Presto中的RowType支持

Question

我实现了BigQuery插件。BigQuery支持struct类型，它可以在Presto中表示为rowtype类。

RowType在RowType::CreateBlockBuilder中创建RowBlockBuilder，它具有RowBlockBuilder::AppendStructure方法，该方法只要求接受AbstractSingleRowBlock类的实例。

这意味着在我实现Presto的RecordCursorBigQueryRecordCursor::GetObject方法时，我必须为具有RowType类型的字段返回AbstractSingleRowBlock。

但是AbstractSingleRowBlock有package private abstract方法，这使我无法实现该类。唯一的子级singlerowblock具有package private constructor，并且没有可以为我构建实例的工厂或构建器。

如何在BigQueryRecordCursor::GetObject中实现Struct支持？（提醒:BigQueryRecordCursor是RecordCursor的子级）。

Answer 1

您需要通过调用beginblockentry来组装该行的块，通过type.writexxx将每列的值附加到列的类型，然后使用closeentry。这里有一些伪代码。

BlockBuilder builder = type.createBlockBuilder(..);

builder = builder.beginBlockEntry();
for each column {
    ...

    columnType.writeXXX(builder, ...);
}
builder.closeEntry();

return (Block) type.getObject(builder, 0);

但是，我建议您改用柱状API（即connectorpageSource和friends）。看看Elasticsearch连接器是如何实现的：

https://github.com/prestoSQL/presto/blob/master/presto-elasticsearch/src/main/java/io/prestoSQL/elasticsearch/elasticsearch/src/main/java/io/prestoSQL/elasticsearch/elasticsearchsourceprovider.java https://github.com/presto/blob/master/presto-elasticsearch/src/main/java/prestoSQL/elasticsearch/elasticsearchsource.java

以下是它处理行类型的方式：

https://github.com/prestoSQL/presto/blob/master/presto-elasticsearch/src/main/java/io/prestoSQL/elasticsearch/decoders/rowdecoder.java

此外，我建议您加入Presto社区Slack上的#dev频道，所有Presto开发人员都在这里活动。