在向Arrow Flight Sql中插入数据时,对于BitVector类型的列,需要设置数据大小,并指定每行数据占用的位数。代码示例如下:
private void insertBitVectorData() throws IOException, FlightException {
// 创建BitVector类型的数据
BitVector bitVector = new BitVector(2, allocator);
bitVector.set(0, true);
bitVector.set(1, false);
// 创建Schema,包含一个BitVector类型的列
Field field = Field.nullable("field1", new ArrowType.Bit(2));
Schema schema = new Schema(Collections.singletonList(field));
// 创建RecordBatch,并插入数据
VectorSchemaRoot root = VectorSchemaRoot.create(schema, allocator);
root.setRowCount(2);
root.setFieldData("field1", bitVector);
root.setValidityBuffer("field1", bitVector.getValidityBuffer(), bitVector.getValidityBuffer().capacity());
root.setOffsetBuffer("field1", bitVector.getOffsetBuffer(), bitVector.getOffsetBuffer().capacity());
root.setValueCount(2);
// 将RecordBatch发送到Flight服务器
FlightDescriptor descriptor = FlightDescriptor.path("example");
FlightClient client = FlightClient.builder().location(location).build();
FlightInfo flightInfo = client.getInfo(descriptor);
FlightStream stream = client.getStream(flightInfo.getEndpoints().get(0).getTicket());
VectorSchemaRoot clientRoot = (VectorSchemaRoot) stream.getRoot();
client.startPut(1, root);
client.putNext();
client.putNext();
client.completed();
}
在使用BitVector类型时,需要注意:
new ArrowType.Bit(位数)
创建BitVector类型。client.startPut(1, root)
开始事务,使用client.putNext()
插入数据,使用client.completed()
提交事务。
上一篇:ApacheArrowC++中的“table->column(0)->GetScalar(0).ValueOrDie()->Value<int32_t>()doesn'tcompile”语句无法编译。