@Override public void sendData(RpcOutcomeListener<Ack> listener, QueryWritableBatch result) { // Save the query results schema and release the buffers. if (fields == null) { fields = result.getHeader().getDef().getFieldList(); } for (ByteBuf buf : result.getBuffers()) { buf.release(); } listener.success(Acks.OK, null); }
final List<SerializedField> fields = def.getFieldList(); int bufOffset = 0; for (final SerializedField field : fields) {
fieldList = rawBatch.getHeader().getDef().getFieldList();
List<SerializedField> fields = def.getFieldList();
final List<SerializedField> fieldList = batchDef.getFieldList(); for (SerializedField metaData : fieldList) { final int dataLength = metaData.getBufferLength();
for (final SerializedField field : batch.getHeader().getDef().getFieldList()) { @SuppressWarnings("resource") final ValueVector v = outgoingContainer.addOrGet(MaterializedField.create(field));
@SuppressWarnings("resource") private void readVectors(InputStream input, RecordBatchDef batchDef) throws IOException { final VectorContainer container = new VectorContainer(); final List<ValueVector> vectorList = Lists.newArrayList(); final List<SerializedField> fieldList = batchDef.getFieldList(); for (SerializedField metaData : fieldList) { final int dataLength = metaData.getBufferLength(); final MaterializedField field = MaterializedField.create(metaData); final DrillBuf buf = allocator.read(dataLength, input); final ValueVector vector = TypeHelper.getNewVector(field, allocator); vector.load(metaData, buf); buf.release(); // Vector now owns the buffer vectorList.add(vector); } container.addCollection(vectorList); container.buildSchema(svMode); container.setRecordCount(recordCount); va = container; }
public void writeTo(com.dyuproject.protostuff.Output output, org.apache.drill.exec.proto.UserBitShared.RecordBatchDef message) throws java.io.IOException { if(message.hasRecordCount()) output.writeInt32(1, message.getRecordCount(), false); for(org.apache.drill.exec.proto.UserBitShared.SerializedField field : message.getFieldList()) output.writeObject(2, field, org.apache.drill.exec.proto.SchemaUserBitShared.SerializedField.WRITE, true); if(message.hasCarriesTwoByteSelectionVector()) output.writeBool(3, message.getCarriesTwoByteSelectionVector(), false); if(message.hasAffectedRowsCount()) output.writeInt32(4, message.getAffectedRowsCount(), false); } public boolean isInitialized(org.apache.drill.exec.proto.UserBitShared.RecordBatchDef message)
public void writeTo(com.dyuproject.protostuff.Output output, org.apache.drill.exec.proto.UserBitShared.RecordBatchDef message) throws java.io.IOException { if(message.hasRecordCount()) output.writeInt32(1, message.getRecordCount(), false); for(org.apache.drill.exec.proto.UserBitShared.SerializedField field : message.getFieldList()) output.writeObject(2, field, org.apache.drill.exec.proto.SchemaUserBitShared.SerializedField.WRITE, true); if(message.hasCarriesTwoByteSelectionVector()) output.writeBool(3, message.getCarriesTwoByteSelectionVector(), false); if(message.hasAffectedRowsCount()) output.writeInt32(4, message.getAffectedRowsCount(), false); } public boolean isInitialized(org.apache.drill.exec.proto.UserBitShared.RecordBatchDef message)