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

Class BeamZetaSqlUncollectRel

java.lang.Object

org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.AbstractRelNode

org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.SingleRel

org.apache.beam.sdk.extensions.sql.zetasql.unnest.ZetaSqlUnnest

org.apache.beam.sdk.extensions.sql.zetasql.unnest.BeamZetaSqlUncollectRel

All Implemented Interfaces:

java.lang.Cloneable, BeamRelNode, org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptNode, org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode

public class BeamZetaSqlUncollectRel
extends ZetaSqlUnnest
implements BeamRelNode

BeamRelNode to implement an uncorrelated ZetaSqlUnnest, aka UNNEST.

This class is a copy of BeamUncollectRel except that in UncollectDoFn it does not treat elements of struct type differently.

Details of why unwrapping structs breaks ZetaSQL UNNEST syntax is in https://issues.apache.org/jira/browse/BEAM-10896.

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode

org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode.Context

Field Summary

Fields inherited from class org.apache.beam.sdk.extensions.sql.zetasql.unnest.ZetaSqlUnnest

withOrdinality

Fields inherited from class org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.SingleRel

input

Fields inherited from class org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.AbstractRelNode

digest, id, rowType, traitSet

Constructor Summary

Constructors

Constructor and Description
BeamZetaSqlUncollectRel(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptCluster cluster, org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelTraitSet traitSet, org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode input, boolean withOrdinality)

Method Summary

Modifier and Type	Method and Description
BeamCostModel	beamComputeSelfCost(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptPlanner planner, org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.metadata.RelMetadataQuery mq) This method is called by org.apache.beam.sdk.extensions.sql.impl.CalciteQueryPlanner.NonCumulativeCostImpl.
PTransform<PCollectionList<Row>,PCollection<Row>>	buildPTransform()
org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode	copy(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelTraitSet traitSet, org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode input)
NodeStats	estimateNodeStats(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.metadata.RelMetadataQuery mq) This method is called by org.apache.beam.sdk.extensions.sql.impl.planner.RelMdNodeStats.

Methods inherited from class org.apache.beam.sdk.extensions.sql.zetasql.unnest.ZetaSqlUnnest

copy, create, deriveRowType, deriveUncollectRowType, explainTerms

Methods inherited from class org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.SingleRel

childrenAccept, estimateRowCount, getInput, getInputs, replaceInput

Methods inherited from class org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.AbstractRelNode

accept, accept, collectVariablesSet, collectVariablesUsed, computeDigest, computeSelfCost, computeSelfCost, explain, getChildExps, getCluster, getCollationList, getConvention, getCorrelVariable, getDescription, getDigest, getExpectedInputRowType, getId, getInput, getQuery, getRelTypeName, getRows, getRowType, getTable, getTraitSet, getVariablesSet, getVariablesStopped, isDistinct, isKey, isValid, isValid, metadata, onRegister, recomputeDigest, register, sole, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.beam.sdk.extensions.sql.impl.rel.BeamRelNode

getPCollectionInputs, getPipelineOptions, isBounded

Methods inherited from interface org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode

accept, accept, childrenAccept, collectVariablesSet, collectVariablesUsed, computeSelfCost, computeSelfCost, copy, estimateRowCount, explain, getChildExps, getCollationList, getConvention, getCorrelVariable, getExpectedInputRowType, getInput, getInputs, getQuery, getRelTypeName, getRows, getRowType, getTable, getVariablesSet, getVariablesStopped, isDistinct, isKey, isValid, isValid, metadata, onRegister, recomputeDigest, register, replaceInput

Methods inherited from interface org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptNode

getCluster, getDescription, getDigest, getId, getTraitSet

Constructor Detail

BeamZetaSqlUncollectRel

public BeamZetaSqlUncollectRel(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptCluster cluster,
                               org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelTraitSet traitSet,
                               org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode input,
                               boolean withOrdinality)

Method Detail

copy

public org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode copy(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelTraitSet traitSet,
                                                                                  org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode input)

Overrides:

copy in class ZetaSqlUnnest

buildPTransform

public PTransform<PCollectionList<Row>,PCollection<Row>> buildPTransform()

Specified by:

buildPTransform in interface BeamRelNode

estimateNodeStats

public NodeStats estimateNodeStats(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.metadata.RelMetadataQuery mq)

Description copied from interface: BeamRelNode

This method is called by org.apache.beam.sdk.extensions.sql.impl.planner.RelMdNodeStats. This is currently only used in SQLTransform Path (and not JDBC path). When a RelNode wants to calculate its BeamCost or estimate its NodeStats, it may need NodeStat of its inputs. However, it should not call this directly (because maybe its inputs are not physical yet). It should call BeamSqlRelUtils.getNodeStats( org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.RelNode, org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.metadata.RelMetadataQuery) instead.

Specified by:

estimateNodeStats in interface BeamRelNode

beamComputeSelfCost

public BeamCostModel beamComputeSelfCost(org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.plan.RelOptPlanner planner,
                                         org.apache.beam.vendor.calcite.v1_20_0.org.apache.calcite.rel.metadata.RelMetadataQuery mq)

Description copied from interface: BeamRelNode

This method is called by org.apache.beam.sdk.extensions.sql.impl.CalciteQueryPlanner.NonCumulativeCostImpl. This is currently only used in SQLTransform Path (and not JDBC path). This is needed when Calcite Query Planner wants to get the cost of a plan. Instead of calling this directly for a node, if we needed that it should be obtained by calling mq.getNonCumulativeCost. This way RelMetadataQuery will call this method instead of ComputeSelfCost if the handler is set correctly (see org.apache.beam.sdk.extensions.sql.impl.CalciteQueryPlanner#convertToBeamRel(String))

Specified by:

beamComputeSelfCost in interface BeamRelNode