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.calcite.plan.RelOptCluster cluster, java.util.Map<java.lang.String,com.google.zetasql.Value> params)

Method Summary

Modifier and Type	Method and Description
org.apache.calcite.rex.RexNode	convertResolvedLiteral(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedLiteral resolvedLiteral) Convert a resolved literal to a RexNode.
org.apache.calcite.rex.RexNode	convertRexNodeFromResolvedExpr(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedExpr expr) Create a RexNode for a corresponding resolved expression.
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.calcite.rel.type.RelDataTypeField> fieldList) Create a RexNode for a corresponding resolved expression node.
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.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.calcite.rel.type.RelDataTypeField> rightFieldList, java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> originalRightColumnList) Extract the RexNode from expression with ref scan.
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.calcite.rex.RexNode>	retrieveRexNode(com.google.zetasql.resolvedast.ResolvedNodes.ResolvedProjectScan node, java.util.List<org.apache.calcite.rel.type.RelDataTypeField> fieldList) Extract expressions from a project scan node.
java.util.List<org.apache.calcite.rex.RexNode>	retrieveRexNodeFromOrderByScan(org.apache.calcite.plan.RelOptCluster cluster, com.google.zetasql.resolvedast.ResolvedNodes.ResolvedOrderByScan node, java.util.List<org.apache.calcite.rel.type.RelDataTypeField> fieldList) Extract expressions from order by scan node.

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.calcite.plan.RelOptCluster cluster,
                           java.util.Map<java.lang.String,com.google.zetasql.Value> params)

Method Detail

retrieveRexNode

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

Extract expressions from a project scan node.

retrieveRexNodeFromOrderByScan

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

Extract expressions from order by scan node.

convertRexNodeFromResolvedExpr

public 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.calcite.rel.type.RelDataTypeField> fieldList)

Create a RexNode for a corresponding resolved expression node.

convertRexNodeFromResolvedExpr

public 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.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.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.calcite.rel.type.RelDataTypeField> rightFieldList,
                                                                                java.util.List<com.google.zetasql.resolvedast.ResolvedColumn> originalRightColumnList)

Extract the RexNode from expression with ref scan.

convertResolvedLiteral

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

Convert a resolved literal to a RexNode.

indexOfProjectionColumnRef

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

Return an index of the projection column reference.