&& ( qrb.getHeader().getRowCount() == 0 || qrb.getData() == null ) ) { schemaChanged = currentBatchHolder.load(qrb.getHeader().getDef(), qrb.getData());
RecordBatchDef batchDef = result.getHeader().getDef(); final List<SerializedField> fields = batchDef.getFieldList(); final ByteBuf[] oldBuffers = result.getBuffers(); return new QueryWritableBatch(QueryData.newBuilder(result.getHeader()).setDef(patchedDef).build(), patchedBuffers.toArray(new ArrowBuf[0]));
/** * Maps internal low-level API protocol to {@link UserResultsListener}-level API protocol. * handles query data messages */ public void batchArrived( ConnectionThrottle throttle, byte[] pBody, ByteBuf dBody ) throws RpcException { final QueryData queryData = RpcBus.get( pBody, QueryData.PARSER ); // Current batch coming in. final ArrowBuf ArrowBuf = (ArrowBuf) dBody; final QueryDataBatch batch = new QueryDataBatch( queryData, ArrowBuf ); final QueryId queryId = queryData.getQueryId(); if (logger.isDebugEnabled()) { logger.debug("batchArrived: queryId = {}", QueryIdHelper.getQueryId(queryId)); } logger.trace( "batchArrived: batch = {}", batch ); final UserResultsListener resultsListener = newUserResultsListener(queryId); // A data case--pass on via dataArrived try { resultsListener.dataArrived(batch, throttle); // That releases batch if successful. } catch ( Exception e ) { batch.release(); resultsListener.submissionFailed(UserException.systemError(e).build(logger)); } }
for(QueryDataBatch b : results) { if (b.getData() != null) { int rows = b.getHeader().getRowCount(); count += rows; RecordBatchLoader loader = new RecordBatchLoader(bootStrapContext.getAllocator()); loader.load(b.getHeader().getDef(), b.getData()); BigIntVector vv1 = loader.getValueAccessorById(BigIntVector.class, loader.getValueVectorId( new SchemaPath("col1")).getFieldIds()).getValueVector();
assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData())); boolean firstColumn = true;
int recordCount = 0; for (final QueryDataBatch batch : results) { final boolean schemaChanged = batchLoader.load(batch.getHeader().getDef(), batch.getData()); boolean firstColumn = true;
batchLoader.load(result.getHeader().getDef(), result.getData()); if (batchLoader.getRecordCount() <= 0) { result.release();
int columnValCounter = 0; FieldInfo currentField; count += result.getHeader().getRowCount(); boolean schemaChanged = false; final RecordBatchLoader batchLoader = new RecordBatchLoader(allocator); try { schemaChanged = batchLoader.load(result.getHeader().getDef(), result.getData());
assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));
for (final QueryDataBatch b : results) { final QueryData queryData = b.getHeader(); final int batchRowCount = queryData.getRowCount(); count += batchRowCount; batchLoader.load(queryData.getDef(), b.getData()); for (final VectorWrapper vw : batchLoader) { final ValueVector vv = vw.getValueVector();
int count = 0; for (QueryDataBatch b : results) { if (b.getHeader().getRowCount() != 0) { count += b.getHeader().getRowCount(); if (b.getHeader().getRowCount() == 0) { continue; loader.load(b.getHeader().getDef(),b.getData()); BigIntVector c1 = loader.getValueAccessorById(BigIntVector.class, loader.getValueVectorId(new SchemaPath("blue")).getFieldIds()).getValueVector();
assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));
batchLoader.load(result.getHeader().getDef(), result.getData()); if (batchLoader.getRecordCount() <= 0) { result.release();
int count = 0; for (QueryDataBatch b : results) { if (b.getHeader().getRowCount() != 0) { count += b.getHeader().getRowCount(); if (b.getHeader().getRowCount() == 0) { break; loader.load(b.getHeader().getDef(),b.getData()); BigIntVector c1 = loader.getValueAccessorById(BigIntVector.class, loader.getValueVectorId(new SchemaPath("blue")).getFieldIds()).getValueVector();
int count = 0; for (QueryDataBatch b : results) { if (b.getHeader().getRowCount() != 0) { count += b.getHeader().getRowCount(); if (b.getHeader().getRowCount() == 0) { break; loader.load(b.getHeader().getDef(),b.getData()); BigIntVector c1 = loader.getValueAccessorById(BigIntVector.class, loader.getValueVectorId(new SchemaPath("blue")).getFieldIds()).getValueVector();
actual = BaseTestQuery.testRunAndReturn(queryType, query); QueryDataBatch batch = actual.get(0); loader.load(batch.getHeader().getDef(), batch.getData());
assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));
assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));
int count = 0; for (QueryDataBatch b : results) { System.out.println(String.format("Got %d results", b.getHeader().getRowCount())); count += b.getHeader().getRowCount(); loader.load(b.getHeader().getDef(), b.getData()); for (VectorWrapper vw : loader) { System.out.print(vw.getValueVector().getField().getName() + ": ");
assertTrue(batchLoader.load(batch.getHeader().getDef(), batch.getData()));