Java Code Examples for org.apache.calcite.sql.SqlAggFunction#getName()
The following examples show how to use
org.apache.calcite.sql.SqlAggFunction#getName() .
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
| Source File: SolrAggregate.java From lucene-solr with Apache License 2.0 | 6 votes |
|
@SuppressWarnings({"fallthrough"})
private Pair<String, String> toSolrMetric(Implementor implementor, AggregateCall aggCall, List<String> inNames) {
SqlAggFunction aggregation = aggCall.getAggregation();
List<Integer> args = aggCall.getArgList();
switch (args.size()) {
case 0:
if (aggregation.equals(SqlStdOperatorTable.COUNT)) {
return new Pair<>(aggregation.getName(), "*");
}
case 1:
String inName = inNames.get(args.get(0));
String name = implementor.fieldMappings.getOrDefault(inName, inName);
if(SUPPORTED_AGGREGATIONS.contains(aggregation)) {
return new Pair<>(aggregation.getName(), name);
}
default:
throw new AssertionError("Invalid aggregation " + aggregation + " with args " + args + " with names" + inNames);
}
}
Example 2
| Source File: DrillCalciteSqlAggFunctionWrapper.java From Bats with Apache License 2.0 | 5 votes |
|
private DrillCalciteSqlAggFunctionWrapper(
SqlAggFunction sqlAggFunction,
SqlReturnTypeInference sqlReturnTypeInference) {
super(sqlAggFunction.getName(),
sqlAggFunction.getSqlIdentifier(),
sqlAggFunction.getKind(),
sqlReturnTypeInference,
sqlAggFunction.getOperandTypeInference(),
Checker.ANY_CHECKER,
sqlAggFunction.getFunctionType(),
sqlAggFunction.requiresOrder(),
sqlAggFunction.requiresOver());
this.operator = sqlAggFunction;
}
Example 3
| Source File: RexImpTable.java From calcite with Apache License 2.0 | 5 votes |
|
@Override public void implementNotNullAdd(AggContext info,
AggAddContext add) {
Expression acc = add.accumulator().get(0);
Expression arg = add.arguments().get(0);
SqlAggFunction aggregation = info.aggregation();
final BuiltInMethod builtInMethod;
switch (aggregation.kind) {
case BIT_AND:
builtInMethod = BuiltInMethod.BIT_AND;
break;
case BIT_OR:
builtInMethod = BuiltInMethod.BIT_OR;
break;
case BIT_XOR:
builtInMethod = BuiltInMethod.BIT_XOR;
break;
default:
throw new IllegalArgumentException("Unknown " + aggregation.getName()
+ ". Only support bit_and, bit_or and bit_xor for bit aggregation function");
}
final Method method = builtInMethod.method;
Expression next = Expressions.call(
method.getDeclaringClass(),
method.getName(),
acc,
Expressions.unbox(arg));
accAdvance(add, acc, next);
}
Example 4
| Source File: ElasticsearchAggregate.java From dk-fitting with Apache License 2.0 | 4 votes |
|
public void implement(Implementor implementor) {
implementor.visitChild(0, getInput());
ElasticsearchTable esTable = implementor.getElasticsearchTable();
List<AggregateCall> aggCallList = getAggCallList();
List<RelDataTypeField> fieldList = esTable.getRowType().getFieldList();
final List<String> inNames = ElasticsearchRules.elasticsearchFieldNames(getInput().getRowType());
final List<String> outNames = ElasticsearchRules.elasticsearchFieldNames(getRowType());
esTable.setOutNames(outNames);
// final List<String> inNames = ElasticsearchRules.elasticsearchFieldNames(getInput().getRowType());
TermsAggregationBuilder groupAggregationBuilder = null;
AggregationBuilder groupAggregationBuilder2 = null; //防止出现group但没有sort出错的情况
for(int i = 0 ;i<groupSet.cardinality();i++) {
final String inName = inNames.get(groupSet.nth(i));
groupAggregationBuilder = AggregationBuilders.terms(inName.toLowerCase()).field(transFieldName(ElasticsearchRules.maybeQuote(inName).toLowerCase(),fieldList));
groupAggregationBuilder2 = AggregationBuilders.terms(inName).field(transFieldName(ElasticsearchRules.maybeQuote(inName),fieldList));
esTable.setIsGroup(true);
}
List<String> out = new ArrayList<String>(); //聚合函数的名称
for(AggregateCall call : aggCallList)
{
SqlAggFunction function = call.getAggregation();
List<Integer> argList = call.getArgList();
String functionName = function.getName();
out.add(functionName);
switch (function.getKind())
{
case MIN:
RelDataTypeField typeField = fieldList.get(argList.get(0));
if (groupAggregationBuilder == null) {
//min值 与 原字段值 的类型是一样的
esTable.addAggregationBuilder(AggregationBuilders.min(functionName).field(typeField.getName().toLowerCase()),
((RelDataTypeFactoryImpl.JavaType) typeField.getType()).getJavaClass());
}else {
MinAggregationBuilder minAgg = AggregationBuilders.min(functionName).field(typeField.getName().toLowerCase());
// esTable.addAggregationBuilder(groupAggregationBuilder.subAggregation(minAgg), call.getType().getClass());
groupAggregationBuilder.subAggregation(minAgg);
groupAggregationBuilder2.subAggregation(minAgg);
}
break;
case MAX:
//max值 与 原字段值 的类型是一样的
RelDataTypeField typeField1 = fieldList.get(argList.get(0));
if(groupAggregationBuilder == null) {
esTable.addAggregationBuilder(AggregationBuilders.max(functionName).field(typeField1.getName().toLowerCase()),
((RelDataTypeFactoryImpl.JavaType) typeField1.getType()).getJavaClass());
}else {
groupAggregationBuilder.subAggregation(AggregationBuilders.max(functionName).field(typeField1.getName().toLowerCase()));
}
break;
case COUNT:
if(groupAggregationBuilder == null) {
if (argList == null || argList.size() == 0)//count(*)
esTable.addAggregationBuilder(AggregationBuilders.count(functionName), Long.class);
else
esTable.addAggregationBuilder(AggregationBuilders.count(functionName).field(transFieldName(fieldList.get(argList.get(0)).getName().toLowerCase(), fieldList)), Long.class);
}else {
groupAggregationBuilder.subAggregation(AggregationBuilders.count(functionName).field(transFieldName(fieldList.get(argList.get(0)).getName().toLowerCase(), fieldList)));
groupAggregationBuilder2.subAggregation(AggregationBuilders.count(functionName).field(transFieldName(fieldList.get(argList.get(0)).getName().toLowerCase(), fieldList)));
}
break;
case SUM:
if(groupAggregationBuilder == null) {
esTable.addAggregationBuilder(AggregationBuilders.sum(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase()), Double.class);
}else {
String s = fieldList.get(argList.get(0)).getName().toLowerCase();
SumAggregationBuilder sumAgg = AggregationBuilders.sum(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase());
groupAggregationBuilder.subAggregation(sumAgg);
groupAggregationBuilder2.subAggregation(sumAgg);
}
break;
case AVG:
if(groupAggregationBuilder == null) {
esTable.addAggregationBuilder(AggregationBuilders.avg(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase()), Double.class);
}else {
AvgAggregationBuilder avgAgg = AggregationBuilders.avg(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase());
groupAggregationBuilder.subAggregation(avgAgg);
groupAggregationBuilder2.subAggregation(avgAgg);
}
break;
default:break;
}
}
if (groupAggregationBuilder != null) {
esTable.addAggregationBuilder((AbstractAggregationBuilder) groupAggregationBuilder,String.class);
esTable.addAggregationBuilderList2(groupAggregationBuilder);
}
esTable.setOut(out);
}
Example 5
| Source File: ElasticsearchAggregate.java From dk-fitting with Apache License 2.0 | 4 votes |
|
public void implement(Implementor implementor) {
implementor.visitChild(0, getInput());
ElasticsearchTable esTable = implementor.getElasticsearchTable();
List<AggregateCall> aggCallList = getAggCallList();
List<RelDataTypeField> fieldList = esTable.getRowType().getFieldList();
final List<String> inNames = ElasticsearchRules.elasticsearchFieldNames(getInput().getRowType());
final List<String> outNames = ElasticsearchRules.elasticsearchFieldNames(getRowType());
esTable.setOutNames(outNames);
// final List<String> inNames = ElasticsearchRules.elasticsearchFieldNames(getInput().getRowType());
TermsAggregationBuilder groupAggregationBuilder = null;
AggregationBuilder groupAggregationBuilder2 = null; //防止出现group但没有sort出错的情况
for(int i = 0 ;i<groupSet.cardinality();i++) {
final String inName = inNames.get(groupSet.nth(i));
groupAggregationBuilder = AggregationBuilders.terms(inName.toLowerCase()).field(transFieldName(ElasticsearchRules.maybeQuote(inName).toLowerCase(),fieldList));
groupAggregationBuilder2 = AggregationBuilders.terms(inName).field(transFieldName(ElasticsearchRules.maybeQuote(inName),fieldList));
esTable.setIsGroup(true);
}
List<String> out = new ArrayList<String>(); //聚合函数的名称
for(AggregateCall call : aggCallList)
{
SqlAggFunction function = call.getAggregation();
List<Integer> argList = call.getArgList();
String functionName = function.getName();
out.add(functionName);
switch (function.getKind())
{
case MIN:
RelDataTypeField typeField = fieldList.get(argList.get(0));
if (groupAggregationBuilder == null) {
//min值 与 原字段值 的类型是一样的
esTable.addAggregationBuilder(AggregationBuilders.min(functionName).field(typeField.getName().toLowerCase()),
((RelDataTypeFactoryImpl.JavaType) typeField.getType()).getJavaClass());
}else {
MinAggregationBuilder minAgg = AggregationBuilders.min(functionName).field(typeField.getName().toLowerCase());
// esTable.addAggregationBuilder(groupAggregationBuilder.subAggregation(minAgg), call.getType().getClass());
groupAggregationBuilder.subAggregation(minAgg);
groupAggregationBuilder2.subAggregation(minAgg);
}
break;
case MAX:
//max值 与 原字段值 的类型是一样的
RelDataTypeField typeField1 = fieldList.get(argList.get(0));
if(groupAggregationBuilder == null) {
esTable.addAggregationBuilder(AggregationBuilders.max(functionName).field(typeField1.getName().toLowerCase()),
((RelDataTypeFactoryImpl.JavaType) typeField1.getType()).getJavaClass());
}else {
groupAggregationBuilder.subAggregation(AggregationBuilders.max(functionName).field(typeField1.getName().toLowerCase()));
}
break;
case COUNT:
if(groupAggregationBuilder == null) {
if (argList == null || argList.size() == 0)//count(*)
esTable.addAggregationBuilder(AggregationBuilders.count(functionName), Long.class);
else
esTable.addAggregationBuilder(AggregationBuilders.count(functionName).field(transFieldName(fieldList.get(argList.get(0)).getName().toLowerCase(), fieldList)), Long.class);
}else {
groupAggregationBuilder.subAggregation(AggregationBuilders.count(functionName).field(transFieldName(fieldList.get(argList.get(0)).getName().toLowerCase(), fieldList)));
groupAggregationBuilder2.subAggregation(AggregationBuilders.count(functionName).field(transFieldName(fieldList.get(argList.get(0)).getName().toLowerCase(), fieldList)));
}
break;
case SUM:
if(groupAggregationBuilder == null) {
esTable.addAggregationBuilder(AggregationBuilders.sum(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase()), Double.class);
}else {
String s = fieldList.get(argList.get(0)).getName().toLowerCase();
SumAggregationBuilder sumAgg = AggregationBuilders.sum(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase());
groupAggregationBuilder.subAggregation(sumAgg);
groupAggregationBuilder2.subAggregation(sumAgg);
}
break;
case AVG:
if(groupAggregationBuilder == null) {
esTable.addAggregationBuilder(AggregationBuilders.avg(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase()), Double.class);
}else {
AvgAggregationBuilder avgAgg = AggregationBuilders.avg(functionName).field(fieldList.get(argList.get(0)).getName().toLowerCase());
groupAggregationBuilder.subAggregation(avgAgg);
groupAggregationBuilder2.subAggregation(avgAgg);
}
break;
default:break;
}
}
if (groupAggregationBuilder != null) {
esTable.addAggregationBuilder((AbstractAggregationBuilder) groupAggregationBuilder,String.class);
esTable.addAggregationBuilderList2(groupAggregationBuilder);
}
esTable.setOut(out);
}
