public StatementClient startQuery(ClientSession session, String query) { return newStatementClient(httpClient, session, query); }
private StatementClient startInternalQuery(ClientSession session, String query) { OkHttpClient.Builder builder = httpClient.newBuilder(); sslSetup.accept(builder); OkHttpClient client = builder.build(); return newStatementClient(client, session, query); }
private StatementClient startInternalQuery(ClientSession session, String query) { OkHttpClient.Builder builder = httpClient.newBuilder(); sslSetup.accept(builder); OkHttpClient client = builder.build(); return newStatementClient(client, session, query); }
private StatementStats execute(ClientSession session, String query, Consumer<QueryData> queryDataConsumer, Consumer<QueryError> queryErrorConsumer) { // start query try (StatementClient client = newStatementClient(okHttpClient, session, query)) { // read query output while (client.isRunning()) { queryDataConsumer.accept(client.currentData()); if (!client.advance()) { break; } } // verify final state if (client.isClientAborted()) { throw new IllegalStateException("Query aborted by user"); } if (client.isClientError()) { throw new IllegalStateException("Query is gone (server restarted?)"); } verify(client.isFinished()); QueryError resultsError = client.finalStatusInfo().getError(); if (resultsError != null) { queryErrorConsumer.accept(resultsError); } return client.finalStatusInfo().getStats(); } }
try (StatementClient client = newStatementClient(httpClient, clientSession, sql)) { while (client.isRunning()) { resultsSession.addResults(client.currentStatusInfo(), client.currentData());
StatementClient client = newStatementClient(httpClient, clientSession, sql);