Java RexNode.accept方法代码示例

import org.eigenbase.rex.RexNode; //导入方法依赖的package包/类
@Override
public ChildVisitor visitCall(RexCall call) {
  final SqlSyntax syntax = call.getOperator().getSyntax();
  switch (syntax) {
    case Binary:
      return visitBinary(call);
    case Function:
      buf.append(call.getOperator().getName().toLowerCase()).append("(");
      for (Ord<RexNode> operand : Ord.zip(call.getOperands())) {
        buf.append(operand.i > 0 ? ", " : "");
        operand.e.accept(this);
      }
      return null;
    case Special:
      switch (call.getKind()) {
        case Cast:
          // Ignore casts. Drill is type-less.
          return call.getOperands().get(0).accept(this);
        default:
          break;
      }
      if (call.getOperator() == SqlStdOperatorTable.itemOp) {
        final RexNode left = call.getOperands().get(0);
        final int length = buf.length();
        left.accept(this);
        if (buf.length() > length) {
          // check before generating empty LHS if inputName is null
          buf.append('.');
        }
        // return buf.append(field);
        return null;
      }
      // fall through
    default:
      throw new AssertionError("todo: implement syntax " + syntax + "(" + call + ")");
  }
}
 

import org.eigenbase.rex.RexNode; //导入方法依赖的package包/类
@Override
public Operator getStratosphereOperator() {
	System.err.println("Preparing operator "+this.getDigest());

	// get Input
	Operator inputOp = StratosphereRelUtils.openSingleInputOperator(getInputs());
	System.err.println("Input is "+inputOp);

	final RexBuilder rexBuilder = getCluster().getRexBuilder();
	final RexExecutorImpl executor = new RexExecutorImpl(null);

	final List<RexNode> complexExps = new ArrayList<RexNode>();


	StratosphereRexUtils.GetInputRefVisitor replaceInputRefsByExternalInputRefsVisitor = new StratosphereRexUtils.GetInputRefVisitor();

	Set<StratosphereRexUtils.ProjectionFieldProperties> fields = new HashSet<StratosphereRexUtils.ProjectionFieldProperties>();
	int pos = 0;
	int rexpos = 0;
	for(RexNode rex : exps) {
		if(rex.getKind() != SqlKind.INPUT_REF) {
			complexExps.add(rex);
		}
		rex.accept(replaceInputRefsByExternalInputRefsVisitor);
		boolean trivialProjection = rex.getKind() == SqlKind.INPUT_REF;
		for(Pair<Integer, RelDataType> rexInput : replaceInputRefsByExternalInputRefsVisitor.getInputPosAndType() ) {
			StratosphereRexUtils.ProjectionFieldProperties field = new StratosphereRexUtils.ProjectionFieldProperties();
			field.positionInOutput = pos;
			field.fieldIndex = pos;
			field.positionInRex = rexpos;
			field.positionInInput = rexInput.getKey();
			field.inFieldType = StratosphereRelUtils.getTypeClass(rexInput.getValue());
			field.outFieldType = StratosphereRelUtils.getTypeClass( getRowType().getFieldList().get(pos).getType() );
			field.trivialProjection = trivialProjection;
			if(fields.add(field)) {
				System.err.println("adding projection field="+field+" for rex="+rex);
			} else {
				System.err.println("fields already contained "+field+" for rex="+rex);
			}
			field.name = rex.toString();
		}
		pos++;
		if(!trivialProjection) {
			rexpos++;
		}
		replaceInputRefsByExternalInputRefsVisitor.resetInputList();
	}

	RexExecutable executable = executor.getExecutable(rexBuilder, complexExps, getInput(0).getRowType() );
	System.err.println("Code="+executable.getSource());
	// create MapOperator
	MapOperator proj = MapOperator	.builder(new StratosphereSqlProjectionMapOperator(executable.getFunction(),
																					fields, executable.getSource()))
									.input(inputOp)
									.name(buildName())
									.build();
	return proj;
}
 

import org.eigenbase.rex.RexNode; //导入方法依赖的package包/类
@Override
public int implement(Plan implementor) {
  
  implementor.visitChild(getChild());
  
  implementor.table = accumuloAccessor;
  
  OperationVisitor visitor = new OperationVisitor(getChild());
  
  cosmos.sql.call.impl.Projection projections = new cosmos.sql.call.impl.Projection();
  for (RexNode node : exps) {
    CallIfc<?> projection = node.accept(visitor);
    projections.addChild(projection.getClass().getSimpleName(), projection);
    
  }
  implementor.add(projections.getClass().getSimpleName(), projections);
  
  return 1;
  
}