コード例 #1
0
 public ClusterMetrics getClusterMetrics() throws IOException, InterruptedException {
   GetClusterMetricsRequest request =
       recordFactory.newRecordInstance(GetClusterMetricsRequest.class);
   GetClusterMetricsResponse response = applicationsManager.getClusterMetrics(request);
   YarnClusterMetrics metrics = response.getClusterMetrics();
   ClusterMetrics oldMetrics =
       new ClusterMetrics(
           1,
           1,
           1,
           1,
           1,
           1,
           metrics.getNumNodeManagers() * 10,
           metrics.getNumNodeManagers() * 2,
           1,
           metrics.getNumNodeManagers(),
           0,
           0);
   return oldMetrics;
 }
コード例 #2
0
ファイル: YarnClientImpl.java プロジェクト: JoeChien23/hadoop
 @Override
 public YarnClusterMetrics getYarnClusterMetrics() throws YarnException, IOException {
   GetClusterMetricsRequest request = Records.newRecord(GetClusterMetricsRequest.class);
   GetClusterMetricsResponse response = rmClient.getClusterMetrics(request);
   return response.getClusterMetrics();
 }
コード例 #3
0
ファイル: TestYARNRunner.java プロジェクト: wangxuemin/coding
  @Test(timeout = 20000)
  public void testResourceMgrDelegate() throws Exception {
    /* we not want a mock of resource mgr delegate */
    final ClientRMProtocol clientRMProtocol = mock(ClientRMProtocol.class);
    ResourceMgrDelegate delegate =
        new ResourceMgrDelegate(conf) {
          @Override
          public synchronized void start() {
            this.rmClient = clientRMProtocol;
          }
        };
    /* make sure kill calls finish application master */
    when(clientRMProtocol.forceKillApplication(any(KillApplicationRequest.class))).thenReturn(null);
    delegate.killApplication(appId);
    verify(clientRMProtocol).forceKillApplication(any(KillApplicationRequest.class));

    /* make sure getalljobs calls get all applications */
    when(clientRMProtocol.getAllApplications(any(GetAllApplicationsRequest.class)))
        .thenReturn(recordFactory.newRecordInstance(GetAllApplicationsResponse.class));
    delegate.getAllJobs();
    verify(clientRMProtocol).getAllApplications(any(GetAllApplicationsRequest.class));

    /* make sure getapplication report is called */
    when(clientRMProtocol.getApplicationReport(any(GetApplicationReportRequest.class)))
        .thenReturn(recordFactory.newRecordInstance(GetApplicationReportResponse.class));
    delegate.getApplicationReport(appId);
    verify(clientRMProtocol).getApplicationReport(any(GetApplicationReportRequest.class));

    /* make sure metrics is called */
    GetClusterMetricsResponse clusterMetricsResponse =
        recordFactory.newRecordInstance(GetClusterMetricsResponse.class);
    clusterMetricsResponse.setClusterMetrics(
        recordFactory.newRecordInstance(YarnClusterMetrics.class));
    when(clientRMProtocol.getClusterMetrics(any(GetClusterMetricsRequest.class)))
        .thenReturn(clusterMetricsResponse);
    delegate.getClusterMetrics();
    verify(clientRMProtocol).getClusterMetrics(any(GetClusterMetricsRequest.class));

    when(clientRMProtocol.getClusterNodes(any(GetClusterNodesRequest.class)))
        .thenReturn(recordFactory.newRecordInstance(GetClusterNodesResponse.class));
    delegate.getActiveTrackers();
    verify(clientRMProtocol).getClusterNodes(any(GetClusterNodesRequest.class));

    GetNewApplicationResponse newAppResponse =
        recordFactory.newRecordInstance(GetNewApplicationResponse.class);
    newAppResponse.setApplicationId(appId);
    when(clientRMProtocol.getNewApplication(any(GetNewApplicationRequest.class)))
        .thenReturn(newAppResponse);
    delegate.getNewJobID();
    verify(clientRMProtocol).getNewApplication(any(GetNewApplicationRequest.class));

    GetQueueInfoResponse queueInfoResponse =
        recordFactory.newRecordInstance(GetQueueInfoResponse.class);
    queueInfoResponse.setQueueInfo(recordFactory.newRecordInstance(QueueInfo.class));
    when(clientRMProtocol.getQueueInfo(any(GetQueueInfoRequest.class)))
        .thenReturn(queueInfoResponse);
    delegate.getQueues();
    verify(clientRMProtocol).getQueueInfo(any(GetQueueInfoRequest.class));

    GetQueueUserAclsInfoResponse aclResponse =
        recordFactory.newRecordInstance(GetQueueUserAclsInfoResponse.class);
    when(clientRMProtocol.getQueueUserAcls(any(GetQueueUserAclsInfoRequest.class)))
        .thenReturn(aclResponse);
    delegate.getQueueAclsForCurrentUser();
    verify(clientRMProtocol).getQueueUserAcls(any(GetQueueUserAclsInfoRequest.class));
  }