slfan1989 commented on code in PR #4540:
URL: https://github.com/apache/hadoop/pull/4540#discussion_r921597065
##########
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-router/src/main/java/org/apache/hadoop/yarn/server/router/clientrm/RouterYarnClientUtils.java:
##########
@@ -417,5 +420,59 @@ public static GetQueueInfoResponse mergeQueues(
queueResponse.setQueueInfo(queueInfo);
return queueResponse;
}
+
+ /**
+ * Merges a list of GetAllResourceProfilesResponse.
+ *
+ * @param responses a list of GetAllResourceProfilesResponse to merge.
+ * @return the merged GetAllResourceProfilesResponse.
+ */
+ public static GetAllResourceProfilesResponse
mergeClusterResourceProfilesResponse(
+ Collection<GetAllResourceProfilesResponse> responses) {
+ GetAllResourceProfilesResponse profilesResponse =
+ Records.newRecord(GetAllResourceProfilesResponse.class);
+ Map<String, Resource> profilesMap = new HashMap<>();
+ for (GetAllResourceProfilesResponse response : responses) {
+ if (response != null && response.getResourceProfiles() != null) {
+ for (Map.Entry<String, Resource> entry :
response.getResourceProfiles().entrySet()) {
+ String key = entry.getKey();
+ Resource value = entry.getValue();
+ if (profilesMap.containsKey(key)) {
+ Resource resourceValue = profilesMap.get(key);
+ resourceValue.setVirtualCores(
+ resourceValue.getVirtualCores() + value.getVirtualCores());
+ resourceValue.setMemorySize(resourceValue.getMemorySize() +
value.getMemorySize());
+ profilesMap.put(key, resourceValue);
+ } else {
+ profilesMap.put(key, value);
+ }
+ }
+ }
+ }
+ profilesResponse.setResourceProfiles(profilesMap);
+ return profilesResponse;
+ }
+
+ /**
+ * Merges a list of GetResourceProfileResponse.
+ *
+ * @param responses a list of GetResourceProfileResponse to merge.
+ * @return the merged GetResourceProfileResponse.
+ */
+ public static GetResourceProfileResponse mergeClusterResourceProfileResponse(
+ Collection<GetResourceProfileResponse> responses) {
+ GetResourceProfileResponse profileResponse =
+ Records.newRecord(GetResourceProfileResponse.class);
+ Resource resource = Resource.newInstance(0, 0);
+ for (GetResourceProfileResponse response : responses) {
+ if (response != null && response.getResource() != null) {
+ resource.setMemorySize(resource.getMemorySize() +
response.getResource().getMemorySize());
Review Comment:
OK, i will fix it.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: common-issues-unsubscr...@hadoop.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: common-issues-unsubscr...@hadoop.apache.org
For additional commands, e-mail: common-issues-h...@hadoop.apache.org
