Ejemplo n.º 1
0
 @Override
 public String toString() {
   return "NodeOp{ "
       + executionPhase.executionPhaseId()
       + " "
       + executionPhase.name()
       + ", downstreamNodes="
       + downstreamNodes
       + ", downstreamPhase="
       + downstreamExecutionPhaseId
       + ", downstreamInputId="
       + downstreamExecutionPhaseInputId
       + '}';
 }
Ejemplo n.º 2
0
  public static NodeOperation withDownstream(
      ExecutionPhase executionPhase,
      ExecutionPhase downstreamExecutionPhase,
      byte inputId,
      String localNodeId) {
    if (downstreamExecutionPhase.executionNodes().isEmpty()) {
      if (executionPhase instanceof UpstreamPhase
          && executionPhase.executionNodes().size() == 1
          && executionPhase.executionNodes().contains(localNodeId)) {
        ((UpstreamPhase) executionPhase).distributionType(DistributionType.SAME_NODE);
        LOGGER.trace(
            "Phase uses SAME_NODE downstream, reason: ON HANDLER, executionNodes: {}, phase: {}",
            executionPhase.executionNodes(),
            executionPhase);
        return new NodeOperation(
            executionPhase,
            ImmutableList.<String>of(),
            downstreamExecutionPhase.executionPhaseId(),
            inputId);
      }
      return new NodeOperation(
          executionPhase,
          ImmutableList.of(ExecutionPhase.DIRECT_RETURN_DOWNSTREAM_NODE),
          downstreamExecutionPhase.executionPhaseId(),
          inputId);
    } else {
      if (executionPhase instanceof UpstreamPhase) {
        if (executionPhase.executionNodes().size() == 1
            && executionPhase.executionNodes().equals(downstreamExecutionPhase.executionNodes())) {
          ((UpstreamPhase) executionPhase).distributionType(DistributionType.SAME_NODE);
          LOGGER.trace(
              "Phase uses SAME_NODE downstream, reason: ON DOWNSTRREAM NODE, executionNodes: {}, phase: {}",
              executionPhase.executionNodes(),
              executionPhase);
        }
      }

      return new NodeOperation(
          executionPhase,
          downstreamExecutionPhase.executionNodes(),
          downstreamExecutionPhase.executionPhaseId(),
          inputId);
    }
  }