org.apache.calcite.sql.type.OperandTypes.family java code examples

Add the Codota plugin to your IDE and get smart completions

private void myMethod () {
 SimpleDateFormat s =String pattern;new SimpleDateFormat(pattern)
String template;Locale locale;new SimpleDateFormat(template, locale)
new SimpleDateFormat()
Smart code suggestions by Tabnine
}

 public SqlFunction build()
 {
  return new SqlFunction(
    name,
    kind,
    Preconditions.checkNotNull(returnTypeInference, "returnTypeInference"),
    null,
    OperandTypes.family(
      Preconditions.checkNotNull(operandTypes, "operandTypes"),
      i -> i + 1 > requiredOperands
    ),
    functionCategory
  );
 }
}

 QuantileSqlAggFunction()
 {
  super(
    NAME,
    null,
    SqlKind.OTHER_FUNCTION,
    ReturnTypes.explicit(SqlTypeName.DOUBLE),
    null,
    OperandTypes.or(
      OperandTypes.and(
        OperandTypes.sequence(SIGNATURE1, OperandTypes.ANY, OperandTypes.LITERAL),
        OperandTypes.family(SqlTypeFamily.ANY, SqlTypeFamily.NUMERIC)
      ),
      OperandTypes.and(
        OperandTypes.sequence(SIGNATURE2, OperandTypes.ANY, OperandTypes.LITERAL, OperandTypes.LITERAL),
        OperandTypes.family(SqlTypeFamily.ANY, SqlTypeFamily.NUMERIC, SqlTypeFamily.EXACT_NUMERIC)
      )
    ),
    SqlFunctionCategory.NUMERIC,
    false,
    false
  );
 }
}

private static CalciteUDFInfo getUDFInfo(String hiveUdfName,
  ImmutableList<RelDataType> calciteArgTypes, RelDataType calciteRetType) {
 CalciteUDFInfo udfInfo = new CalciteUDFInfo();
 udfInfo.udfName = hiveUdfName;
 udfInfo.returnTypeInference = ReturnTypes.explicit(calciteRetType);
 udfInfo.operandTypeInference = InferTypes.explicit(calciteArgTypes);
 ImmutableList.Builder<SqlTypeFamily> typeFamilyBuilder = new ImmutableList.Builder<SqlTypeFamily>();
 for (RelDataType at : calciteArgTypes) {
  typeFamilyBuilder.add(Util.first(at.getSqlTypeName().getFamily(), SqlTypeFamily.ANY));
 }
 udfInfo.operandTypeChecker = OperandTypes.family(typeFamilyBuilder.build());
 return udfInfo;
}

SqlAggFunction createCustomAggFunction(String funcName, RelDataType returnType, Class<?> customAggFuncClz) {
  RelDataTypeFactory typeFactory = getCluster().getTypeFactory();
  SqlIdentifier sqlIdentifier = new SqlIdentifier(funcName, new SqlParserPos(1, 1));
  AggregateFunction aggFunction = AggregateFunctionImpl.create(customAggFuncClz);
  List<RelDataType> argTypes = new ArrayList<RelDataType>();
  List<SqlTypeFamily> typeFamilies = new ArrayList<SqlTypeFamily>();
  for (FunctionParameter o : aggFunction.getParameters()) {
    final RelDataType type = o.getType(typeFactory);
    argTypes.add(type);
    typeFamilies.add(Util.first(type.getSqlTypeName().getFamily(), SqlTypeFamily.ANY));
  }
  return new SqlUserDefinedAggFunction(sqlIdentifier, ReturnTypes.explicit(returnType),
      InferTypes.explicit(argTypes), OperandTypes.family(typeFamilies), aggFunction, false, false,
      typeFactory);
}

private static CalciteUDFInfo getUDFInfo(String hiveUdfName,
  ImmutableList<RelDataType> calciteArgTypes, RelDataType calciteRetType) {
 CalciteUDFInfo udfInfo = new CalciteUDFInfo();
 udfInfo.udfName = hiveUdfName;
 udfInfo.returnTypeInference = ReturnTypes.explicit(calciteRetType);
 udfInfo.operandTypeInference = InferTypes.explicit(calciteArgTypes);
 ImmutableList.Builder<SqlTypeFamily> typeFamilyBuilder = new ImmutableList.Builder<SqlTypeFamily>();
 for (RelDataType at : calciteArgTypes) {
  typeFamilyBuilder.add(Util.first(at.getSqlTypeName().getFamily(), SqlTypeFamily.ANY));
 }
 udfInfo.operandTypeChecker = OperandTypes.family(typeFamilyBuilder.build());
 return udfInfo;
}

 SqlTimestampDiffFunction() {
  super("TIMESTAMPDIFF", SqlKind.TIMESTAMP_DIFF,
    RETURN_TYPE_INFERENCE, null,
    OperandTypes.family(SqlTypeFamily.ANY, SqlTypeFamily.DATETIME,
      SqlTypeFamily.DATETIME),
    SqlFunctionCategory.TIMEDATE);
 }
}

 /** Creates a SqlTimestampAddFunction. */
 SqlTimestampAddFunction() {
  super("TIMESTAMPADD", SqlKind.TIMESTAMP_ADD, RETURN_TYPE_INFERENCE, null,
    OperandTypes.family(SqlTypeFamily.ANY, SqlTypeFamily.INTEGER,
      SqlTypeFamily.DATETIME),
    SqlFunctionCategory.TIMEDATE);
 }
}

 /** Creates a SqlTimestampAddFunction. */
 SqlTimestampAddFunction() {
  super("TIMESTAMPADD", SqlKind.TIMESTAMP_ADD, RETURN_TYPE_INFERENCE, null,
    OperandTypes.family(SqlTypeFamily.ANY, SqlTypeFamily.INTEGER,
      SqlTypeFamily.DATETIME),
    SqlFunctionCategory.TIMEDATE);
 }
}

 SqlTimestampDiffFunction() {
  super("TIMESTAMPDIFF", SqlKind.TIMESTAMP_DIFF,
    RETURN_TYPE_INFERENCE, null,
    OperandTypes.family(SqlTypeFamily.ANY, SqlTypeFamily.DATETIME,
      SqlTypeFamily.DATETIME),
    SqlFunctionCategory.TIMEDATE);
 }
}

/**
 * Creates a checker that passes if each operand is a member of a
 * corresponding family.
 */
public static FamilyOperandTypeChecker family(List<SqlTypeFamily> families) {
 return family(families, i -> false);
}

/**
 * Creates a checker that passes if each operand is a member of a
 * corresponding family.
 */
public static FamilyOperandTypeChecker family(List<SqlTypeFamily> families) {
 return family(families, i -> false);
}

public SqlJsonArrayAggAggFunction(String name,
  SqlJsonConstructorNullClause nullClause) {
 super(name, null, SqlKind.JSON_ARRAYAGG, ReturnTypes.VARCHAR_2000, null,
   OperandTypes.family(SqlTypeFamily.ANY), SqlFunctionCategory.SYSTEM,
   false, false, Optionality.FORBIDDEN);
 this.nullClause = Objects.requireNonNull(nullClause);
}

/** Creates a SqlJsonObjectAggAggFunction. */
public SqlJsonObjectAggAggFunction(String name,
  SqlJsonConstructorNullClause nullClause) {
 super(name, null, SqlKind.JSON_OBJECTAGG, ReturnTypes.VARCHAR_2000, null,
   OperandTypes.family(SqlTypeFamily.CHARACTER, SqlTypeFamily.ANY),
   SqlFunctionCategory.SYSTEM, false, false, Optionality.FORBIDDEN);
 this.nullClause = Objects.requireNonNull(nullClause);
}

public SqlJsonQueryFunction() {
 super("JSON_QUERY", SqlKind.OTHER_FUNCTION,
   ReturnTypes.cascade(ReturnTypes.VARCHAR_2000,
     SqlTypeTransforms.FORCE_NULLABLE),
   null,
   OperandTypes.family(SqlTypeFamily.ANY,
     SqlTypeFamily.ANY, SqlTypeFamily.ANY, SqlTypeFamily.ANY),
   SqlFunctionCategory.SYSTEM);
}

public SqlJsonApiCommonSyntaxOperator() {
 super("JSON_API_COMMON_SYNTAX", SqlKind.JSON_API_COMMON_SYNTAX, 100, true,
   ReturnTypes.explicit(SqlTypeName.ANY), null,
   OperandTypes.family(SqlTypeFamily.ANY, SqlTypeFamily.STRING));
}

private SqlSingleOperandTypeChecker getChecker(RelDataType operandType) {
 switch (operandType.getSqlTypeName()) {
 case ARRAY:
  return OperandTypes.family(SqlTypeFamily.INTEGER);
 case MAP:
  return OperandTypes.family(
    operandType.getKeyType().getSqlTypeName().getFamily());
 case ANY:
 case DYNAMIC_STAR:
  return OperandTypes.or(
    OperandTypes.family(SqlTypeFamily.INTEGER),
    OperandTypes.family(SqlTypeFamily.CHARACTER));
 default:
  throw new AssertionError(operandType.getSqlTypeName());
 }
}

private SqlSingleOperandTypeChecker getChecker(RelDataType operandType) {
 switch (operandType.getSqlTypeName()) {
 case ROW:
  return OperandTypes.family(SqlTypeFamily.STRING);
 default:
  throw new AssertionError(operandType.getSqlTypeName());
 }
}

private SqlSingleOperandTypeChecker getChecker(RelDataType operandType) {
 switch (operandType.getSqlTypeName()) {
 case ROW:
  return OperandTypes.family(SqlTypeFamily.STRING);
 default:
  throw new AssertionError(operandType.getSqlTypeName());
 }
}

SqlAggFunction createCustomAggFunction(String funcName, RelDataType returnType, Class<?> customAggFuncClz) {
  RelDataTypeFactory typeFactory = getCluster().getTypeFactory();
  SqlIdentifier sqlIdentifier = new SqlIdentifier(funcName, new SqlParserPos(1, 1));
  AggregateFunction aggFunction = AggregateFunctionImpl.create(customAggFuncClz);
  List<RelDataType> argTypes = new ArrayList<RelDataType>();
  List<SqlTypeFamily> typeFamilies = new ArrayList<SqlTypeFamily>();
  for (FunctionParameter o : aggFunction.getParameters()) {
    final RelDataType type = o.getType(typeFactory);
    argTypes.add(type);
    typeFamilies.add(Util.first(type.getSqlTypeName().getFamily(), SqlTypeFamily.ANY));
  }
  return new SqlUserDefinedAggFunction(sqlIdentifier, ReturnTypes.explicit(returnType),
      InferTypes.explicit(argTypes), OperandTypes.family(typeFamilies), aggFunction, false, false,
      typeFactory);
}

private static CalciteUDFInfo getUDFInfo(String hiveUdfName,
  ImmutableList<RelDataType> calciteArgTypes, RelDataType calciteRetType) {
 CalciteUDFInfo udfInfo = new CalciteUDFInfo();
 udfInfo.udfName = hiveUdfName;
 udfInfo.returnTypeInference = ReturnTypes.explicit(calciteRetType);
 udfInfo.operandTypeInference = InferTypes.explicit(calciteArgTypes);
 ImmutableList.Builder<SqlTypeFamily> typeFamilyBuilder = new ImmutableList.Builder<SqlTypeFamily>();
 for (RelDataType at : calciteArgTypes) {
  typeFamilyBuilder.add(Util.first(at.getSqlTypeName().getFamily(), SqlTypeFamily.ANY));
 }
 udfInfo.operandTypeChecker = OperandTypes.family(typeFamilyBuilder.build());
 udfInfo.argTypes = ImmutableList.<RelDataType> copyOf(calciteArgTypes);
 udfInfo.retType = calciteRetType;
 return udfInfo;
}

Javadoc

Creates a checker that passes if each operand is a member of a corresponding family.

Popular methods of OperandTypes

sequence
Creates an operand checker from a sequence of single-operand checkers.
or
Creates a single-operand checker that passes if any one of the rules passes.
and
Creates a single-operand checker that passes if all of the rules pass.
repeat
Creates a checker that passes if all of the rules pass for each operand, using a given operand count

Popular in Java

Running tasks concurrently on multiple threads
compareTo (BigDecimal)
getExternalFilesDir (Context)
startActivity (Activity)
Pointer (com.sun.jna)
An abstraction for a native pointer data type. A Pointer instance represents, on the Java side, a na
PrintWriter (java.io)
Wraps either an existing OutputStream or an existing Writerand provides convenience methods for prin
InetAddress (java.net)
An Internet Protocol (IP) address. This can be either an IPv4 address or an IPv6 address, and in pra
Path (java.nio.file)
HashMap (java.util)
HashMap is an implementation of Map. All optional operations are supported.All elements are permitte
Annotation (javassist.bytecode.annotation)
The annotation structure.An instance of this class is returned bygetAnnotations() in AnnotationsAttr

How to use familymethodin org.apache.calcite.sql.type.OperandTypes

Best Java code snippets using org.apache.calcite.sql.type.OperandTypes.family (Showing top 20 results out of 315)

How to use
family
method
in
org.apache.calcite.sql.type.OperandTypes