@Override public void addResults(QueryStatusInfo statusInfo, QueryData data) { if (!loggedUri.getAndSet(true)) { log.info("Query %s: %s", statusInfo.getId(), statusInfo.getInfoUri()); } if (types.get() == null && statusInfo.getColumns() != null) { types.set(getTypes(statusInfo.getColumns())); } if (data.getData() != null) { checkState(types.get() != null, "data received without types"); rows.addAll(transform(data.getData(), dataToRow(types.get()))); } }
@Override public void addResults(QueryStatusInfo statusInfo, QueryData data) { if (types.get() == null && statusInfo.getColumns() != null) { types.set(getTypes(statusInfo.getColumns())); } if (data.getData() != null) { checkState(types.get() != null, "Data without types received!"); List<Column> columns = statusInfo.getColumns(); for (List<Object> fields : data.getData()) { ImmutableMap.Builder<String, Object> builder = ImmutableMap.builder(); for (int i = 0; i < fields.size(); i++) { Type type = types.get().get(i); Object value = convertValue(fields.get(i), type); if (value != null) { builder.put(columns.get(i).getName(), value); } } producer.send(new KeyedMessage<>(topicName, count.getAndIncrement(), builder.build())); } } }
@Override public void addResults(QueryStatusInfo statusInfo, QueryData data) { if (types.get() == null && statusInfo.getColumns() != null) { types.set(getTypes(statusInfo.getColumns())); } if (data.getData() == null) { return; } checkState(types.get() != null, "Type information is missing"); List<Column> columns = statusInfo.getColumns(); for (List<Object> fields : data.getData()) { try { XContentBuilder dataBuilder = jsonBuilder().startObject(); for (int i = 0; i < fields.size(); i++) { Type type = types.get().get(i); Object value = convertValue(fields.get(i), type); dataBuilder.field(columns.get(i).getName(), value); } dataBuilder.endObject(); client.prepareIndex(tableName, "doc") .setSource(dataBuilder.string(), JSON) .get(); } catch (IOException e) { throw new UncheckedIOException("Error loading data into Elasticsearch index: " + tableName, e); } } }
@Override public void addResults(QueryStatusInfo statusInfo, QueryData data) if (types.get() == null && statusInfo.getColumns() != null) { types.set(getTypes(statusInfo.getColumns())); List<Column> columns = statusInfo.getColumns(); for (List<Object> fields : data.getData()) { String redisKey = tableName + ":" + count.getAndIncrement();
renderUpdate(errorChannel, results); else if (results.getColumns() == null) { errorChannel.printf("Query %s has no columns\n", results.getId()); return false; renderResults(out, outputFormat, interactive, results.getColumns());
private void transformAndAdd() { QueryStatusInfo queryStatusInfo = client.isValid() ? client.currentStatusInfo() : client.finalStatusInfo(); if (queryStatusInfo.getError() != null || queryStatusInfo.getColumns() == null || !client.isValid()) { return; columns = queryStatusInfo.getColumns().stream() .map(c -> { List<ClientTypeSignatureParameter> arguments = c.getTypeSignature().getArguments(); String type = queryStatusInfo.getColumns().get(i).getTypeSignature().getRawType(); Object value = objects.get(i); if (value != null) {
@Override public void addResults(QueryStatusInfo statusInfo, QueryData data) { if (!loggedUri.getAndSet(true)) { log.info("Query %s: %s", statusInfo.getId(), statusInfo.getInfoUri()); } if (types.get() == null && statusInfo.getColumns() != null) { types.set(getTypes(statusInfo.getColumns())); } if (data.getData() != null) { checkState(types.get() != null, "data received without types"); rows.addAll(transform(data.getData(), dataToRow(types.get()))); } }
renderUpdate(errorChannel, results); else if (results.getColumns() == null) { errorChannel.printf("Query %s has no columns\n", results.getId()); return false; renderResults(out, outputFormat, interactive, results.getColumns());