Java NodeState.DECOMMISSIONING属性代码示例

/**
 * Refresh the nodes gracefully
 *
 * @param conf
 * @throws IOException
 * @throws YarnException
 */
public void refreshNodesGracefully(Configuration conf) throws IOException,
    YarnException {
  refreshHostsReader(conf);
  for (Entry<NodeId, RMNode> entry:rmContext.getRMNodes().entrySet()) {
    NodeId nodeId = entry.getKey();
    if (!isValidNode(nodeId.getHost())) {
      this.rmContext.getDispatcher().getEventHandler().handle(
          new RMNodeEvent(nodeId, RMNodeEventType.GRACEFUL_DECOMMISSION));
    } else {
      // Recommissioning the nodes
      if (entry.getValue().getState() == NodeState.DECOMMISSIONING
          || entry.getValue().getState() == NodeState.DECOMMISSIONED) {
        this.rmContext.getDispatcher().getEventHandler()
            .handle(new RMNodeEvent(nodeId, RMNodeEventType.RECOMMISSION));
      }
    }
  }
}
 

/**
 * It checks for any nodes in decommissioning state
 *
 * @return decommissioning nodes
 */
public Set<NodeId> checkForDecommissioningNodes() {
  Set<NodeId> decommissioningNodes = new HashSet<NodeId>();
  for (Entry<NodeId, RMNode> entry : rmContext.getRMNodes().entrySet()) {
    if (entry.getValue().getState() == NodeState.DECOMMISSIONING) {
      decommissioningNodes.add(entry.getKey());
    }
  }
  return decommissioningNodes;
}
 

/**
 * Forcefully decommission the nodes if they are in DECOMMISSIONING state
 */
public void refreshNodesForcefully() {
  for (Entry<NodeId, RMNode> entry : rmContext.getRMNodes().entrySet()) {
    if (entry.getValue().getState() == NodeState.DECOMMISSIONING) {
      this.rmContext.getDispatcher().getEventHandler().handle(
          new RMNodeEvent(entry.getKey(), RMNodeEventType.DECOMMISSION));
    }
  }
}
 

@Override
public NodeState transition(RMNodeImpl rmNode, RMNodeEvent event) {
  RMNodeReconnectEvent reconnectEvent = (RMNodeReconnectEvent) event;
  RMNode newNode = reconnectEvent.getReconnectedNode();
  rmNode.nodeManagerVersion = newNode.getNodeManagerVersion();
  List<ApplicationId> runningApps = reconnectEvent.getRunningApplications();
  boolean noRunningApps = 
      (runningApps == null) || (runningApps.size() == 0);
  
  // No application running on the node, so send node-removal event with 
  // cleaning up old container info.
  if (noRunningApps) {
    if (rmNode.getState() == NodeState.DECOMMISSIONING) {
      // When node in decommissioning, and no running apps on this node,
      // it will return as decommissioned state.
      deactivateNode(rmNode, NodeState.DECOMMISSIONED);
      return NodeState.DECOMMISSIONED;
    }
    rmNode.nodeUpdateQueue.clear();
    rmNode.context.getDispatcher().getEventHandler().handle(
        new NodeRemovedSchedulerEvent(rmNode));

    if (rmNode.getHttpPort() == newNode.getHttpPort()) {
      if (!rmNode.getTotalCapability().equals(
          newNode.getTotalCapability())) {
        rmNode.totalCapability = newNode.getTotalCapability();
      }
      if (rmNode.getState().equals(NodeState.RUNNING)) {
        // Only add old node if old state is RUNNING
        rmNode.context.getDispatcher().getEventHandler().handle(
            new NodeAddedSchedulerEvent(rmNode));
      }
    } else {
      // Reconnected node differs, so replace old node and start new node
      switch (rmNode.getState()) {
        case RUNNING:
          ClusterMetrics.getMetrics().decrNumActiveNodes();
          break;
        case UNHEALTHY:
          ClusterMetrics.getMetrics().decrNumUnhealthyNMs();
          break;
        default:
          LOG.debug("Unexpected Rmnode state");
        }
        rmNode.context.getRMNodes().put(newNode.getNodeID(), newNode);
        rmNode.context.getDispatcher().getEventHandler().handle(
            new RMNodeStartedEvent(newNode.getNodeID(), null, null));
    }

  } else {
    rmNode.httpPort = newNode.getHttpPort();
    rmNode.httpAddress = newNode.getHttpAddress();
    boolean isCapabilityChanged = false;
    if (!rmNode.getTotalCapability().equals(
        newNode.getTotalCapability())) {
      rmNode.totalCapability = newNode.getTotalCapability();
      isCapabilityChanged = true;
    }
  
    handleNMContainerStatus(reconnectEvent.getNMContainerStatuses(), rmNode);

    for (ApplicationId appId : reconnectEvent.getRunningApplications()) {
      handleRunningAppOnNode(rmNode, rmNode.context, appId, rmNode.nodeId);
    }

    if (isCapabilityChanged
        && rmNode.getState().equals(NodeState.RUNNING)) {
      // Update scheduler node's capacity for reconnect node.
      rmNode.context
          .getDispatcher()
          .getEventHandler()
          .handle(
              new NodeResourceUpdateSchedulerEvent(rmNode, ResourceOption
                  .newInstance(newNode.getTotalCapability(), -1)));
    }

  }
  return rmNode.getState();
}