根据官方文档指引,在PipelineOptions中添加ZetaSQL分析函数相关的选项,启用分析函数支持。如下示例代码:
from apache_beam.options.pipeline_options import PipelineOptions
options = PipelineOptions()
options.view_as(BeamSqlPipelineOptions).set_table_planner("org.apache.beam.sdk.extensions.sql.zetasql.ZetaSQLTablePlanner")
options.view_as(BeamSqlPipelineOptions).set_fn_execution("org.apache.beam.sdk.extensions.sql.zetasql.ZetaSqlCalciteFnExecutor")
其中set_table_planner
和set_fn_execution
是设置ZetaSQL分析函数的选项,值的设置根据具体情况进行调整。