Java Cursor类代码示例

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
public void close() {
    closed = true;
    final Cursor cursor = this.cursor;
    if (cursor != null) {
      this.cursor = null;
      cursor.close();
    }
    statement.onResultSetClose(this);
    // TODO: for timeout, see IteratorResultSet.close
/*
        if (timeoutCursor != null) {
            final long noTimeout = 0;
            timeoutCursor.close(noTimeout);
            timeoutCursor = null;
        }
*/
  }
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
@Override
public ResultSet create(ColumnMetaData.AvaticaType elementType,
                        Iterable<Object> iterable) {
    final List<ColumnMetaData> columnMetaDataList;
    if (elementType instanceof ColumnMetaData.StructType) {
        columnMetaDataList = ((ColumnMetaData.StructType) elementType).columns;
    } else {
        columnMetaDataList =
                ImmutableList.of(ColumnMetaData.dummy(elementType, false));
    }
    final CalcitePrepare.CalciteSignature signature =
            (CalcitePrepare.CalciteSignature) this.signature;
    final CalcitePrepare.CalciteSignature<Object> newSignature =
            new CalcitePrepare.CalciteSignature<>(signature.sql,
                    signature.parameters, signature.internalParameters,
                    signature.rowType, columnMetaDataList, Meta.CursorFactory.ARRAY,
                    ImmutableList.<RelCollation>of(), -1, null);
    ResultSetMetaData subResultSetMetaData =
            new AvaticaResultSetMetaData(statement, null, newSignature);
    final CalciteResultSet resultSet =
            new CalciteResultSet(statement, signature, subResultSetMetaData,
                    localCalendar.getTimeZone(), new Meta.Frame(0, true, iterable));
    final Cursor cursor = resultSet.createCursor(elementType, iterable);
    return resultSet.execute2(cursor, columnMetaDataList);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
@Override
public ResultSet create(ColumnMetaData.AvaticaType elementType,
                        Iterable<Object> iterable) {
  final List<ColumnMetaData> columnMetaDataList;
  if (elementType instanceof ColumnMetaData.StructType) {
    columnMetaDataList = ((ColumnMetaData.StructType) elementType).columns;
  } else {
    columnMetaDataList =
        ImmutableList.of(ColumnMetaData.dummy(elementType, false));
  }
  final CalcitePrepare.CalciteSignature signature =
      (CalcitePrepare.CalciteSignature) this.signature;
  final CalcitePrepare.CalciteSignature<Object> newSignature =
      new CalcitePrepare.CalciteSignature<>(signature.sql,
          signature.parameters, signature.internalParameters,
          signature.rowType, columnMetaDataList, Meta.CursorFactory.ARRAY,
          signature.rootSchema, ImmutableList.<RelCollation>of(), -1, null);
  ResultSetMetaData subResultSetMetaData =
      new AvaticaResultSetMetaData(statement, null, newSignature);
  final QuarkResultSet resultSet =
      new QuarkResultSet(statement, signature, subResultSetMetaData,
          localCalendar.getTimeZone(), new Meta.Frame(0, true, iterable));
  final Cursor cursor = resultSet.createCursor(elementType, iterable);
  return resultSet.execute2(cursor, columnMetaDataList);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
@Override public ResultSet create(ColumnMetaData.AvaticaType elementType,
    Iterable<Object> iterable) {
  final List<ColumnMetaData> columnMetaDataList;
  if (elementType instanceof ColumnMetaData.StructType) {
    columnMetaDataList = ((ColumnMetaData.StructType) elementType).columns;
  } else {
    columnMetaDataList =
        ImmutableList.of(ColumnMetaData.dummy(elementType, false));
  }
  final CalcitePrepare.CalciteSignature signature =
      (CalcitePrepare.CalciteSignature) this.signature;
  final CalcitePrepare.CalciteSignature<Object> newSignature =
      new CalcitePrepare.CalciteSignature<>(signature.sql,
          signature.parameters, signature.internalParameters,
          signature.rowType, columnMetaDataList, Meta.CursorFactory.ARRAY,
          signature.rootSchema, ImmutableList.<RelCollation>of(), -1, null,
          statement.getStatementType());
  ResultSetMetaData subResultSetMetaData =
      new AvaticaResultSetMetaData(statement, null, newSignature);
  final CalciteResultSet resultSet =
      new CalciteResultSet(statement, signature, subResultSetMetaData,
          localCalendar.getTimeZone(), new Meta.Frame(0, true, iterable));
  final Cursor cursor = resultSet.createCursor(elementType, iterable);
  return resultSet.execute2(cursor, columnMetaDataList);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
/** Uses a {@link org.apache.calcite.avatica.Meta.CursorFactory} to convert
 * an {@link Iterable} into a
 * {@link org.apache.calcite.avatica.util.Cursor}. */
public static Cursor createCursor(CursorFactory cursorFactory,
    Iterable<Object> iterable) {
  switch (cursorFactory.style) {
  case OBJECT:
    return new IteratorCursor<Object>(iterable.iterator()) {
      protected Getter createGetter(int ordinal) {
        return new ObjectGetter(ordinal);
      }
    };
  case ARRAY:
    @SuppressWarnings("unchecked") final Iterable<Object[]> iterable1 =
        (Iterable<Object[]>) (Iterable) iterable;
    return new ArrayIteratorCursor(iterable1.iterator());
  case RECORD:
    @SuppressWarnings("unchecked") final Class<Object> clazz =
        cursorFactory.clazz;
    return new RecordIteratorCursor<Object>(iterable.iterator(), clazz);
  case RECORD_PROJECTION:
    @SuppressWarnings("unchecked") final Class<Object> clazz2 =
        cursorFactory.clazz;
    return new RecordIteratorCursor<Object>(iterable.iterator(), clazz2,
        cursorFactory.fields);
  case LIST:
    @SuppressWarnings("unchecked") final Iterable<List<Object>> iterable2 =
        (Iterable<List<Object>>) (Iterable) iterable;
    return new ListIteratorCursor(iterable2.iterator());
  case MAP:
    @SuppressWarnings("unchecked") final Iterable<Map<String, Object>>
        iterable3 =
        (Iterable<Map<String, Object>>) (Iterable) iterable;
    return new MapIteratorCursor(iterable3.iterator(),
        cursorFactory.fieldNames);
  default:
    throw new AssertionError("unknown style: " + cursorFactory.style);
  }
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
/**
 * Returns the accessor for column with a given index.
 *
 * @param columnIndex 1-based column index
 * @return Accessor
 * @throws SQLException if index is not valid
 */
private Cursor.Accessor getAccessor(int columnIndex) throws SQLException {
  try {
    return accessorList.get(columnIndex - 1);
  } catch (IndexOutOfBoundsException e) {
    throw new SQLException("invalid column ordinal: " + columnIndex);
  }
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
public AvaticaResultSet execute2(Cursor cursor,
    List<ColumnMetaData> columnMetaDataList) {
  this.cursor = cursor;
  this.accessorList =
      cursor.createAccessors(columnMetaDataList, localCalendar, this);
  this.row = -1;
  this.afterLast = false;
  return this;
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
@Override
public Object getObject( int columnIndex ) throws SQLException {
  throwIfClosed();

  final Cursor.Accessor accessor;
  try {
    accessor = accessorList.get(columnIndex - 1);
  } catch (IndexOutOfBoundsException e) {
    throw new SQLException("invalid column ordinal: " + columnIndex);
  }
  final ColumnMetaData metaData = columnMetaDataList.get(columnIndex - 1);
  // Dremio returns a float (4bytes) for a SQL Float whereas Calcite would return a double (8bytes)
  int typeId = (metaData.type.id != Types.FLOAT) ? metaData.type.id : Types.REAL;
  return AvaticaSite.get(accessor, typeId, localCalendar);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
@Override
public Object getObject( int columnIndex ) throws SQLException {
  throwIfClosed();

  final Cursor.Accessor accessor;
  try {
    accessor = accessorList.get(columnIndex - 1);
  } catch (IndexOutOfBoundsException e) {
    throw new SQLException("invalid column ordinal: " + columnIndex);
  }
  final ColumnMetaData metaData = columnMetaDataList.get(columnIndex - 1);
  // Drill returns a float (4bytes) for a SQL Float whereas Calcite would return a double (8bytes)
  int typeId = (metaData.type.id != Types.FLOAT) ? metaData.type.id : Types.REAL;
  return AvaticaSite.get(accessor, typeId, localCalendar);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
private Cursor createCursor(ColumnMetaData.AvaticaType elementType,
                            Iterable iterable) {
    final Enumerator enumerator = Linq4j.iterableEnumerator(iterable);
    //noinspection unchecked
    return !(elementType instanceof ColumnMetaData.StructType)
            || ((ColumnMetaData.StructType) elementType).columns.size() == 1
            ? new ObjectEnumeratorCursor(enumerator)
            : new ArrayEnumeratorCursor(enumerator);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
private Cursor createCursor(ColumnMetaData.AvaticaType elementType,
                            Iterable iterable) {
  final Enumerator enumerator = Linq4j.iterableEnumerator(iterable);
  //noinspection unchecked
  return !(elementType instanceof ColumnMetaData.StructType)
      || ((ColumnMetaData.StructType) elementType).columns.size() == 1
      ? new ObjectEnumeratorCursor(enumerator)
      : new ArrayEnumeratorCursor(enumerator);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
private Cursor createCursor(ColumnMetaData.AvaticaType elementType,
    Iterable iterable) {
  final Enumerator enumerator = Linq4j.iterableEnumerator(iterable);
  //noinspection unchecked
  return !(elementType instanceof ColumnMetaData.StructType)
      || ((ColumnMetaData.StructType) elementType).columns.size() == 1
      ? new ObjectEnumeratorCursor(enumerator)
      : new ArrayEnumeratorCursor(enumerator);
}
 

import org.apache.calcite.avatica.util.Cursor; //导入依赖的package包/类
public Object getObject(int columnIndex) throws SQLException {
  final Cursor.Accessor accessor = getAccessor(columnIndex);
  final ColumnMetaData metaData = columnMetaDataList.get(columnIndex - 1);
  return AvaticaSite.get(accessor, metaData.type.id, localCalendar);
}