Examples with Trace org.apache.drill.exec.physical.config.Trace used on opensource projects

Example 1 with Trace

use of org.apache.drill.exec.physical.config.Trace in project drill by apache.

the class TraceInjector method visitOp.

/**
     * Traverse the physical plan and inject the trace operator after
     * every operator.
     * @param op Physical operator under which the trace operator will be injected
     * @param context Fragment context
     * @return same physical operator as passed in, but its child will be a trace operator
     *         whose child will be the original child of this operator
     * @throws ExecutionSetupException
     */
@Override
public PhysicalOperator visitOp(PhysicalOperator op, FragmentContext context) throws ExecutionSetupException {
    List<PhysicalOperator> newChildren = Lists.newArrayList();
    List<PhysicalOperator> list = null;
    PhysicalOperator newOp = op;
    /* Get the list of child operators */
    for (PhysicalOperator child : op) {
        newChildren.add(child.accept(this, context));
    }
    list = Lists.newArrayList();
    /* For every child operator create a trace operator as its parent */
    for (int i = 0; i < newChildren.size(); i++) {
        String traceTag = newChildren.get(i).toString() + Integer.toString(traceTagCount++);
        /* Trace operator */
        Trace traceOp = new Trace(newChildren.get(i), traceTag);
        list.add(traceOp);
    }
    /* Inject trace operator */
    if (list.size() > 0) {
        newOp = op.getNewWithChildren(list);
    }
    newOp.setOperatorId(op.getOperatorId());
    return newOp;
}