diff --git a/hadoop-hdfs-project/hadoop-hdfs-rbf/src/main/java/org/apache/hadoop/hdfs/server/federation/resolver/order/AvailableSpaceResolver.java b/hadoop-hdfs-project/hadoop-hdfs-rbf/src/main/java/org/apache/hadoop/hdfs/server/federation/resolver/order/AvailableSpaceResolver.java index 591ac5b3c37ca..6462a9b22a85e 100644 --- a/hadoop-hdfs-project/hadoop-hdfs-rbf/src/main/java/org/apache/hadoop/hdfs/server/federation/resolver/order/AvailableSpaceResolver.java +++ b/hadoop-hdfs-project/hadoop-hdfs-rbf/src/main/java/org/apache/hadoop/hdfs/server/federation/resolver/order/AvailableSpaceResolver.java @@ -29,6 +29,7 @@ import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.hdfs.server.federation.resolver.PathLocation; +import org.apache.hadoop.hdfs.server.federation.resolver.RemoteLocation; import org.apache.hadoop.hdfs.server.federation.resolver.order.AvailableSpaceResolver.SubclusterAvailableSpace; import org.apache.hadoop.hdfs.server.federation.router.RBFConfigKeys; import org.apache.hadoop.hdfs.server.federation.router.Router; @@ -116,8 +117,10 @@ protected Map getSubclusterInfo( protected String chooseFirstNamespace(String path, PathLocation loc) { Map subclusterInfo = getSubclusterMapping(); - List subclusterList = new LinkedList<>( - subclusterInfo.values()); + List subclusterList = new LinkedList<>(); + for (RemoteLocation dest : loc.getDestinations()) { + subclusterList.add(subclusterInfo.get(dest.getNameserviceId())); + } Collections.sort(subclusterList, comparator); return subclusterList.size() > 0 ? subclusterList.get(0).getNameserviceId() diff --git a/hadoop-hdfs-project/hadoop-hdfs-rbf/src/test/java/org/apache/hadoop/hdfs/server/federation/resolver/order/TestAvailableSpaceResolver.java b/hadoop-hdfs-project/hadoop-hdfs-rbf/src/test/java/org/apache/hadoop/hdfs/server/federation/resolver/order/TestAvailableSpaceResolver.java index 3d6b61aca0e35..84bf57f2b0a10 100644 --- a/hadoop-hdfs-project/hadoop-hdfs-rbf/src/test/java/org/apache/hadoop/hdfs/server/federation/resolver/order/TestAvailableSpaceResolver.java +++ b/hadoop-hdfs-project/hadoop-hdfs-rbf/src/test/java/org/apache/hadoop/hdfs/server/federation/resolver/order/TestAvailableSpaceResolver.java @@ -229,4 +229,12 @@ private void verifyRank(float balancerPreference, boolean shouldOrdered, } subclusters.clear(); } + + @Test + public void testChooseFirstNamespace() throws Exception { + MultipleDestinationMountTableResolver mountTableResolver = + mockAvailableSpaceResolver(1.0f); + PathLocation loc = mountTableResolver.getDestinationForPath("/space"); + assertEquals("subcluster9", loc.getDefaultLocation().getNameserviceId()); + } }