invokeSingle

Author: KeeProMise

hadoop-hdfs-project/hadoop-hdfs-rbf/src/test/java/org/apache/hadoop/hdfs/server/federation/router/TestRouterAsyncRpcClient.java

@@ -59,6 +59,9 @@
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
 
+/**
+ * Used to test the functionality of {@link RouterAsyncRpcClient}.
+ */
 public class TestRouterAsyncRpcClient {
   private static Configuration routerConf;
   /** Federated HDFS cluster. */
@@ -72,8 +75,12 @@ public class TestRouterAsyncRpcClient {
   private RouterRpcServer routerRpcServer;
   private RouterAsyncRpcClient asyncRpcClient;
   private FederationRPCMetrics rpcMetrics;
+  private final String testFile = "/test.file";
 
-
+  /**
+   * Start a cluster using a router service that includes 2 namespaces,
+   * 6 namenodes and 6 datanodes.
+   */
   @BeforeClass
   public static void setUpCluster() throws Exception {
     cluster = new MiniRouterDFSCluster(true, 2, 3,
@@ -116,12 +123,15 @@ public static void setUpCluster() throws Exception {
   }
 
   @AfterClass
-  public static void shutdownCluster() throws Exception {
+  public static void shutdownCluster() {
     if (cluster != null) {
       cluster.shutdown();
     }
   }
 
+  /**
+   * Initialize the mount table, create a RouterAsyncRpcClient object, and create test file.
+   */
   @Before
   public void setup() throws Exception {
     // Create mock locations
@@ -132,13 +142,16 @@ public void setup() throws Exception {
     routerFs = router.getFileSystem();
     routerRpcServer = router.getRouterRpcServer();
     routerRpcServer.initAsyncThreadPool();
+
+    // Create a RouterAsyncRpcClient object
     asyncRpcClient = new RouterAsyncRpcClient(
         routerConf, router.getRouter(), routerRpcServer.getNamenodeResolver(),
         routerRpcServer.getRPCMonitor(),
         routerRpcServer.getRouterStateIdContext());
 
+    // Create a test file
     FSDataOutputStream fsDataOutputStream = routerFs.create(
-        new Path("/test.file"), true);
+        new Path(testFile), true);
     fsDataOutputStream.write(new byte[1024]);
     fsDataOutputStream.close();
   }
@@ -151,13 +164,17 @@ public void down() throws IOException {
     asyncRpcClient.getNamenodeResolver().updateActiveNamenode(
         ns0, NetUtils.createSocketAddr(cluster
             .getNamenode(ns0, NAMENODES[0]).getRpcAddress()));
-    boolean delete = routerFs.delete(new Path("/test.file"));
+    // Delete the test file
+    boolean delete = routerFs.delete(new Path(testFile));
     assertTrue(delete);
     if (routerFs != null) {
       routerFs.close();
     }
   }
 
+  /**
+   * Test the functionality of the asynchronous invokeSingle method.
+   */
   @Test
   public void testInvokeSingle() throws Exception {
     long proxyOps = rpcMetrics.getProxyOps();
@@ -173,6 +190,9 @@ public void testInvokeSingle() throws Exception {
     assertTrue(rpcMetrics.getProxyAvg() > 0);
   }
 
+  /**
+   * Test the functionality of the asynchronous invokeAll and invokeConcurrent methods.
+   */
   @Test
   public void testInvokeAll() throws Exception {
     long proxyOps = rpcMetrics.getProxyOps();
@@ -206,6 +226,9 @@ public void testInvokeAll() throws Exception {
     assertTrue(rpcMetrics.getProxyAvg() > 0);
   }
 
+  /**
+   * Test the functionality of the asynchronous invokeMethod method.
+   */
   @Test
   public void testInvokeMethod() throws Exception {
     long proxyOps = rpcMetrics.getProxyOps();
@@ -214,7 +237,7 @@ public void testInvokeMethod() throws Exception {
         new Class<?>[] {String.class}, new RemoteParam());
     UserGroupInformation ugi = RouterRpcServer.getRemoteUser();
     Class<?> protocol = method.getProtocol();
-    Object[] params = new String[]{"/test.file"};
+    Object[] params = new String[]{testFile};
     List<? extends FederationNamenodeContext> namenodes =
         asyncRpcClient.getOrderedNamenodes(ns0, false);
     asyncRpcClient.invokeMethod(ugi, namenodes, false,
@@ -257,10 +280,13 @@ public void testInvokeMethod() throws Exception {
     assertEquals(1, rpcMetrics.getProxyOpFailureCommunicate());
   }
 
+  /**
+   * Test the functionality of the asynchronous invokeSequential method.
+   */
   @Test
   public void testInvokeSequential() throws Exception {
     List<RemoteLocation> locations =
-        routerRpcServer.getLocationsForPath("/test.file", false, false);
+        routerRpcServer.getLocationsForPath(testFile, false, false);
     RemoteMethod remoteMethod = new RemoteMethod("getBlockLocations",
         new Class<?>[] {String.class, long.class, long.class},
         new RemoteParam(), 0, 1024);
@@ -271,6 +297,9 @@ public void testInvokeSequential() throws Exception {
     assertEquals(1, locatedBlocks.getLocatedBlocks().size());
   }
 
+  /**
+   * Initialize the mount information.
+   */
   private void installMockLocations() {
     List<MiniRouterDFSCluster.RouterContext> routers = cluster.getRouters();
 
@@ -282,4 +311,4 @@ private void installMockLocations() {
       resolver.addLocation("/multDes", ns1, "/multDes");
     }
   }
-}
+}