コード例 #1
0
 /**
  * Add multiple node labels to repository
  *
  * @param labels new node labels added
  */
 @VisibleForTesting
 public void addToCluserNodeLabelsWithDefaultExclusivity(Set<String> labels) throws IOException {
   Set<NodeLabel> nodeLabels = new HashSet<NodeLabel>();
   for (String label : labels) {
     nodeLabels.add(NodeLabel.newInstance(label));
   }
   addToCluserNodeLabels(nodeLabels);
 }
コード例 #2
0
 public List<NodeLabel> getClusterNodeLabels() {
   try {
     readLock.lock();
     List<NodeLabel> nodeLabels = new ArrayList<>();
     for (RMNodeLabel label : labelCollections.values()) {
       if (!label.getLabelName().equals(NO_LABEL)) {
         nodeLabels.add(NodeLabel.newInstance(label.getLabelName(), label.getIsExclusive()));
       }
     }
     return nodeLabels;
   } finally {
     readLock.unlock();
   }
 }
コード例 #3
0
  @Test
  public void testNodeRegistrationWithLabels() throws Exception {
    writeToHostsFile("host2");
    Configuration conf = new Configuration();
    conf.set(YarnConfiguration.RM_NODES_INCLUDE_FILE_PATH, hostFile.getAbsolutePath());
    conf.set(
        YarnConfiguration.NODELABEL_CONFIGURATION_TYPE,
        YarnConfiguration.DISTRIBUTED_NODELABEL_CONFIGURATION_TYPE);

    final RMNodeLabelsManager nodeLabelsMgr = new NullRMNodeLabelsManager();

    rm =
        new MockRM(conf) {
          @Override
          protected RMNodeLabelsManager createNodeLabelManager() {
            return nodeLabelsMgr;
          }
        };
    rm.start();

    try {
      nodeLabelsMgr.addToCluserNodeLabelsWithDefaultExclusivity(toSet("A", "B", "C"));
    } catch (IOException e) {
      Assert.fail("Caught Exception while intializing");
      e.printStackTrace();
    }

    ResourceTrackerService resourceTrackerService = rm.getResourceTrackerService();
    RegisterNodeManagerRequest registerReq = Records.newRecord(RegisterNodeManagerRequest.class);
    NodeId nodeId = NodeId.newInstance("host2", 1234);
    Resource capability = BuilderUtils.newResource(1024, 1);
    registerReq.setResource(capability);
    registerReq.setNodeId(nodeId);
    registerReq.setHttpPort(1234);
    registerReq.setNMVersion(YarnVersionInfo.getVersion());
    registerReq.setNodeLabels(toSet(NodeLabel.newInstance("A")));
    RegisterNodeManagerResponse response = resourceTrackerService.registerNodeManager(registerReq);

    Assert.assertEquals(
        "Action should be normal on valid Node Labels",
        NodeAction.NORMAL,
        response.getNodeAction());
    assertCollectionEquals(
        nodeLabelsMgr.getNodeLabels().get(nodeId),
        ResourceTrackerService.convertToStringSet(registerReq.getNodeLabels()));
    Assert.assertTrue(
        "Valid Node Labels were not accepted by RM", response.getAreNodeLabelsAcceptedByRM());
    rm.stop();
  }