.setQueryId(context.getHandle().getQueryId()) .setRowCount(0) .setDef(RecordBatchDef.getDefaultInstance())
public void mergeFrom(com.dyuproject.protostuff.Input input, org.apache.drill.exec.proto.UserBitShared.QueryData.Builder builder) throws java.io.IOException { for(int number = input.readFieldNumber(this);; number = input.readFieldNumber(this)) { switch(number) { case 0: return; case 1: builder.setQueryId(input.mergeObject(org.apache.drill.exec.proto.UserBitShared.QueryId.newBuilder(), org.apache.drill.exec.proto.SchemaUserBitShared.QueryId.MERGE)); break; case 2: builder.setRowCount(input.readInt32()); break; case 3: builder.setDef(input.mergeObject(org.apache.drill.exec.proto.UserBitShared.RecordBatchDef.newBuilder(), org.apache.drill.exec.proto.SchemaUserBitShared.RecordBatchDef.MERGE)); break; case 4: builder.setAffectedRowsCount(input.readInt32()); break; default: input.handleUnknownField(number, this); } } } public boolean isInitialized(org.apache.drill.exec.proto.UserBitShared.QueryData.Builder builder)
public QueryWritableBatch convertNext() { WritableBatch w = batch.getWritableBatch().transfer(allocator); QueryData.Builder builder = QueryData.newBuilder() .setQueryId(queryId) .setRowCount(batch.getRecordCount()) .setDef(w.getDef()); if (!options.getBoolean(ExecConstants.RETURN_RESULT_SET_FOR_DDL)) { int count = w.getDef().getAffectedRowsCount(); builder.setAffectedRowsCount(count == -1 ? 0 : count); } return new QueryWritableBatch(builder.build(), w.getBuffers()); } }
public void mergeFrom(com.dyuproject.protostuff.Input input, org.apache.drill.exec.proto.UserBitShared.QueryData.Builder builder) throws java.io.IOException { for(int number = input.readFieldNumber(this);; number = input.readFieldNumber(this)) { switch(number) { case 0: return; case 1: builder.setQueryId(input.mergeObject(org.apache.drill.exec.proto.UserBitShared.QueryId.newBuilder(), org.apache.drill.exec.proto.SchemaUserBitShared.QueryId.MERGE)); break; case 2: builder.setRowCount(input.readInt32()); break; case 3: builder.setDef(input.mergeObject(org.apache.drill.exec.proto.UserBitShared.RecordBatchDef.newBuilder(), org.apache.drill.exec.proto.SchemaUserBitShared.RecordBatchDef.MERGE)); break; case 4: builder.setAffectedRowsCount(input.readInt32()); break; default: input.handleUnknownField(number, this); } } } public boolean isInitialized(org.apache.drill.exec.proto.UserBitShared.QueryData.Builder builder)