Package org.apache.druid.sql.calcite.expression

Class OperatorConversions

java.lang.Object

org.apache.druid.sql.calcite.expression.OperatorConversions

public class OperatorConversions
extends Object

Utilities for assisting in writing SqlOperatorConversion implementations.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	OperatorConversions.AggregatorBuilder
static class	OperatorConversions.OperatorBuilder<T extends org.apache.calcite.sql.SqlFunction>	Helps in creating the operator builder along with validations and type inference for simple operator conversions.

Constructor Summary

Constructors

Constructor	Description
OperatorConversions()

Method Summary

Modifier and Type	Method	Description
static OperatorConversions.OperatorBuilder<org.apache.calcite.sql.SqlAggFunction>	aggregatorBuilder(String name)	Returns a builder that helps SqlAggregator implementations create the SqlAggFunction objects they need to return from SqlAggregator.calciteFunction().
static DruidExpression	convertCall(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, String functionName)	Deprecated.
static DruidExpression	convertCall(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, String functionName, Function<List<DruidExpression>,SimpleExtraction> simpleExtractionFunction)	Deprecated.
static DruidExpression	convertCall(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, DruidExpression.DruidExpressionCreator expressionFunction)
static DruidExpression	convertCallBuilder(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, DruidExpression.ExpressionGenerator expressionGenerator)
static DruidExpression	convertCallWithPostAggOperands(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, DruidExpression.DruidExpressionCreator expressionFunction, PostAggregatorVisitor postAggregatorVisitor)
static DruidExpression	convertDirectCall(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, String functionName)
static DruidExpression	convertDirectCallWithExtraction(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, String functionName, Function<List<DruidExpression>,SimpleExtraction> simpleExtractionFunction)
static DirectOperatorConversion	druidBinaryLongFn(String sqlOperator, String druidFunctionName)
static DirectOperatorConversion	druidUnaryDoubleFn(String sqlOperator, String druidFunctionName)
static DirectOperatorConversion	druidUnaryLongFn(String sqlOperator, String druidFunctionName)
static <T> T	getOperandWithDefault(List<org.apache.calcite.rex.RexNode> operands, int i, Function<org.apache.calcite.rex.RexNode,T> f, T defaultReturnValue)	Gets operand "i" from "operands", or returns a default value if it doesn't exist (operands is too short) or is null.
static OperatorConversions.OperatorBuilder<org.apache.calcite.sql.SqlFunction>	operatorBuilder(String name)	Returns a builder that helps SqlOperatorConversion implementations create the SqlFunction objects they need to return from SqlOperatorConversion.calciteOperator().
static boolean	throwOrReturn(boolean throwOnFailure, org.apache.calcite.sql.SqlCallBinding callBinding, Function<org.apache.calcite.sql.SqlCallBinding,org.apache.calcite.runtime.CalciteException> exceptionMapper)
static PostAggregator	toPostAggregator(PlannerContext plannerContext, RowSignature rowSignature, org.apache.calcite.rex.RexNode rexNode, PostAggregatorVisitor postAggregatorVisitor, boolean useExpressions)	Translate a Calcite RexNode to a Druid PostAggregator

Methods inherited from class java.lang.Object

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

Constructor Details

OperatorConversions

public OperatorConversions()

Method Details

convertDirectCall

@Nullable
public static DruidExpression convertDirectCall(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 String functionName)

convertDirectCallWithExtraction

@Nullable
public static DruidExpression convertDirectCallWithExtraction(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 String functionName,
 Function<List<DruidExpression>,SimpleExtraction> simpleExtractionFunction)

convertCallBuilder

@Nullable
public static DruidExpression convertCallBuilder(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 DruidExpression.ExpressionGenerator expressionGenerator)

convertCall

@Nullable
public static DruidExpression convertCall(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 DruidExpression.DruidExpressionCreator expressionFunction)

convertCall

@Deprecated
@Nullable
public static DruidExpression convertCall(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 String functionName)

Deprecated.

convertCall

@Deprecated
@Nullable
public static DruidExpression convertCall(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 String functionName,
 Function<List<DruidExpression>,SimpleExtraction> simpleExtractionFunction)

Deprecated.

getOperandWithDefault

public static <T> T getOperandWithDefault(List<org.apache.calcite.rex.RexNode> operands,
 int i,
 Function<org.apache.calcite.rex.RexNode,T> f,
 T defaultReturnValue)

Gets operand "i" from "operands", or returns a default value if it doesn't exist (operands is too short) or is null.

convertCallWithPostAggOperands

@Nullable
public static DruidExpression convertCallWithPostAggOperands(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 DruidExpression.DruidExpressionCreator expressionFunction,
 PostAggregatorVisitor postAggregatorVisitor)

toPostAggregator

@Nullable
public static PostAggregator toPostAggregator(PlannerContext plannerContext,
 RowSignature rowSignature,
 org.apache.calcite.rex.RexNode rexNode,
 PostAggregatorVisitor postAggregatorVisitor,
 boolean useExpressions)

Translate a Calcite RexNode to a Druid PostAggregator

Parameters:

plannerContext - SQL planner context

rowSignature - signature of the rows to be extracted from

rexNode - expression meant to be applied on top of the rows

postAggregatorVisitor - visitor that manages postagg names and tracks postaggs that were created by the translation

useExpressions - whether we should consider ExpressionPostAggregator as a target

Returns:

rexNode referring to fields in rowOrder, or null if not possible

operatorBuilder

public static OperatorConversions.OperatorBuilder<org.apache.calcite.sql.SqlFunction> operatorBuilder(String name)

Returns a builder that helps SqlOperatorConversion implementations create the SqlFunction objects they need to return from SqlOperatorConversion.calciteOperator().

aggregatorBuilder

public static OperatorConversions.OperatorBuilder<org.apache.calcite.sql.SqlAggFunction> aggregatorBuilder(String name)

Returns a builder that helps SqlAggregator implementations create the SqlAggFunction objects they need to return from SqlAggregator.calciteFunction().

throwOrReturn

public static boolean throwOrReturn(boolean throwOnFailure,
 org.apache.calcite.sql.SqlCallBinding callBinding,
 Function<org.apache.calcite.sql.SqlCallBinding,org.apache.calcite.runtime.CalciteException> exceptionMapper)

druidUnaryLongFn

public static DirectOperatorConversion druidUnaryLongFn(String sqlOperator,
 String druidFunctionName)

druidBinaryLongFn

public static DirectOperatorConversion druidBinaryLongFn(String sqlOperator,
 String druidFunctionName)

druidUnaryDoubleFn

public static DirectOperatorConversion druidUnaryDoubleFn(String sqlOperator,
 String druidFunctionName)