private void updateScreen() { console.repositionCursor(); printQueryInfo(client.current()); }
private List<String> queryMetadata(String query) { ImmutableList.Builder<String> cache = ImmutableList.builder(); try (StatementClient client = queryRunner.startInternalQuery(query)) { while (client.isValid() && !Thread.currentThread().isInterrupted()) { QueryResults results = client.current(); if (results.getData() != null) { for (List<Object> row : results.getData()) { cache.add((String) row.get(0)); } } client.advance(); } } return cache.build(); }
private void waitForData() { while (client.isValid() && (client.current().getData() == null)) { client.advance(); } }
public void processRows(StatementClient client) throws IOException { while (client.isValid()) { Iterable<List<Object>> data = client.current().getData(); if (data != null) { for (List<Object> row : data) { processRow(unmodifiableList(row)); } } if (nanosSince(bufferStart).compareTo(MAX_BUFFER_TIME) >= 0) { flush(false); } client.advance(); } }
public boolean cancelLeafStage(Duration timeout) { checkState(!isClosed(), "client is closed"); URI uri = current().getPartialCancelUri(); if (uri == null) { return false; } Request request = prepareDelete() .setHeader(USER_AGENT, USER_AGENT_VALUE) .setUri(uri) .build(); HttpResponseFuture<StatusResponse> response = httpClient.executeAsync(request, createStatusResponseHandler()); try { StatusResponse status = response.get(timeout.toMillis(), MILLISECONDS); return familyForStatusCode(status.getStatusCode()) == Family.SUCCESSFUL; } catch (InterruptedException e) { Thread.currentThread().interrupt(); throw Throwables.propagate(e); } catch (ExecutionException e) { throw Throwables.propagate(e.getCause()); } catch (TimeoutException e) { return false; } }
public boolean advance() URI nextUri = current().getNextUri(); if (isClosed() || (nextUri == null)) { valid.set(false);
@Override protected Iterable<List<Object>> computeNext() { while (client.isValid()) { Iterable<List<Object>> data = client.current().getData(); client.advance(); if (data != null) { return data; } } if (client.isFailed()) { throw internalServerError(failureMessage(client.finalResults())); } return endOfData(); } }
private static List<Column> getColumns(StatementClient client) { while (client.isValid()) { List<Column> columns = client.current().getColumns(); if (columns != null) { return columns; } client.advance(); } if (!client.isFailed()) { throw internalServerError("No columns"); } throw internalServerError(failureMessage(client.finalResults())); }
try { if (client.current().getData() != null) { return;
QueryResults results = client.isValid() ? client.current() : client.finalResults(); if (results.getUpdateType() != null) { renderUpdate(errorChannel, results);