Package org.apache.hadoop.yarn.api.records

Examples of org.apache.hadoop.yarn.api.records.NodeId


    launcher.close();
  }

  private static Container createMockContainer() {
    Container container = mock(Container.class);
    NodeId nodeId = NodeId.newInstance("foo.bar.org", 1234);
    when(container.getNodeId()).thenReturn(nodeId);
    return container;
  }
View Full Code Here


    String hostname3 = "localhost3";
    Resource capability = BuilderUtils.newResource(1024, 1);

    RegisterNodeManagerRequest request1 = recordFactory
        .newRecordInstance(RegisterNodeManagerRequest.class);
    NodeId nodeId1 = NodeId.newInstance(hostname1, 0);
    request1.setNodeId(nodeId1);
    request1.setHttpPort(0);
    request1.setResource(capability);
    resourceTrackerService.registerNodeManager(request1);

    RegisterNodeManagerRequest request2 = recordFactory
        .newRecordInstance(RegisterNodeManagerRequest.class);
    NodeId nodeId2 = NodeId.newInstance(hostname2, 0);
    request2.setNodeId(nodeId2);
    request2.setHttpPort(0);
    request2.setResource(capability);
    resourceTrackerService.registerNodeManager(request2);
   
    int waitCount = 0;
    while(ClusterMetrics.getMetrics().getNumLostNMs()!=2 && waitCount ++<20){
      synchronized (this) {
        wait(100);
      }
    }
    Assert.assertEquals(2, ClusterMetrics.getMetrics().getNumLostNMs());

    request3 = recordFactory
        .newRecordInstance(RegisterNodeManagerRequest.class);
    NodeId nodeId3 = NodeId.newInstance(hostname3, 0);
    request3.setNodeId(nodeId3);
    request3.setHttpPort(0);
    request3.setResource(capability);
    resourceTrackerService
        .registerNodeManager(request3);
View Full Code Here

        HashMap<TaskAttemptId, Container> taskSet = i == 0 ? assignedRequests.maps
            : assignedRequests.reduces;
        // kill running containers
        for (Map.Entry<TaskAttemptId, Container> entry : taskSet.entrySet()) {
          TaskAttemptId tid = entry.getKey();
          NodeId taskAttemptNodeId = entry.getValue().getNodeId();
          if (unusableNodes.contains(taskAttemptNodeId)) {
            LOG.info("Killing taskAttempt:" + tid
                + " because it is running on unusable node:"
                + taskAttemptNodeId);
            eventHandler.handle(new TaskAttemptKillEvent(tid,
View Full Code Here

    return allocateContainer(appmaster, id, null);
  }

  protected Container allocateContainer(Object appmaster, int id, String host) throws Exception {
    ContainerId containerId = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 0);
    NodeId nodeId = MockUtils.getMockNodeId(host, 0);
    Priority priority = MockUtils.getMockPriority(0);
    Container container = MockUtils.getMockContainer(containerId, nodeId, null, priority);
    TestUtils.callMethod("onContainerAllocated", appmaster, new Object[]{container}, new Class<?>[]{Container.class});
    return container;
  }
View Full Code Here

    return allocateContainer(appmaster, id, null);
  }

  protected Container allocateContainer(Object appmaster, int id, String host) throws Exception {
    ContainerId containerId = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 0);
    NodeId nodeId = MockUtils.getMockNodeId(host, 0);
    Priority priority = MockUtils.getMockPriority(0);
    Container container = MockUtils.getMockContainer(containerId, nodeId, null, priority);
    TestUtils.callMethod("onContainerAllocated", appmaster, new Object[]{container}, new Class<?>[]{Container.class});
    return container;
  }
View Full Code Here

    assertThat(satisfyState.getAllocateData().getRacks().get("/rack2"), is(2));
    assertThat(satisfyState.getAllocateData().getRacks().get("/rack3"), is(2));
    assertThat(satisfyState.getRemoveData().size(), is(0));

    ContainerId id1 = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 1);
    NodeId nodeId1 = MockUtils.getMockNodeId("host1", 0);
    Priority priority = MockUtils.getMockPriority(0);
    Container container1 = MockUtils.getMockContainer(id1, nodeId1, null, priority);
    DefaultGridMember member1 = new DefaultGridMember(container1);
    boolean accepted = projection.acceptMember(member1);
    assertThat(accepted, is(true));
View Full Code Here

    return allocateContainer(appmaster, id, null);
  }

  protected Container allocateContainer(Object appmaster, int id, String host) throws Exception {
    ContainerId containerId = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 0);
    NodeId nodeId = MockUtils.getMockNodeId(host, 0);
    Priority priority = MockUtils.getMockPriority(0);
    Container container = MockUtils.getMockContainer(containerId, nodeId, null, priority);
    TestUtils.callMethod("onContainerAllocated", appmaster, new Object[]{container}, new Class<?>[]{Container.class});
    return container;
  }
View Full Code Here

    assertThat(satisfyState.getAllocateData().getHosts().size(), is(3));
    assertThat(satisfyState.getAllocateData().getRacks().size(), is(0));
    assertThat(satisfyState.getRemoveData().size(), is(0));

    ContainerId id1 = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 1);
    NodeId nodeId1 = MockUtils.getMockNodeId("host1", 0);
    Priority priority = MockUtils.getMockPriority(0);
    Container container1 = MockUtils.getMockContainer(id1, nodeId1, null, priority);
    DefaultGridMember member1 = new DefaultGridMember(container1);
    boolean accepted = projection.acceptMember(member1);
    assertThat(accepted, is(true));
View Full Code Here

    assertThat(satisfyState.getAllocateData().getHosts().size(), is(2));
    assertThat(satisfyState.getAllocateData().getRacks().size(), is(0));
    assertThat(satisfyState.getRemoveData().size(), is(0));

    ContainerId id1 = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 1);
    NodeId nodeId1 = MockUtils.getMockNodeId("host1", 0);
    Priority priority1 = MockUtils.getMockPriority(0);
    Container container1 = MockUtils.getMockContainer(id1, nodeId1, null, priority1);
    DefaultGridMember member1 = new DefaultGridMember(container1);
    assertThat(projection.acceptMember(member1), is(true));

    ContainerId id2 = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 2);
    NodeId nodeId2 = MockUtils.getMockNodeId("host1", 0);
    Priority priority2 = MockUtils.getMockPriority(0);
    Container container2 = MockUtils.getMockContainer(id2, nodeId2, null, priority2);
    DefaultGridMember member2 = new DefaultGridMember(container2);
    assertThat(projection.acceptMember(member2), is(true));

    ContainerId id3 = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 3);
    NodeId nodeId3 = MockUtils.getMockNodeId("host2", 0);
    Priority priority3 = MockUtils.getMockPriority(0);
    Container container3 = MockUtils.getMockContainer(id3, nodeId3, null, priority3);
    DefaultGridMember member3 = new DefaultGridMember(container3);
    assertThat(projection.acceptMember(member3), is(true));

    ContainerId id4 = MockUtils.getMockContainerId(MockUtils.getMockApplicationAttemptId(0, 0), 4);
    NodeId nodeId4 = MockUtils.getMockNodeId("host2", 0);
    Priority priority4 = MockUtils.getMockPriority(0);
    Container container4 = MockUtils.getMockContainer(id4, nodeId4, null, priority4);
    DefaultGridMember member4 = new DefaultGridMember(container4);
    assertThat(projection.acceptMember(member4), is(true));
View Full Code Here

    when(pri.getPriority()).thenReturn(priority);
    return pri;
  }

  public static NodeId getMockNodeId(String host, int port) {
    NodeId nodeId = mock(NodeId.class);
    doReturn(host).when(nodeId).getHost();
    doReturn(port).when(nodeId).getPort();
    return nodeId;
  }
View Full Code Here

TOP

Related Classes of org.apache.hadoop.yarn.api.records.NodeId

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.