From 4ea295016e20add7ab45ddc558acf77a8f1f5925 Mon Sep 17 00:00:00 2001 From: Vinod Kumar Vavilapalli Date: Tue, 17 Sep 2013 00:24:17 +0000 Subject: [PATCH] YARN-1001. Added a web-service to get statistics about per application-type per state for consumption by downstream projects. Contributed by Zhijie Shen. git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/trunk@1523855 13f79535-47bb-0310-9956-ffa450edef68 --- hadoop-yarn-project/CHANGES.txt | 3 + .../webapp/JAXBContextResolver.java | 5 +- .../resourcemanager/webapp/RMWebServices.java | 172 +++++++++++----- .../webapp/dao/ApplicationStatisticsInfo.java | 44 +++++ .../webapp/dao/StatisticsItemInfo.java | 56 ++++++ .../webapp/TestRMWebServicesApps.java | 185 +++++++++++++++++- .../src/site/apt/ResourceManagerRest.apt.vm | 123 ++++++++++++ 7 files changed, 534 insertions(+), 54 deletions(-) create mode 100644 hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ApplicationStatisticsInfo.java create mode 100644 hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/StatisticsItemInfo.java diff --git a/hadoop-yarn-project/CHANGES.txt b/hadoop-yarn-project/CHANGES.txt index 17e8a4ef75..4640e35b70 100644 --- a/hadoop-yarn-project/CHANGES.txt +++ b/hadoop-yarn-project/CHANGES.txt @@ -106,6 +106,9 @@ Release 2.1.1-beta - UNRELEASED YARN-1137. Add support whitelist for system users to Yarn container-executor.c. (rvs via tucu) + YARN-1001. Added a web-service to get statistics about per application-type + per state for consumption by downstream projects. (Zhijie Shen via vinodkv) + OPTIMIZATIONS BUG FIXES diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/JAXBContextResolver.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/JAXBContextResolver.java index 5f50ed18f1..c5776294ea 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/JAXBContextResolver.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/JAXBContextResolver.java @@ -34,12 +34,14 @@ import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptsInfo; +import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ApplicationStatisticsInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppsInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.CapacitySchedulerInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.CapacitySchedulerQueueInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.CapacitySchedulerQueueInfoList; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ClusterInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ClusterMetricsInfo; +import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.StatisticsItemInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.FifoSchedulerInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.NodeInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.NodesInfo; @@ -65,7 +67,8 @@ public class JAXBContextResolver implements ContextResolver { CapacitySchedulerInfo.class, ClusterMetricsInfo.class, SchedulerInfo.class, AppsInfo.class, NodesInfo.class, RemoteExceptionData.class, CapacitySchedulerQueueInfoList.class, - ResourceInfo.class, UsersInfo.class, UserInfo.class}; + ResourceInfo.class, UsersInfo.class, UserInfo.class, + ApplicationStatisticsInfo.class, StatisticsItemInfo.class}; public JAXBContextResolver() throws Exception { this.types = new HashSet(Arrays.asList(cTypes)); diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java index a5c2f441e0..dcc8874e46 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/RMWebServices.java @@ -22,7 +22,9 @@ import java.util.Arrays; import java.util.Collection; import java.util.EnumSet; +import java.util.HashMap; import java.util.HashSet; +import java.util.Map; import java.util.Set; import java.util.concurrent.ConcurrentMap; @@ -42,12 +44,12 @@ import org.apache.hadoop.yarn.api.records.FinalApplicationStatus; import org.apache.hadoop.yarn.api.records.NodeId; import org.apache.hadoop.yarn.api.records.NodeState; +import org.apache.hadoop.yarn.api.records.YarnApplicationState; import org.apache.hadoop.yarn.factories.RecordFactory; import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider; import org.apache.hadoop.yarn.server.resourcemanager.RMServerUtils; import org.apache.hadoop.yarn.server.resourcemanager.ResourceManager; import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp; -import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppState; import org.apache.hadoop.yarn.server.resourcemanager.rmapp.attempt.RMAppAttempt; import org.apache.hadoop.yarn.server.resourcemanager.rmnode.RMNode; import org.apache.hadoop.yarn.server.resourcemanager.scheduler.ResourceScheduler; @@ -58,6 +60,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptsInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppInfo; +import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ApplicationStatisticsInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppsInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.CapacitySchedulerInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.ClusterInfo; @@ -68,6 +71,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.NodesInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.SchedulerInfo; import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.SchedulerTypeInfo; +import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.StatisticsItemInfo; import org.apache.hadoop.yarn.server.security.ApplicationACLsManager; import org.apache.hadoop.yarn.util.ConverterUtils; import org.apache.hadoop.yarn.webapp.BadRequestException; @@ -80,6 +84,7 @@ @Path("/ws/v1/cluster") public class RMWebServices { private static final String EMPTY = ""; + private static final String ANY = "*"; private final ResourceManager rm; private static RecordFactory recordFactory = RecordFactoryProvider .getRecordFactory(null); @@ -303,53 +308,16 @@ public AppsInfo getApps(@Context HttpServletRequest hsr, "finishTimeEnd must be greater than finishTimeBegin"); } - Set appTypes = new HashSet(); - if (!applicationTypes.isEmpty()) { - for (String applicationType : applicationTypes) { - if (applicationType != null && !applicationType.trim().isEmpty()) { - if (applicationType.indexOf(",") == -1) { - appTypes.add(applicationType.trim()); - } else { - String[] types = applicationType.split(","); - for (String type : types) { - if (!type.trim().isEmpty()) { - appTypes.add(type.trim()); - } - } - } - } - } - } + Set appTypes = parseQueries(applicationTypes, false); if (!appTypes.isEmpty()) { checkAppTypes = true; } - String allAppStates; - RMAppState[] stateArray = RMAppState.values(); - allAppStates = Arrays.toString(stateArray); - - Set appStates = new HashSet(); // stateQuery is deprecated. if (stateQuery != null && !stateQuery.isEmpty()) { statesQuery.add(stateQuery); } - if (!statesQuery.isEmpty()) { - for (String applicationState : statesQuery) { - if (applicationState != null && !applicationState.isEmpty()) { - String[] states = applicationState.split(","); - for (String state : states) { - try { - RMAppState.valueOf(state.trim()); - } catch (IllegalArgumentException iae) { - throw new BadRequestException( - "Invalid application-state " + state - + " specified. It should be one of " + allAppStates); - } - appStates.add(state.trim().toLowerCase()); - } - } - } - } + Set appStates = parseQueries(statesQuery, true); if (!appStates.isEmpty()) { checkAppStates = true; } @@ -363,8 +331,8 @@ public AppsInfo getApps(@Context HttpServletRequest hsr, break; } - if (checkAppStates - && !appStates.contains(rmapp.getState().toString().toLowerCase())) { + if (checkAppStates && !appStates.contains( + rmapp.createApplicationState().toString().toLowerCase())) { continue; } if (finalStatusQuery != null && !finalStatusQuery.isEmpty()) { @@ -394,8 +362,8 @@ public AppsInfo getApps(@Context HttpServletRequest hsr, continue; } } - if (checkAppTypes - && !appTypes.contains(rmapp.getApplicationType())) { + if (checkAppTypes && !appTypes.contains( + rmapp.getApplicationType().trim().toLowerCase())) { continue; } @@ -415,6 +383,122 @@ public AppsInfo getApps(@Context HttpServletRequest hsr, return allApps; } + @GET + @Path("/appstatistics") + @Produces({ MediaType.APPLICATION_JSON, MediaType.APPLICATION_XML }) + public ApplicationStatisticsInfo getAppStatistics( + @Context HttpServletRequest hsr, + @QueryParam("states") Set stateQueries, + @QueryParam("applicationTypes") Set typeQueries) { + init(); + + // parse the params and build the scoreboard + // converting state/type name to lowercase + Set states = parseQueries(stateQueries, true); + Set types = parseQueries(typeQueries, false); + // if no types, counts the applications of any types + if (types.size() == 0) { + types.add(ANY); + } else if (types.size() != 1) { + throw new BadRequestException("# of applicationTypes = " + types.size() + + ", we temporarily support at most one applicationType"); + } + // if no states, returns the counts of all RMAppStates + if (states.size() == 0) { + for (YarnApplicationState state : YarnApplicationState.values()) { + states.add(state.toString().toLowerCase()); + } + } + // in case we extend to multiple applicationTypes in the future + Map> scoreboard = + buildScoreboard(states, types); + + // go through the apps in RM to count the numbers, ignoring the case of + // the state/type name + ConcurrentMap apps = rm.getRMContext().getRMApps(); + for (RMApp rmapp : apps.values()) { + YarnApplicationState state = rmapp.createApplicationState(); + String type = rmapp.getApplicationType().trim().toLowerCase(); + if (states.contains(state.toString().toLowerCase())) { + if (types.contains(ANY)) { + countApp(scoreboard, state, ANY); + } else if (types.contains(type)) { + countApp(scoreboard, state, type); + } + } + } + + // fill the response object + ApplicationStatisticsInfo appStatInfo = new ApplicationStatisticsInfo(); + for (Map.Entry> partScoreboard + : scoreboard.entrySet()) { + for (Map.Entry statEntry + : partScoreboard.getValue().entrySet()) { + StatisticsItemInfo statItem = new StatisticsItemInfo( + partScoreboard.getKey(), statEntry.getKey(), statEntry.getValue()); + appStatInfo.add(statItem); + } + } + return appStatInfo; + } + + private static Set parseQueries( + Set queries, boolean isState) { + Set params = new HashSet(); + if (!queries.isEmpty()) { + for (String query : queries) { + if (query != null && !query.trim().isEmpty()) { + String[] paramStrs = query.split(","); + for (String paramStr : paramStrs) { + if (paramStr != null && !paramStr.trim().isEmpty()) { + if (isState) { + try { + // enum string is in the uppercase + YarnApplicationState.valueOf(paramStr.trim().toUpperCase()); + } catch (RuntimeException e) { + YarnApplicationState[] stateArray = + YarnApplicationState.values(); + String allAppStates = Arrays.toString(stateArray); + throw new BadRequestException( + "Invalid application-state " + paramStr.trim() + + " specified. It should be one of " + allAppStates); + } + } + params.add(paramStr.trim().toLowerCase()); + } + } + } + } + } + return params; + } + + private static Map> buildScoreboard( + Set states, Set types) { + Map> scoreboard + = new HashMap>(); + // default states will result in enumerating all YarnApplicationStates + assert !states.isEmpty(); + for (String state : states) { + Map partScoreboard = new HashMap(); + scoreboard.put( + YarnApplicationState.valueOf(state.toUpperCase()), partScoreboard); + // types is verified no to be empty + for (String type : types) { + partScoreboard.put(type, 0L); + } + } + return scoreboard; + } + + private static void countApp( + Map> scoreboard, + YarnApplicationState state, String type) { + Map partScoreboard = scoreboard.get(state); + Long count = partScoreboard.get(type); + partScoreboard.put(type, count + 1L); + } + @GET @Path("/apps/{appid}") @Produces({ MediaType.APPLICATION_JSON, MediaType.APPLICATION_XML }) diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ApplicationStatisticsInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ApplicationStatisticsInfo.java new file mode 100644 index 0000000000..b77ffb4ed6 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/ApplicationStatisticsInfo.java @@ -0,0 +1,44 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; + +import java.util.ArrayList; + +import javax.xml.bind.annotation.XmlAccessType; +import javax.xml.bind.annotation.XmlAccessorType; +import javax.xml.bind.annotation.XmlRootElement; + +@XmlRootElement(name = "appStatInfo") +@XmlAccessorType(XmlAccessType.FIELD) +public class ApplicationStatisticsInfo { + + protected ArrayList statItem + = new ArrayList(); + + public ApplicationStatisticsInfo() { + } // JAXB needs this + + public void add(StatisticsItemInfo statItem) { + this.statItem.add(statItem); + } + + public ArrayList getStatItems() { + return statItem; + } + +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/StatisticsItemInfo.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/StatisticsItemInfo.java new file mode 100644 index 0000000000..e12dd5fb63 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/dao/StatisticsItemInfo.java @@ -0,0 +1,56 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.hadoop.yarn.server.resourcemanager.webapp.dao; + +import javax.xml.bind.annotation.XmlAccessType; +import javax.xml.bind.annotation.XmlAccessorType; +import javax.xml.bind.annotation.XmlRootElement; + +import org.apache.hadoop.yarn.api.records.YarnApplicationState; + +@XmlRootElement(name = "statItem") +@XmlAccessorType(XmlAccessType.FIELD) +public class StatisticsItemInfo { + + protected YarnApplicationState state; + protected String type; + protected long count; + + public StatisticsItemInfo() { + } // JAXB needs this + + public StatisticsItemInfo( + YarnApplicationState state, String type, long count) { + this.state = state; + this.type = type; + this.count = count; + } + + public YarnApplicationState getState() { + return state; + } + + public String getType() { + return type; + } + + public long getCount() { + return count; + } + +} \ No newline at end of file diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesApps.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesApps.java index 77ccbaddf5..380db7ac6f 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesApps.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/webapp/TestRMWebServicesApps.java @@ -33,6 +33,7 @@ import org.apache.hadoop.security.UserGroupInformation; import org.apache.hadoop.yarn.api.records.ContainerState; import org.apache.hadoop.yarn.api.records.FinalApplicationStatus; +import org.apache.hadoop.yarn.api.records.YarnApplicationState; import org.apache.hadoop.yarn.conf.YarnConfiguration; import org.apache.hadoop.yarn.server.resourcemanager.MockAM; import org.apache.hadoop.yarn.server.resourcemanager.MockNM; @@ -227,7 +228,8 @@ public void testAppsQueryState() throws JSONException, Exception { WebResource r = resource(); ClientResponse response = r.path("ws").path("v1").path("cluster") - .path("apps").queryParam("state", RMAppState.ACCEPTED.toString()) + .path("apps") + .queryParam("state", YarnApplicationState.ACCEPTED.toString()) .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); JSONObject json = response.getEntity(JSONObject.class); @@ -252,7 +254,7 @@ public void testAppsQueryStates() throws JSONException, Exception { WebResource r = resource(); MultivaluedMapImpl params = new MultivaluedMapImpl(); - params.add("states", RMAppState.ACCEPTED.toString()); + params.add("states", YarnApplicationState.ACCEPTED.toString()); ClientResponse response = r.path("ws").path("v1").path("cluster") .path("apps").queryParams(params) .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); @@ -268,8 +270,8 @@ public void testAppsQueryStates() throws JSONException, Exception { r = resource(); params = new MultivaluedMapImpl(); - params.add("states", RMAppState.ACCEPTED.toString()); - params.add("states", RMAppState.KILLED.toString()); + params.add("states", YarnApplicationState.ACCEPTED.toString()); + params.add("states", YarnApplicationState.KILLED.toString()); response = r.path("ws").path("v1").path("cluster") .path("apps").queryParams(params) .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); @@ -301,7 +303,7 @@ public void testAppsQueryStatesComma() throws JSONException, Exception { WebResource r = resource(); MultivaluedMapImpl params = new MultivaluedMapImpl(); - params.add("states", RMAppState.ACCEPTED.toString()); + params.add("states", YarnApplicationState.ACCEPTED.toString()); ClientResponse response = r.path("ws").path("v1").path("cluster") .path("apps").queryParams(params) .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); @@ -317,8 +319,8 @@ public void testAppsQueryStatesComma() throws JSONException, Exception { r = resource(); params = new MultivaluedMapImpl(); - params.add("states", RMAppState.ACCEPTED.toString() + "," - + RMAppState.KILLED.toString()); + params.add("states", YarnApplicationState.ACCEPTED.toString() + "," + + YarnApplicationState.KILLED.toString()); response = r.path("ws").path("v1").path("cluster") .path("apps").queryParams(params) .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); @@ -347,7 +349,8 @@ public void testAppsQueryStatesNone() throws JSONException, Exception { WebResource r = resource(); ClientResponse response = r.path("ws").path("v1").path("cluster") - .path("apps").queryParam("states", RMAppState.RUNNING.toString()) + .path("apps") + .queryParam("states", YarnApplicationState.RUNNING.toString()) .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); JSONObject json = response.getEntity(JSONObject.class); @@ -365,7 +368,8 @@ public void testAppsQueryStateNone() throws JSONException, Exception { WebResource r = resource(); ClientResponse response = r.path("ws").path("v1").path("cluster") - .path("apps").queryParam("state", RMAppState.RUNNING.toString()) + .path("apps") + .queryParam("state", YarnApplicationState.RUNNING.toString()) .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); JSONObject json = response.getEntity(JSONObject.class); @@ -975,6 +979,169 @@ public void testAppsQueryAppTypes() throws JSONException, Exception { rm.stop(); } + @Test + public void testAppStatistics() throws JSONException, Exception { + try { + rm.start(); + MockNM amNodeManager = rm.registerNode("127.0.0.1:1234", 4096); + Thread.sleep(1); + RMApp app1 = rm.submitApp(1024, "", UserGroupInformation.getCurrentUser() + .getShortUserName(), null, false, null, 2, null, "MAPREDUCE"); + amNodeManager.nodeHeartbeat(true); + // finish App + MockAM am = rm + .sendAMLaunched(app1.getCurrentAppAttempt().getAppAttemptId()); + am.registerAppAttempt(); + am.unregisterAppAttempt(); + amNodeManager.nodeHeartbeat(app1.getCurrentAppAttempt().getAppAttemptId(), + 1, ContainerState.COMPLETE); + + rm.submitApp(1024, "", UserGroupInformation.getCurrentUser() + .getShortUserName(), null, false, null, 2, null, "MAPREDUCE"); + rm.submitApp(1024, "", UserGroupInformation.getCurrentUser() + .getShortUserName(), null, false, null, 2, null, "OTHER"); + + // zero type, zero state + WebResource r = resource(); + ClientResponse response = r.path("ws").path("v1").path("cluster") + .path("appstatistics") + .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); + assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); + JSONObject json = response.getEntity(JSONObject.class); + assertEquals("incorrect number of elements", 1, json.length()); + JSONObject appsStatInfo = json.getJSONObject("appStatInfo"); + assertEquals("incorrect number of elements", 1, appsStatInfo.length()); + JSONArray statItems = appsStatInfo.getJSONArray("statItem"); + assertEquals("incorrect number of elements", + YarnApplicationState.values().length, statItems.length()); + for (int i = 0; i < YarnApplicationState.values().length; ++i) { + assertEquals("*", statItems.getJSONObject(0).getString("type")); + if (statItems.getJSONObject(0).getString("state").equals("ACCEPTED")) { + assertEquals("2", statItems.getJSONObject(0).getString("count")); + } else if ( + statItems.getJSONObject(0).getString("state").equals("FINISHED")) { + assertEquals("1", statItems.getJSONObject(0).getString("count")); + } else { + assertEquals("0", statItems.getJSONObject(0).getString("count")); + } + } + + // zero type, one state + r = resource(); + response = r.path("ws").path("v1").path("cluster") + .path("appstatistics") + .queryParam("states", YarnApplicationState.ACCEPTED.toString()) + .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); + assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); + json = response.getEntity(JSONObject.class); + assertEquals("incorrect number of elements", 1, json.length()); + appsStatInfo = json.getJSONObject("appStatInfo"); + assertEquals("incorrect number of elements", 1, appsStatInfo.length()); + statItems = appsStatInfo.getJSONArray("statItem"); + assertEquals("incorrect number of elements", 1, statItems.length()); + assertEquals("ACCEPTED", statItems.getJSONObject(0).getString("state")); + assertEquals("*", statItems.getJSONObject(0).getString("type")); + assertEquals("2", statItems.getJSONObject(0).getString("count")); + + // one type, zero state + r = resource(); + response = r.path("ws").path("v1").path("cluster") + .path("appstatistics") + .queryParam("applicationTypes", "MAPREDUCE") + .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); + assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); + json = response.getEntity(JSONObject.class); + assertEquals("incorrect number of elements", 1, json.length()); + appsStatInfo = json.getJSONObject("appStatInfo"); + assertEquals("incorrect number of elements", 1, appsStatInfo.length()); + statItems = appsStatInfo.getJSONArray("statItem"); + assertEquals("incorrect number of elements", + YarnApplicationState.values().length, statItems.length()); + for (int i = 0; i < YarnApplicationState.values().length; ++i) { + assertEquals("mapreduce", statItems.getJSONObject(0).getString("type")); + if (statItems.getJSONObject(0).getString("state").equals("ACCEPTED")) { + assertEquals("1", statItems.getJSONObject(0).getString("count")); + } else if ( + statItems.getJSONObject(0).getString("state").equals("FINISHED")) { + assertEquals("1", statItems.getJSONObject(0).getString("count")); + } else { + assertEquals("0", statItems.getJSONObject(0).getString("count")); + } + } + + // two types, zero state + r = resource(); + response = r.path("ws").path("v1").path("cluster") + .path("appstatistics") + .queryParam("applicationTypes", "MAPREDUCE,OTHER") + .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); + assertEquals(Status.BAD_REQUEST, response.getClientResponseStatus()); + assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); + json = response.getEntity(JSONObject.class); + assertEquals("incorrect number of elements", 1, json.length()); + JSONObject exception = json.getJSONObject("RemoteException"); + assertEquals("incorrect number of elements", 3, exception.length()); + String message = exception.getString("message"); + String type = exception.getString("exception"); + String className = exception.getString("javaClassName"); + WebServicesTestUtils.checkStringContains("exception message", + "we temporarily support at most one applicationType", message); + WebServicesTestUtils.checkStringEqual("exception type", + "BadRequestException", type); + WebServicesTestUtils.checkStringEqual("exception className", + "org.apache.hadoop.yarn.webapp.BadRequestException", className); + + // one type, two states + r = resource(); + response = r.path("ws").path("v1").path("cluster") + .path("appstatistics") + .queryParam("states", YarnApplicationState.FINISHED.toString() + + "," + YarnApplicationState.ACCEPTED.toString()) + .queryParam("applicationTypes", "MAPREDUCE") + .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); + assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); + json = response.getEntity(JSONObject.class); + assertEquals("incorrect number of elements", 1, json.length()); + appsStatInfo = json.getJSONObject("appStatInfo"); + assertEquals("incorrect number of elements", 1, appsStatInfo.length()); + statItems = appsStatInfo.getJSONArray("statItem"); + assertEquals("incorrect number of elements", 2, statItems.length()); + JSONObject statItem1 = statItems.getJSONObject(0); + JSONObject statItem2 = statItems.getJSONObject(1); + assertTrue((statItem1.getString("state").equals("ACCEPTED") && + statItem2.getString("state").equals("FINISHED")) || + (statItem2.getString("state").equals("ACCEPTED") && + statItem1.getString("state").equals("FINISHED"))); + assertEquals("mapreduce", statItem1.getString("type")); + assertEquals("1", statItem1.getString("count")); + assertEquals("mapreduce", statItem2.getString("type")); + assertEquals("1", statItem2.getString("count")); + + // invalid state + r = resource(); + response = r.path("ws").path("v1").path("cluster") + .path("appstatistics").queryParam("states", "wrong_state") + .accept(MediaType.APPLICATION_JSON).get(ClientResponse.class); + assertEquals(Status.BAD_REQUEST, response.getClientResponseStatus()); + assertEquals(MediaType.APPLICATION_JSON_TYPE, response.getType()); + json = response.getEntity(JSONObject.class); + assertEquals("incorrect number of elements", 1, json.length()); + exception = json.getJSONObject("RemoteException"); + assertEquals("incorrect number of elements", 3, exception.length()); + message = exception.getString("message"); + type = exception.getString("exception"); + className = exception.getString("javaClassName"); + WebServicesTestUtils.checkStringContains("exception message", + "Invalid application-state wrong_state", message); + WebServicesTestUtils.checkStringEqual("exception type", + "BadRequestException", type); + WebServicesTestUtils.checkStringEqual("exception className", + "org.apache.hadoop.yarn.webapp.BadRequestException", className); + } finally { + rm.stop(); + } + } + @Test public void testSingleApp() throws JSONException, Exception { rm.start(); diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-site/src/site/apt/ResourceManagerRest.apt.vm b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-site/src/site/apt/ResourceManagerRest.apt.vm index 22011124f2..b3eb94f21b 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-site/src/site/apt/ResourceManagerRest.apt.vm +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-site/src/site/apt/ResourceManagerRest.apt.vm @@ -1269,6 +1269,129 @@ _01_000001 +---+ +* Cluster Application Statistics API + + With the Application Statistics API, you can obtain a collection of triples, each of which contains the application type, the application state and the number of applications of this type and this state in ResourceManager context. Note that with the performance concern, we currently only support at most one applicationType per query. We may support multiple applicationTypes per query as well as more statistics in the future. When you run a GET operation on this resource, you obtain a collection of statItem objects. + +** URI + +------ + * http:///ws/v1/cluster/appstatistics +------ + +** HTTP Operations Supported + +------ + * GET +------ + +** Query Parameters Required + + Two paramters can be specified. The parameters are case insensitive. + +------ + * states - states of the applications, specified as a comma-separated list. If states is not provided, the API will enumerate all application states and return the counts of them. + * applicationTypes - types of the applications, specified as a comma-separated list. If applicationTypes is not provided, the API will count the applications of any application type. In this case, the response shows * to indicate any application type. Note that we only support at most one applicationType temporarily. Otherwise, users will expect an BadRequestException. +------ + +** Elements of the (statItems) object + + When you make a request for the list of statistics items, the information will be returned as a collection of statItem objects + +*-----------+----------------------------------------------------------------------+-------------------------------------+ +|| Item || Data Type || Description | +*-----------+----------------------------------------------------------------------+-------------------------------------+ +| statItem | array of statItem objects(JSON)/zero or more statItem objects(XML) | The collection of statItem objects | +*-----------+----------------------------------------------------------------------+-------------------------------------+ + +** Response Examples + + <> + + HTTP Request: + +------ + GET http:///ws/v1/cluster/appstatistics?states=accepted,running,finished&applicationTypes=mapreduce +------ + + Response Header: + ++---+ + HTTP/1.1 200 OK + Content-Type: application/json + Transfer-Encoding: chunked + Server: Jetty(6.1.26) ++---+ + + Response Body: + ++---+ +{ + "appStatInfo": + { + "statItem": + [ + { + "state" : "accepted", + "type" : "mapreduce", + "count" : 4 + }, + { + "state" : "running", + "type" : "mapreduce", + "count" : 1 + }, + { + "state" : "finished", + "type" : "mapreduce", + "count" : 7 + } + ] + } +} ++---+ + + <> + + HTTP Request: + +------ + GET http:///ws/v1/cluster/appstatistics?states=accepted,running,finished&applicationTypes=mapreduce + Accept: application/xml +------ + + Response Header: + ++---+ + HTTP/1.1 200 OK + Content-Type: application/xml + Content-Length: 2459 + Server: Jetty(6.1.26) ++---+ + + Response Body: + ++---+ + + + + accepted + mapreduce + 4 + + + running + mapreduce + 1 + + + finished + mapreduce + 7 + + ++---+ + * Cluster {Application API} An application resource contains information about a particular application that was submitted to a cluster.