protected List<ApplicationReport> fetchAppReports() throws YarnException, IOException { List<ApplicationReport> ret; EnumSet<YarnApplicationState> states = EnumSet.of(YarnApplicationState.ACCEPTED, YarnApplicationState.RUNNING); GetApplicationsRequest req = GetApplicationsRequest.newInstance(types, states); req.setQueues(queues); req.setUsers(users); ret = applicationReportsCache.getIfPresent(req); if (ret != null) { return ret; } ret = client.getApplications(queues, users, types, states); applicationReportsCache.put(req, ret); return ret; }
protected List<ApplicationReport> fetchAppReports() throws YarnException, IOException { List<ApplicationReport> ret; EnumSet<YarnApplicationState> states = EnumSet.of(YarnApplicationState.ACCEPTED, YarnApplicationState.RUNNING); GetApplicationsRequest req = GetApplicationsRequest.newInstance(types, states); req.setQueues(queues); req.setUsers(users); ret = applicationReportsCache.getIfPresent(req); if (ret != null) { return ret; } ret = client.getApplications(queues, users, types, states); applicationReportsCache.put(req, ret); return ret; }
@Override public List<ApplicationReport> getApplications(Set<String> queues, Set<String> users, Set<String> applicationTypes, EnumSet<YarnApplicationState> applicationStates) throws YarnException, IOException { GetApplicationsRequest request = GetApplicationsRequest.newInstance(applicationTypes, applicationStates); request.setQueues(queues); request.setUsers(users); GetApplicationsResponse response = rmClient.getApplications(request); return response.getApplicationList(); }
@Override public List<ApplicationReport> getApplications(Set<String> queues, Set<String> users, Set<String> applicationTypes, EnumSet<YarnApplicationState> applicationStates) throws YarnException, IOException { GetApplicationsRequest request = GetApplicationsRequest.newInstance(applicationTypes, applicationStates); request.setQueues(queues); request.setUsers(users); GetApplicationsResponse response = rmClient.getApplications(request); return response.getApplicationList(); }
public GetApplicationsRequest build() { validate(); GetApplicationsRequest request = GetApplicationsRequest.newInstance(); Set<String> appStates = parseQueries(statesQuery, true); if (!appStates.isEmpty()) { request.setApplicationStates(appStates); } if (!users.isEmpty()) { request.setUsers(users); } if (!queues.isEmpty()) { request.setQueues(queues); } if (limitNumber != null) { request.setLimit(limitNumber); } request.setStartRange(startedTimeBegin, startedTimeEnd); request.setFinishRange(finishTimeBegin, finishTimeEnd); if (!appTypes.isEmpty()) { request.setApplicationTypes(appTypes); } if (!appTags.isEmpty()) { request.setApplicationTags(appTags); } return request; } }
@Test(expected = BadRequestException.class) public void testRequestWithQueueQueryNotExistingQueue() throws IOException { CapacityScheduler cs = mock(CapacityScheduler.class); when(cs.getQueueInfo(eq("queue1"), anyBoolean(), anyBoolean())) .thenThrow(new IOException()); ResourceManager rm = mock(ResourceManager.class); when(rm.getResourceScheduler()).thenReturn(cs); GetApplicationsRequest request = ApplicationsRequestBuilder.create() .withQueueQuery(rm, "queue1").build(); GetApplicationsRequest expectedRequest = getDefaultRequest(); expectedRequest.setQueues(Sets.newHashSet("queue1")); assertEquals(expectedRequest, request); }
request.setQueues(queues);
@Test public void testRequestWithQueueQueryExistingQueue() { ResourceManager rm = mock(ResourceManager.class); GetApplicationsRequest request = ApplicationsRequestBuilder.create() .withQueueQuery(rm, "queue1").build(); GetApplicationsRequest expectedRequest = getDefaultRequest(); expectedRequest.setQueues(Sets.newHashSet("queue1")); assertEquals(expectedRequest, request); }
request.setQueues(queueSet);
request.setQueues(queueSet);
request.setQueues(queues);
request.setQueues(queues);
request.setQueues(queues);
request.setQueues(queues); request.setApplicationTypes(applicationTypes); request.setApplicationTags(applicationTags);
request.setQueues(queues); request.setApplicationTypes(applicationTypes); request.setApplicationTags(applicationTags);
request.setQueues(queues); request.setApplicationTypes(applicationTypes); request.setApplicationTags(applicationTags);
request.setQueues(queues); request.setApplicationTypes(applicationTypes); request.setApplicationTags(applicationTags);