org.apache.beam.sdk.extensions.sql.zetasql.translation

Class ExpressionConverter

java.lang.Object

org.apache.beam.sdk.extensions.sql.zetasql.translation.ExpressionConverter

@Internal
public class ExpressionConverter
extends java.lang.Object

Extracts expressions (function calls, field accesses) from the resolve query nodes, converts them to RexNodes.

Constructor Summary

Constructors

Constructor and Description
ExpressionConverter(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptCluster cluster, QueryPlanner.QueryParameters params, java.util.Map<java.lang.String,com.google.zetasql.resolvedast.ResolvedNodes.ResolvedCreateFunctionStmt> userDefinedFunctions)

Method Summary

Modifier and Type	Method and Description
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode	convertRelNodeToRexRangeRef(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode rel)
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode	convertResolvedLiteral(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedLiteral resolvedLiteral) Convert a resolved literal to a RexNode.
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode	convertRexNodeFromResolvedExpr(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedExpr expr) Create a RexNode for a corresponding resolved expression.
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode	convertRexNodeFromResolvedExpr(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedExpr expr, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> columnList, java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> fieldList, java.util.Map<java.lang.String,org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode> functionArguments) Create a RexNode for a corresponding resolved expression node.
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode	convertRexNodeFromResolvedExprWithRefScan(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedExpr expr, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> refScanLeftColumnList, java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> leftFieldList, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> originalLeftColumnList, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> refScanRightColumnList, java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> rightFieldList, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> originalRightColumnList) Extract the RexNode from expression with ref scan.
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexCall	convertTableValuedFunction(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode input, com.google.zetasql.TableValuedFunction tvf, java.util.List<com.google.zetasql.resolvedast.ResolvedNodes.ResolvedTVFArgument> argumentList, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> inputTableColumns) Convert a TableValuedFunction in ZetaSQL to a RexCall in Calcite.
int	indexOfProjectionColumnRef(long colId, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> columnList) Return an index of the projection column reference.
java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode>	retrieveRexNode(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedProjectScan node, java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> fieldList) Extract expressions from a project scan node.
java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode>	retrieveRexNodeFromOrderByScan(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptCluster cluster, com.google.zetasql.resolvedast.ResolvedNodes.ResolvedOrderByScan node, java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> fieldList) Extract expressions from order by scan node.
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode	trueLiteral()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ExpressionConverter

public ExpressionConverter(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptCluster cluster,
                           QueryPlanner.QueryParameters params,
                           java.util.Map<java.lang.String,com.google.zetasql.resolvedast.ResolvedNodes.ResolvedCreateFunctionStmt> userDefinedFunctions)

Method Detail

retrieveRexNode

public java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode> retrieveRexNode(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedProjectScan node,
                                                                                                             java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> fieldList)

Extract expressions from a project scan node.

retrieveRexNodeFromOrderByScan

public java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode> retrieveRexNodeFromOrderByScan(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptCluster cluster,
                                                                                                                            com.google.zetasql.resolvedast.ResolvedNodes.ResolvedOrderByScan node,
                                                                                                                            java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> fieldList)

Extract expressions from order by scan node.

convertRexNodeFromResolvedExpr

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode convertRexNodeFromResolvedExpr(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedExpr expr,
                                                                                                            java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> columnList,
                                                                                                            java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> fieldList,
                                                                                                            java.util.Map<java.lang.String,org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode> functionArguments)

Create a RexNode for a corresponding resolved expression node.

convertRelNodeToRexRangeRef

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode convertRelNodeToRexRangeRef(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode rel)

convertRexNodeFromResolvedExpr

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode convertRexNodeFromResolvedExpr(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedExpr expr)

Create a RexNode for a corresponding resolved expression.

convertRexNodeFromResolvedExprWithRefScan

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode convertRexNodeFromResolvedExprWithRefScan(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedExpr expr,
                                                                                                                       java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> refScanLeftColumnList,
                                                                                                                       java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> leftFieldList,
                                                                                                                       java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> originalLeftColumnList,
                                                                                                                       java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> refScanRightColumnList,
                                                                                                                       java.util.List<org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.type.RelDataTypeField> rightFieldList,
                                                                                                                       java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> originalRightColumnList)

Extract the RexNode from expression with ref scan.

trueLiteral

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode trueLiteral()

convertResolvedLiteral

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexNode convertResolvedLiteral(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedLiteral resolvedLiteral)

Convert a resolved literal to a RexNode.

convertTableValuedFunction

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rex.RexCall convertTableValuedFunction(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode input,
                                                                                                        com.google.zetasql.TableValuedFunction tvf,
                                                                                                        java.util.List<com.google.zetasql.resolvedast.ResolvedNodes.ResolvedTVFArgument> argumentList,
                                                                                                        java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> inputTableColumns)

Convert a TableValuedFunction in ZetaSQL to a RexCall in Calcite.

indexOfProjectionColumnRef

public int indexOfProjectionColumnRef(long colId,
                                      java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> columnList)

Return an index of the projection column reference.