import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
@Test public void testPrepareAndExecuteSetsMaxRows() throws Exception {
  final String id = UUID.randomUUID().toString();
  final int statementId = 12345;
  final String sql = "SELECT * FROM FOO";
  final int maxRows = 500;

  JdbcMeta meta = Mockito.mock(JdbcMeta.class);
  PreparedStatement statement = Mockito.mock(PreparedStatement.class);
  @SuppressWarnings("unchecked")
  Cache<Integer, StatementInfo> statementCache =
      (Cache<Integer, StatementInfo>) Mockito.mock(Cache.class);
  Signature signature = Mockito.mock(Signature.class);

  final StatementInfo statementInfo = new StatementInfo(statement);
  final StatementHandle statementHandle = new StatementHandle(id, statementId, signature);

  Mockito.when(meta.getStatementCache()).thenReturn(statementCache);
  Mockito.when(statementCache.getIfPresent(statementId)).thenReturn(statementInfo);
  Mockito.when(statement.getResultSet()).thenReturn(null);
  // The real methods
  Mockito.when(meta.prepareAndExecute(statementHandle, sql, maxRows, 50, null)).
      thenCallRealMethod();
  Mockito.doCallRealMethod().when(meta).setMaxRows(statement, maxRows);

  // Call our method
  meta.prepareAndExecute(statementHandle, sql, maxRows, 50, null);

  // Verify we called setMaxRows with the right value
  Mockito.verify(statement).setMaxRows(maxRows);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
/**
 * Creates an AvaticaPreparedStatement.
 *
 * @param connection Connection
 * @param h Statement handle
 * @param signature Result of preparing statement
 * @param resultSetType Result set type
 * @param resultSetConcurrency Result set concurrency
 * @param resultSetHoldability Result set holdability
 * @throws SQLException If fails due to underlying implementation reasons.
 */
protected AvaticaPreparedStatement(AvaticaConnection connection,
    Meta.StatementHandle h,
    Meta.Signature signature,
    int resultSetType,
    int resultSetConcurrency,
    int resultSetHoldability) throws SQLException {
  super(connection, h, resultSetType, resultSetConcurrency,
      resultSetHoldability, signature);
  this.slots = new TypedValue[signature.parameters.size()];
  this.resultSetMetaData =
      connection.factory.newResultSetMetaData(this, signature);
  this.parameterValueBatch = new ArrayList<>();
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
/**
 *
 * @param statement
 * @param signature
 */
DremioCursor(DremioConnectionImpl connection, AvaticaStatement statement, Signature signature) {
  this.connection = connection;
  this.statement = statement;
  this.signature = signature;

  DremioClient client = connection.getClient();
  final int batchQueueThrottlingThreshold =
      client.getConfig().getInt(
          ExecConstants.JDBC_BATCH_QUEUE_THROTTLING_THRESHOLD );
  resultsListener = new ResultsListener(batchQueueThrottlingThreshold);
  currentBatchHolder = new RecordBatchLoader(client.getAllocator());
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
/**
 *
 * @param statement
 * @param signature
 */
DrillCursor(DrillConnectionImpl connection, AvaticaStatement statement, Signature signature) {
  this.connection = connection;
  this.statement = statement;
  this.signature = signature;

  DrillClient client = connection.getClient();
  final int batchQueueThrottlingThreshold =
      client.getConfig().getInt(
          ExecConstants.JDBC_BATCH_QUEUE_THROTTLING_THRESHOLD );
  resultsListener = new ResultsListener(batchQueueThrottlingThreshold);
  currentBatchHolder = new RecordBatchLoader(client.getAllocator());
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
public ContentMetadata(String scheme, String sql, String catalog, String connectionId, int statementId, boolean assignColumnNames, boolean odbcCompatible)
		throws AtsdException, IOException {
	this.metadataList = StringUtils.isNotEmpty(scheme) ? buildMetadataList(scheme, catalog, assignColumnNames, odbcCompatible)
			: Collections.<ColumnMetaData>emptyList();
	this.sign = new Signature(metadataList, sql, Collections.<AvaticaParameter>emptyList(), null, CursorFactory.LIST,
			StatementType.SELECT);
	this.list = Collections.unmodifiableList(
			Collections.singletonList(MetaResultSet.create(connectionId, statementId, false, sign, null)));
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
public AtsdResultSet(AvaticaStatement statement, QueryState state, Signature signature,
					 ResultSetMetaData resultSetMetaData, TimeZone timeZone, Frame firstFrame) {
	super(statement, state, signature, resultSetMetaData, timeZone, firstFrame);
	final AtsdConnection connection = (AtsdConnection) statement.connection;
	this.meta = connection.getMeta();
	this.handle = statement.handle;
	this.context = meta.getContextFromMap(statement.handle);
	logger.trace("[ctor] {}", this.handle.id);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
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);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
public ResultSet executeQuery() throws SQLException {
  this.updateCount = -1;
  final Signature sig = getSignature();
  return getConnection().executeQueryInternal(this, sig, null,
      new QueryState(sig.sql), false);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
/**
 * Generates a collection of Requests whose serialization will be tested.
 */
private static List<Request> getRequests() {
  LinkedList<Request> requests = new LinkedList<>();

  requests.add(new CatalogsRequest());
  requests.add(new DatabasePropertyRequest());
  requests.add(new SchemasRequest("connectionId", "catalog", "schemaPattern"));
  requests.add(
      new TablesRequest("connectionId", "catalog", "schemaPattern", "tableNamePattern",
          Arrays.asList("STRING", "BOOLEAN", "INT")));
  requests.add(new TableTypesRequest());
  requests.add(
      new ColumnsRequest("connectionId", "catalog", "schemaPattern", "tableNamePattern",
          "columnNamePattern"));
  requests.add(new TypeInfoRequest());
  requests.add(
      new PrepareAndExecuteRequest("connectionId", Integer.MAX_VALUE, "sql",
          Long.MAX_VALUE));
  requests.add(new PrepareRequest("connectionId", "sql", Long.MAX_VALUE));

  List<TypedValue> paramValues =
      Arrays.asList(TypedValue.create(Rep.BOOLEAN.name(), Boolean.TRUE),
          TypedValue.create(Rep.STRING.name(), "string"));
  FetchRequest fetchRequest = new FetchRequest("connectionId", Integer.MAX_VALUE,
      Long.MAX_VALUE, Integer.MAX_VALUE);
  requests.add(fetchRequest);

  requests.add(new CreateStatementRequest("connectionId"));
  requests.add(new CloseStatementRequest("connectionId", Integer.MAX_VALUE));
  Map<String, String> info = new HashMap<>();
  info.put("param1", "value1");
  info.put("param2", "value2");
  requests.add(new OpenConnectionRequest("connectionId", info));
  requests.add(new CloseConnectionRequest("connectionId"));
  requests.add(
      new ConnectionSyncRequest("connectionId",
          new ConnectionPropertiesImpl(Boolean.FALSE, Boolean.FALSE,
              Integer.MAX_VALUE, "catalog", "schema")));

  requests.add(new SyncResultsRequest("connectionId", 12345, getSqlQueryState(), 150));
  requests.add(new SyncResultsRequest("connectionId2", 54321, getMetadataQueryState1(), 0));
  requests.add(new SyncResultsRequest("connectionId3", 5, getMetadataQueryState2(), 10));

  requests.add(new CommitRequest("connectionId"));
  requests.add(new RollbackRequest("connectionId"));

  // ExecuteBatchRequest omitted because of the special protobuf conversion it does

  List<String> commands = Arrays.asList("command1", "command2", "command3");
  requests.add(new PrepareAndExecuteBatchRequest("connectionId", 12345, commands));


  List<ColumnMetaData> columns = Collections.emptyList();
  List<AvaticaParameter> params = Collections.emptyList();
  Meta.CursorFactory cursorFactory = Meta.CursorFactory.create(Style.LIST, Object.class,
      Collections.<String>emptyList());
  Signature signature = Signature.create(columns, "sql", params, cursorFactory,
      Meta.StatementType.SELECT);
  Meta.StatementHandle handle = new Meta.StatementHandle("1234", 1, signature);
  requests.add(new ExecuteRequest(handle, Arrays.<TypedValue>asList((TypedValue) null), 10));
  requests.add(new ExecuteRequest(handle, Arrays.asList(TypedValue.EXPLICIT_NULL), 10));

  return requests;
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
private Signature getSignature() {
  return null;
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
protected AtsdPreparedStatement(AvaticaConnection connection, StatementHandle h, Signature signature,
								int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException {
	super(connection, h, signature, resultSetType, resultSetConcurrency, resultSetHoldability);
	logger.trace("[new] {}", this.handle.id);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
protected AtsdStatement(AvaticaConnection connection, StatementHandle statementHandle, int resultSetType,
		int resultSetConcurrency, int resultSetHoldability, Signature signature) {
	super(connection, statementHandle, resultSetType, resultSetConcurrency, resultSetHoldability, signature);
	logger.trace("[AtsdStatement#new] {}", this.handle.id);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
protected KylinPreparedStatement(AvaticaConnection connection, StatementHandle h, Signature signature, int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException {
    super(connection, h, signature, resultSetType, resultSetConcurrency, resultSetHoldability);
    if (this.handle.signature == null)
        this.handle.signature = signature;
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
@Override
public AvaticaPreparedStatement newPreparedStatement(AvaticaConnection connection, StatementHandle h, Signature signature, int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException {
    return new KylinPreparedStatement((KylinConnection) connection, h, signature, resultSetType, resultSetConcurrency, resultSetHoldability);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
@Override
public AvaticaResultSet newResultSet(AvaticaStatement statement, QueryState state, Signature signature, TimeZone timeZone, Frame firstFrame) throws SQLException {
    AvaticaResultSetMetaData resultSetMetaData = new AvaticaResultSetMetaData(statement, null, signature);
    return new KylinResultSet(statement, state, signature, resultSetMetaData, timeZone, firstFrame);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
@Override
public ResultSetMetaData newResultSetMetaData(AvaticaStatement statement, Signature signature) throws SQLException {
    return new AvaticaResultSetMetaData(statement, null, signature);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
public KylinResultSet(AvaticaStatement statement, QueryState state, Signature signature, ResultSetMetaData resultSetMetaData, TimeZone timeZone, Frame firstFrame) {
    super(statement, state, signature, resultSetMetaData, timeZone, firstFrame);
}
 

import org.apache.calcite.avatica.Meta.Signature; //导入依赖的package包/类
@Override
public PreparedStatement prepareStatement(String sql, int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException {
    Meta.Signature sig = mockPreparedSignature(sql);
    return factory().newPreparedStatement(this, null, sig, resultSetType, resultSetConcurrency, resultSetHoldability);
}