@Inject RMAppAttemptBlock(ViewContext ctx, ResourceManager rm, Configuration conf) { super(rm.getClientRMService(), ctx); this.rm = rm; this.conf = conf; }
@Override public RenewDelegationTokenResponse run() throws YarnException { return rm.getClientRMService().renewDelegationToken(req); } });
@Override public CancelDelegationTokenResponse run() throws IOException, YarnException { return rm.getClientRMService().cancelDelegationToken(req); } });
@Override public RenewDelegationTokenResponse run() throws IOException, YarnException { return rm.getClientRMService().renewDelegationToken(req); } });
@Override public CancelDelegationTokenResponse run() throws IOException, YarnException { return rm.getClientRMService().cancelDelegationToken(req); } });
@Override public SubmitApplicationResponse run() throws IOException, YarnException { return rm.getClientRMService().submitApplication(req); } });
@Override public CancelDelegationTokenResponse run() throws IOException, YarnException { return rm.getClientRMService().cancelDelegationToken(req); } });
@Override public SubmitApplicationResponse run() throws IOException, YarnException { return rm.getClientRMService().submitApplication(req); } });
@Override protected ApplicationReport getApplicationReport( GetApplicationReportRequest request) throws YarnException, IOException { return rm.getClientRMService().getApplicationReport(request) .getApplicationReport(); }
@Override protected List<ContainerReport> getContainersReport( GetContainersRequest request) throws YarnException, IOException { return rm.getClientRMService().getContainers(request).getContainerList(); }
@Override protected ApplicationReport getApplicationReport( final GetApplicationReportRequest request) throws YarnException, IOException { return rm.getClientRMService().getApplicationReport(request) .getApplicationReport(); }
@Override public KillApplicationResponse run() throws IOException, YarnException { KillApplicationRequest req = KillApplicationRequest.newInstance(appid); return rm.getClientRMService().forceKillApplication(req); } });
@Override protected List<ApplicationAttemptReport> getApplicationAttemptsReport( GetApplicationAttemptsRequest request) throws YarnException, IOException { return rm.getClientRMService().getApplicationAttempts(request) .getApplicationAttemptList(); }
@Override protected List<ApplicationAttemptReport> getApplicationAttemptsReport( final GetApplicationAttemptsRequest request) throws YarnException, IOException { return rm.getClientRMService().getApplicationAttempts(request) .getApplicationAttemptList(); }
@Override public KillApplicationResponse run() throws IOException, YarnException { KillApplicationRequest req = KillApplicationRequest.newInstance(appid); if (diagnostic != null) { req.setDiagnostics(diagnostic); } return rm.getClientRMService().forceKillApplication(req); } });
@Override public void configure(Binder binder) { try { ResourceManager mockRm = mockRm(3, 1, 2, 8*GiB); binder.bind(ResourceManager.class).toInstance(mockRm); binder.bind(ApplicationBaseProtocol.class) .toInstance(mockRm.getClientRMService()); } catch (IOException e) { throw new IllegalStateException(e); } } });
@Override public Void run() throws IOException, YarnException { Priority priority = Priority.newInstance(appPriority); UpdateApplicationPriorityRequest request = UpdateApplicationPriorityRequest .newInstance(app.getApplicationId(), priority); rm.getClientRMService().updateApplicationPriority(request); return null; } });
private static ResourceManager mockRmWithApps(RMContext rmContext) throws IOException { ResourceManager rm = mock(ResourceManager.class); ResourceScheduler rs = mockFairSchedulerWithoutApps(rmContext); ClientRMService clientRMService = mockClientRMService(rmContext); when(rm.getResourceScheduler()).thenReturn(rs); when(rm.getRMContext()).thenReturn(rmContext); when(rm.getClientRMService()).thenReturn(clientRMService); return rm; }
private static ResourceManager mockRmWithApps(RMContext rmContext) throws IOException { ResourceManager rm = mock(ResourceManager.class); ResourceScheduler rs = mockFairSchedulerWithoutApps(rmContext); ClientRMService clientRMService = mockClientRMService(rmContext); when(rm.getResourceScheduler()).thenReturn(rs); when(rm.getRMContext()).thenReturn(rmContext); when(rm.getClientRMService()).thenReturn(clientRMService); return rm; }
public static ResourceManager mockRm(RMContext rmContext) throws IOException { ResourceManager rm = mock(ResourceManager.class); ResourceScheduler rs = mockCapacityScheduler(); ApplicationACLsManager aclMgr = mockAppACLsManager(); ClientRMService clientRMService = mockClientRMService(rmContext); when(rm.getResourceScheduler()).thenReturn(rs); when(rm.getRMContext()).thenReturn(rmContext); when(rm.getApplicationACLsManager()).thenReturn(aclMgr); when(rm.getClientRMService()).thenReturn(clientRMService); return rm; }