How to use

InferTypes

in

org.apache.calcite.sql.type

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;
}

oldCall.isDistinct(),
ReturnTypes.explicit(sumSquaredReturnType),
InferTypes.explicit(Collections.singletonList(argSquared.getType())),
oldCall.isDistinct(),
ReturnTypes.explicit(sumReturnType),
InferTypes.explicit(Collections.singletonList(argOrdinalType)),

if (function instanceof ScalarFunction) {
 return new SqlUserDefinedFunction(name, infer((ScalarFunction) function),
   InferTypes.explicit(argTypes), typeChecker, paramTypes, function);
} else if (function instanceof AggregateFunction) {
 return new SqlUserDefinedAggFunction(name,
   infer((AggregateFunction) function), InferTypes.explicit(argTypes),
   typeChecker, (AggregateFunction) function, false, false, typeFactory);
} else if (function instanceof TableMacro) {
 return new SqlUserDefinedTableMacro(name, ReturnTypes.CURSOR,
   InferTypes.explicit(argTypes), typeChecker, paramTypes,
   (TableMacro) function);
} else if (function instanceof TableFunction) {
 return new SqlUserDefinedTableFunction(name, ReturnTypes.CURSOR,
   InferTypes.explicit(argTypes), typeChecker, paramTypes,
   (TableFunction) function);
} else {

if (function instanceof ScalarFunction) {
 return new SqlUserDefinedFunction(name, infer((ScalarFunction) function),
   InferTypes.explicit(argTypes), typeChecker, paramTypes, function);
} else if (function instanceof AggregateFunction) {
 return new SqlUserDefinedAggFunction(name,
   infer((AggregateFunction) function), InferTypes.explicit(argTypes),
   typeChecker, (AggregateFunction) function, false, false,
   Optionality.FORBIDDEN, typeFactory);
} else if (function instanceof TableMacro) {
 return new SqlUserDefinedTableMacro(name, ReturnTypes.CURSOR,
   InferTypes.explicit(argTypes), typeChecker, paramTypes,
   (TableMacro) function);
} else if (function instanceof TableFunction) {
 return new SqlUserDefinedTableFunction(name, ReturnTypes.CURSOR,
   InferTypes.explicit(argTypes), typeChecker, paramTypes,
   (TableFunction) function);
} else {

if (function instanceof ScalarFunction) {
 return new SqlUserDefinedFunction(name, infer((ScalarFunction) function),
   InferTypes.explicit(argTypes), typeChecker, paramTypes, function);
} else if (function instanceof AggregateFunction) {
 return new SqlUserDefinedAggFunction(name,
   infer((AggregateFunction) function), InferTypes.explicit(argTypes),
   typeChecker, (AggregateFunction) function, false, false, typeFactory);
} else if (function instanceof TableMacro) {
 return new SqlUserDefinedTableMacro(name, ReturnTypes.CURSOR,
   InferTypes.explicit(argTypes), typeChecker, paramTypes,
   (TableMacro) function);
} else if (function instanceof TableFunction) {
 return new SqlUserDefinedTableFunction(name, ReturnTypes.CURSOR,
   InferTypes.explicit(argTypes), typeChecker, paramTypes,
   (TableFunction) function);
} else {

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;
}