YARN-10885. Fix CheckStyle.

apache · Aug 9, 2022 · fb42275 · fb42275
1 parent ce5ca51
commit fb42275
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 23 deletions.
diff --git a/.../main/java/org/apache/hadoop/yarn/server/federation/utils/FederationStateStoreFacade.java b/.../main/java/org/apache/hadoop/yarn/server/federation/utils/FederationStateStoreFacade.java
@@ -380,13 +380,9 @@ public SubClusterId getApplicationHomeSubCluster(ApplicationId appId)
       throws YarnException {
     try {
       if (isCachingEnabled()) {
-        Object value =
-            cache.get(buildGetApplicationHomeSubClusterRequest(appId));
-        if (value instanceof SubClusterId) {
-          return (SubClusterId) value;
-        } else {
-          throw new YarnException("Cannot be converted to SubClusterId.");
-        }
+        SubClusterId value = SubClusterId.class.cast(
+            cache.get(buildGetApplicationHomeSubClusterRequest(appId)));
+        return value;
       } else {
         GetApplicationHomeSubClusterResponse response = stateStore.getApplicationHomeSubCluster(
             GetApplicationHomeSubClusterRequest.newInstance(appId));
@@ -531,20 +527,19 @@ public Map<String, SubClusterPolicyConfiguration> invoke(
   private Object buildGetApplicationHomeSubClusterRequest(ApplicationId applicationId) {
     final String cacheKey = buildCacheKey(getClass().getSimpleName(),
         GET_APPLICATION_HOME_SUBCLUSTER_CACHEID, applicationId.toString());
-    CacheRequest<String, SubClusterId> cacheRequest =
-        new CacheRequest<>(
-            cacheKey,
-            input -> {
-              GetApplicationHomeSubClusterResponse response =
-                  stateStore.getApplicationHomeSubCluster(
-                      GetApplicationHomeSubClusterRequest.newInstance(applicationId));
-
-              ApplicationHomeSubCluster applicationHomeSubCluster =
-                  response.getApplicationHomeSubCluster();
-              SubClusterId subClusterId = applicationHomeSubCluster.getHomeSubCluster();
-
-              return subClusterId;
-            });
+    CacheRequest<String, SubClusterId> cacheRequest = new CacheRequest<>(
+        cacheKey,
+        input -> {
+            GetApplicationHomeSubClusterResponse response =
+                stateStore.getApplicationHomeSubCluster(
+                    GetApplicationHomeSubClusterRequest.newInstance(applicationId));
+
+            ApplicationHomeSubCluster applicationHomeSubCluster =
+                response.getApplicationHomeSubCluster();
+            SubClusterId subClusterId = applicationHomeSubCluster.getHomeSubCluster();
+
+            return subClusterId;
+         });
     return cacheRequest;
   }
 

diff --git a/...t/java/org/apache/hadoop/yarn/server/federation/utils/TestFederationStateStoreFacade.java b/...t/java/org/apache/hadoop/yarn/server/federation/utils/TestFederationStateStoreFacade.java
@@ -217,8 +217,7 @@ public void testGetApplicationHomeSubClusterCache() throws YarnException {
 
     ApplicationHomeSubCluster appHomeSubCluster =
         ApplicationHomeSubCluster.newInstance(appId, subClusterId1);
-    SubClusterId subClusterIdAdd =
-        facade.addApplicationHomeSubCluster(appHomeSubCluster);
+    SubClusterId subClusterIdAdd = facade.addApplicationHomeSubCluster(appHomeSubCluster);
 
     SubClusterId subClusterIdByFacade = facade.getApplicationHomeSubCluster(appId);
     Assert.assertEquals(subClusterIdByFacade, subClusterIdAdd);