Java Code Examples for org.apache.calcite.avatica.Meta#Signature

private CalcitePreparedStatement prepareStatement_(
    CalcitePrepare.Query<?> query,
    int resultSetType,
    int resultSetConcurrency,
    int resultSetHoldability) throws SQLException {
  try {
    final Meta.Signature signature =
        parseQuery(query, createPrepareContext(), -1);
    final CalcitePreparedStatement calcitePreparedStatement =
        (CalcitePreparedStatement) factory.newPreparedStatement(this, null,
            signature, resultSetType, resultSetConcurrency, resultSetHoldability);
    server.getStatement(calcitePreparedStatement.handle).setSignature(signature);
    return calcitePreparedStatement;
  } catch (Exception e) {
    throw Helper.INSTANCE.createException(
        "Error while preparing statement [" + query.sql + "]", e);
  }
}
 

public CalciteResultSet newResultSet(AvaticaStatement statement, QueryState state,
    Meta.Signature signature, TimeZone timeZone, Meta.Frame firstFrame)
    throws SQLException {
  final ResultSetMetaData metaData =
      newResultSetMetaData(statement, signature);
  final CalcitePrepare.CalciteSignature calciteSignature =
      (CalcitePrepare.CalciteSignature) signature;
  return new CalciteResultSet(statement, calciteSignature, metaData, timeZone,
      firstFrame);
}
 

Meta.StatementHandle finagle(Meta.StatementHandle h) {
  final Meta.Signature signature = finagle(h.signature);
  if (signature == h.signature) {
    return h;
  }
  return new Meta.StatementHandle(h.connectionId, h.id, signature);
}
 

@Override
public DremioResultSetImpl newResultSet(AvaticaStatement statement,
                                       QueryState state,
                                       Meta.Signature signature,
                                       TimeZone timeZone,
                                       Meta.Frame firstFrame) throws SQLException {
  final ResultSetMetaData metaData = newResultSetMetaData(statement, signature);
  return new DremioResultSetImpl(statement, state, signature, metaData, timeZone, firstFrame);
}
 

protected QuarkMetaResultSet(String connectionId,
                             int statementId,
                             boolean ownStatement,
                             Meta.Signature signature,
                             Meta.Frame firstFrame,
                             long updateCount) {
  super(connectionId, statementId, ownStatement, signature, firstFrame, updateCount);
}
 

protected DremioPreparedStatementImpl(DremioConnectionImpl connection,
                                     StatementHandle h,
                                     Meta.Signature signature,
                                     PreparedStatement preparedStatementHandle,
                                     int resultSetType,
                                     int resultSetConcurrency,
                                     int resultSetHoldability) throws SQLException {
  super(connection, h, signature,
        resultSetType, resultSetConcurrency, resultSetHoldability);
  connection.openStatementsRegistry.addStatement(this);
  this.preparedStatementHandle = preparedStatementHandle;
  if (preparedStatementHandle != null) {
    ((DremioColumnMetaDataList) signature.columns).updateColumnMetaData(preparedStatementHandle.getColumnsList());
  }
}
 

/** Creates a result set with maxRowCount.
 *
 * <p>If {@code maxRowCount} is -2 ({@link JdbcMeta#UNLIMITED_COUNT}),
 * returns an unlimited number of rows in a single frame; any other
 * negative value (typically -1) returns an unlimited number of rows
 * in frames of the default frame size. */
public static JdbcResultSet create(String connectionId, int statementId,
    ResultSet resultSet, int maxRowCount) {
  try {
    Meta.Signature sig = JdbcMeta.signature(resultSet.getMetaData());
    return create(connectionId, statementId, resultSet, maxRowCount, sig);
  } catch (SQLException e) {
    throw new RuntimeException(e);
  }
}
 

Signature mockPreparedSignature(String sql) {
    List<AvaticaParameter> params = new ArrayList<AvaticaParameter>();
    int startIndex = 0;
    while (sql.indexOf("?", startIndex) >= 0) {
        AvaticaParameter param = new AvaticaParameter(false, 0, 0, 0, null, null, null);
        params.add(param);
        startIndex = sql.indexOf("?", startIndex) + 1;
    }

    ArrayList<ColumnMetaData> columns = new ArrayList<ColumnMetaData>();
    Map<String, Object> internalParams = Collections.<String, Object> emptyMap();

    return new Meta.Signature(columns, sql, params, internalParams, CursorFactory.ARRAY, Meta.StatementType.SELECT);
}
 

/**
 * Creates a QuicksqlResultSet.
 */
public QuicksqlResultSet(AvaticaStatement statement,
    Meta.Signature calciteSignature,
    ResultSetMetaData resultSetMetaData, TimeZone timeZone,
    Meta.Frame firstFrame) throws SQLException {
    super(statement, null, calciteSignature, resultSetMetaData, timeZone, firstFrame);
}
 

/** Creates a frame containing a given number or unlimited number of rows
 * from a result set. */
static Meta.Frame frame(StatementInfo info, ResultSet resultSet, long offset,
    int fetchMaxRowCount, Calendar calendar, Optional<Meta.Signature> sig) throws SQLException {
  final ResultSetMetaData metaData = resultSet.getMetaData();
  final int columnCount = metaData.getColumnCount();
  final int[] types = new int[columnCount];
  Set<Integer> arrayOffsets = new HashSet<>();
  for (int i = 0; i < types.length; i++) {
    types[i] = metaData.getColumnType(i + 1);
    if (Types.ARRAY == types[i]) {
      arrayOffsets.add(i);
    }
  }
  final List<Object> rows = new ArrayList<>();
  // Meta prepare/prepareAndExecute 0 return 0 row and done
  boolean done = fetchMaxRowCount == 0;
  for (int i = 0; fetchMaxRowCount < 0 || i < fetchMaxRowCount; i++) {
    final boolean hasRow;
    if (null != info) {
      hasRow = info.next();
    } else {
      hasRow = resultSet.next();
    }
    if (!hasRow) {
      done = true;
      resultSet.close();
      break;
    }
    Object[] columns = new Object[columnCount];
    for (int j = 0; j < columnCount; j++) {
      columns[j] = getValue(resultSet, types[j], j, calendar);
      if (arrayOffsets.contains(j)) {
        // If we have an Array type, our Signature is lacking precision. We can't extract the
        // component type of an Array from metadata, we have to update it as we're serializing
        // the ResultSet.
        final Array array = resultSet.getArray(j + 1);
        // Only attempt to determine the component type for the array when non-null
        if (null != array && sig.isPresent()) {
          ColumnMetaData columnMetaData = sig.get().columns.get(j);
          ArrayType arrayType = (ArrayType) columnMetaData.type;
          SqlType componentSqlType = SqlType.valueOf(array.getBaseType());

          // Avatica Server will always return non-primitives to ensure nullable is guaranteed.
          ColumnMetaData.Rep rep = ColumnMetaData.Rep.serialRepOf(componentSqlType);
          AvaticaType componentType = ColumnMetaData.scalar(array.getBaseType(),
              array.getBaseTypeName(), rep);
          // Update the ArrayType from the Signature
          arrayType.updateComponentType(componentType);

          // We only need to update the array's type once.
          arrayOffsets.remove(j);
        }
      }
    }
    rows.add(columns);
  }
  return new Meta.Frame(offset, done, rows);
}
 

public ResultSetMetaData newResultSetMetaData(AvaticaStatement statement,
    Meta.Signature signature) {
  return new AvaticaResultSetMetaData(statement, null, signature);
}
 

protected JdbcResultSet(String connectionId, int statementId,
    boolean ownStatement, Meta.Signature signature, Meta.Frame firstFrame) {
  this(connectionId, statementId, ownStatement, signature, firstFrame, -1L);
}
 

QuarkResultSet(AvaticaStatement statement,
               Meta.Signature signature,
               ResultSetMetaData resultSetMetaData, TimeZone timeZone,
               Meta.Frame firstFrame) {
  super(statement, null, signature, resultSetMetaData, timeZone, firstFrame);
}
 

public void setSignature(Meta.Signature signature) {
  this.signature = signature;
}
 

@Override
public ResultSetMetaData newResultSetMetaData(AvaticaStatement statement,
                                              Meta.Signature signature) {
  return new AvaticaResultSetMetaData(statement, null, signature);
}
 

private QuarkMetaResultSet getMetaResultSetFromIterator(Iterator<Object> iterator,
                                                QuarkConnectionImpl connection,
                                                ParserResult result,
                                                String sql,
                                                QuarkJdbcStatement stmt,
                                                Meta.StatementHandle h,
                                                long maxRowCount,
                                                Class clazz) throws SQLException {
  QuarkMetaResultSet metaResultSet;
  final JavaTypeFactory typeFactory =
      connection.getSqlQueryParser().getTypeFactory();
  final RelDataType x;
  switch (result.getKind()) {
    case INSERT:
    case EXPLAIN:
      x = RelOptUtil.createDmlRowType(result.getKind(), typeFactory);
      break;
    case OTHER_DDL:
      x = getRowType(clazz);
      break;
    default:
      x = result.getRelNode().getRowType();
  }
  RelDataType jdbcType = makeStruct(typeFactory, x);
  final List<ColumnMetaData> columns =
      getColumnMetaDataList(typeFactory, x, jdbcType);
  Meta.Signature signature = new Meta.Signature(columns,
      sql,
      new ArrayList<AvaticaParameter>(),
      new HashMap<String, Object>(),
      Meta.CursorFactory.ARRAY,
      Meta.StatementType.SELECT);
  stmt.setSignature(signature);
  stmt.setResultSet(iterator);
  if (signature.statementType.canUpdate()) {
    metaResultSet = QuarkMetaResultSet.count(h.connectionId, h.id,
        ((Number) iterator.next()).intValue());
  } else {
    metaResultSet = QuarkMetaResultSet.create(h.connectionId, h.id,
        iterator, maxRowCount, signature);
  }
  return metaResultSet;
}
 

public Meta.Signature getSignature() {
  return signature;
}
 

public ResultSetMetaData newResultSetMetaData(AvaticaStatement statement,
    Meta.Signature signature) {
  return new AvaticaResultSetMetaData(statement, null, signature);
}
 

void setSignature(Meta.Signature signature); 

void setSignature(Meta.Signature signature);