Java AggregateCall.getName方法代码示例

import org.apache.calcite.rel.core.AggregateCall; //导入方法依赖的package包/类
@Override
public PhysicalOperator getPhysicalOperator(PhysicalPlanCreator creator) throws IOException {
  Prel child = (Prel) this.getInput();

  PhysicalOperator childPOP = child.getPhysicalOperator(creator);

  final List<String> childFields = getInput().getRowType().getFieldNames();

  // We don't support distinct partitions
  checkState(groups.size() == 1, "Only one window is expected in WindowPrel");

  Group window = groups.get(0);
  List<NamedExpression> withins = Lists.newArrayList();
  List<NamedExpression> aggs = Lists.newArrayList();
  List<Order.Ordering> orderings = Lists.newArrayList();

  for (int group : BitSets.toIter(window.keys)) {
    FieldReference fr = new FieldReference(childFields.get(group), ExpressionPosition.UNKNOWN);
    withins.add(new NamedExpression(fr, fr));
  }

  for (AggregateCall aggCall : window.getAggregateCalls(this)) {
    FieldReference ref = new FieldReference(aggCall.getName());
    LogicalExpression expr = toDrill(aggCall, childFields);
    aggs.add(new NamedExpression(expr, ref));
  }

  for (RelFieldCollation fieldCollation : window.orderKeys.getFieldCollations()) {
    orderings.add(new Order.Ordering(fieldCollation.getDirection(), new FieldReference(childFields.get(fieldCollation.getFieldIndex())), fieldCollation.nullDirection));
  }

  WindowPOP windowPOP = new WindowPOP(
      childPOP,
      withins.toArray(new NamedExpression[withins.size()]),
      aggs.toArray(new NamedExpression[aggs.size()]),
      orderings.toArray(new Order.Ordering[orderings.size()]),
      Long.MIN_VALUE, //TODO: Get first/last to work
      Long.MIN_VALUE);

  creator.addMetadata(this, windowPOP);
  return windowPOP;
}
 

import org.apache.calcite.rel.core.AggregateCall; //导入方法依赖的package包/类
@Override
public PhysicalOperator getPhysicalOperator(PhysicalPlanCreator creator) throws IOException {
  Prel child = (Prel) this.getInput();

  PhysicalOperator childPOP = child.getPhysicalOperator(creator);

  final List<String> childFields = getInput().getRowType().getFieldNames();

  // We don't support distinct partitions
  checkState(groups.size() == 1, "Only one window is expected in WindowPrel");

  Group window = groups.get(0);
  List<NamedExpression> withins = Lists.newArrayList();
  List<NamedExpression> aggs = Lists.newArrayList();
  List<Order.Ordering> orderings = Lists.newArrayList();

  for (int group : BitSets.toIter(window.keys)) {
    FieldReference fr = new FieldReference(childFields.get(group));
    withins.add(new NamedExpression(fr, fr));
  }

  for (AggregateCall aggCall : window.getAggregateCalls(this)) {
    FieldReference ref = new FieldReference(aggCall.getName());
    LogicalExpression expr = toExpr(aggCall, childFields);
    aggs.add(new NamedExpression(expr, ref));
  }

  for (RelFieldCollation fieldCollation : window.orderKeys.getFieldCollations()) {
    orderings.add(new Order.Ordering(fieldCollation.getDirection(), new FieldReference(childFields.get(fieldCollation.getFieldIndex())), fieldCollation.nullDirection));
  }

  WindowPOP windowPOP = new WindowPOP(
      childPOP,
      withins,
      aggs,
      orderings,
      window.isRows,
      WindowPOP.newBound(window.lowerBound),
      WindowPOP.newBound(window.upperBound));

  creator.addMetadata(this, windowPOP);
  return windowPOP;
}
 

import org.apache.calcite.rel.core.AggregateCall; //导入方法依赖的package包/类
@Override
public PhysicalOperator getPhysicalOperator(PhysicalPlanCreator creator) throws IOException {
  Prel child = (Prel) this.getInput();

  PhysicalOperator childPOP = child.getPhysicalOperator(creator);

  final List<String> childFields = getInput().getRowType().getFieldNames();

  // We don't support distinct partitions
  checkState(groups.size() == 1, "Only one window is expected in WindowPrel");

  Group window = groups.get(0);
  List<NamedExpression> withins = Lists.newArrayList();
  List<NamedExpression> aggs = Lists.newArrayList();
  List<Order.Ordering> orderings = Lists.newArrayList();

  for (int group : BitSets.toIter(window.keys)) {
    FieldReference fr = new FieldReference(childFields.get(group), ExpressionPosition.UNKNOWN);
    withins.add(new NamedExpression(fr, fr));
  }

  for (AggregateCall aggCall : window.getAggregateCalls(this)) {
    FieldReference ref = new FieldReference(aggCall.getName());
    LogicalExpression expr = toDrill(aggCall, childFields);
    aggs.add(new NamedExpression(expr, ref));
  }

  for (RelFieldCollation fieldCollation : window.orderKeys.getFieldCollations()) {
    orderings.add(new Order.Ordering(fieldCollation.getDirection(), new FieldReference(childFields.get(fieldCollation.getFieldIndex())), fieldCollation.nullDirection));
  }

  WindowPOP windowPOP = new WindowPOP(
      childPOP,
      withins,
      aggs,
      orderings,
      window.isRows,
      WindowPOP.newBound(window.lowerBound),
      WindowPOP.newBound(window.upperBound));

  creator.addMetadata(this, windowPOP);
  return windowPOP;
}
 

import org.apache.calcite.rel.core.AggregateCall; //导入方法依赖的package包/类
/**
 * Collects counts for each aggregation call.
 * Will return empty result map if was not able to determine count for at least one aggregation call,
 *
 * For each aggregate call will determine if count can be calculated. Collects counts only for COUNT function.
 * For star, not null expressions and implicit columns sets count to total record number.
 * For other cases obtains counts from group scan operator. Also count can not be calculated for parition columns.
 *
 * @param agg aggregate relational expression
 * @param scan scan relational expression
 * @param project project relational expression
 * @return result map where key is count column name, value is count value
 */
private Map<String, Long> collectCounts(PlannerSettings settings, DrillAggregateRel agg, DrillScanRel scan, DrillProjectRel project) {
  final Set<String> implicitColumnsNames = ColumnExplorer.initImplicitFileColumns(settings.getOptions()).keySet();
  final GroupScan oldGrpScan = scan.getGroupScan();
  final long totalRecordCount = oldGrpScan.getScanStats(settings).getRecordCount();
  final LinkedHashMap<String, Long> result = new LinkedHashMap<>();

  for (int i = 0; i < agg.getAggCallList().size(); i++) {
    AggregateCall aggCall = agg.getAggCallList().get(i);
  //for (AggregateCall aggCall : agg.getAggCallList()) {
    long cnt;

    // rule can be applied only for count function, return empty counts
    if (!"count".equalsIgnoreCase(aggCall.getAggregation().getName()) ) {
      return ImmutableMap.of();
    }

    if (containsStarOrNotNullInput(aggCall, agg)) {
      cnt = totalRecordCount;

    } else if (aggCall.getArgList().size() == 1) {
      // count(columnName) ==> Agg ( Scan )) ==> columnValueCount
      int index = aggCall.getArgList().get(0);

      if (project != null) {
        // project in the middle of Agg and Scan : Only when input of AggCall is a RexInputRef in Project, we find the index of Scan's field.
        // For instance,
        // Agg - count($0)
        //  \
        //  Proj - Exp={$1}
        //    \
        //   Scan (col1, col2).
        // return count of "col2" in Scan's metadata, if found.
        if (!(project.getProjects().get(index) instanceof RexInputRef)) {
          return ImmutableMap.of(); // do not apply for all other cases.
        }

        index = ((RexInputRef) project.getProjects().get(index)).getIndex();
      }

      String columnName = scan.getRowType().getFieldNames().get(index).toLowerCase();

      // for implicit column count will the same as total record count
      if (implicitColumnsNames.contains(columnName)) {
        cnt = totalRecordCount;
      } else {
        SchemaPath simplePath = SchemaPath.getSimplePath(columnName);

        if (ColumnExplorer.isPartitionColumn(settings.getOptions(), simplePath)) {
          return ImmutableMap.of();
        }

        cnt = oldGrpScan.getColumnValueCount(simplePath);
        if (cnt == GroupScan.NO_COLUMN_STATS) {
          // if column stats is not available don't apply this rule, return empty counts
          return ImmutableMap.of();
        }
      }
    } else {
      return ImmutableMap.of();
    }

    String name = "count" + i + "$" + (aggCall.getName() == null ? aggCall.toString() : aggCall.getName());
    result.put(name, cnt);
  }

  return ImmutableMap.copyOf(result);
}
 

import org.apache.calcite.rel.core.AggregateCall; //导入方法依赖的package包/类
private String getPigAggregateCall(String relAlias, AggregateCall aggCall) {
  final PigAggFunction aggFunc = toPigAggFunc(aggCall);
  final String alias = aggCall.getName();
  final String fields = Joiner.on(", ").join(getArgNames(relAlias, aggCall));
  return aggFunc.name() + "(" + fields + ") AS " + alias;
}