This is an automated email from the ASF dual-hosted git repository.

vinoth pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-hudi.git

commit 80c27f23510e9c4232ee87e06475d8af4823bb38
Author: yanghua <yanghua1...@gmail.com>
AuthorDate: Tue Sep 3 19:00:37 2019 +0800

    Optimize hoodie client after implementat auto closeable interface
---
 .../java/org/apache/hudi/TestHoodieClientBase.java | 25 ++--------------------
 .../src/test/java/org/apache/hudi/TestMultiFS.java |  7 +-----
 .../java/org/apache/hudi/index/TestHbaseIndex.java | 12 +----------
 .../apache/hudi/io/TestHoodieCommitArchiveLog.java |  2 --
 .../org/apache/hudi/io/TestHoodieCompactor.java    |  7 +-----
 .../org/apache/hudi/io/TestHoodieMergeHandle.java  |  8 +------
 .../apache/hudi/table/TestMergeOnReadTable.java    |  8 +------
 7 files changed, 7 insertions(+), 62 deletions(-)

diff --git 
a/hudi-client/src/test/java/org/apache/hudi/TestHoodieClientBase.java 
b/hudi-client/src/test/java/org/apache/hudi/TestHoodieClientBase.java
index 6d1adcd..dca55d1 100644
--- a/hudi-client/src/test/java/org/apache/hudi/TestHoodieClientBase.java
+++ b/hudi-client/src/test/java/org/apache/hudi/TestHoodieClientBase.java
@@ -65,9 +65,6 @@ public class TestHoodieClientBase extends 
HoodieClientTestHarness {
 
   protected static Logger logger = 
LogManager.getLogger(TestHoodieClientBase.class);
 
-  private HoodieWriteClient writeClient;
-  private HoodieReadClient readClient;
-
   protected HoodieWriteClient getHoodieWriteClient(HoodieWriteConfig cfg) {
     return getHoodieWriteClient(cfg, false);
   }
@@ -78,29 +75,11 @@ public class TestHoodieClientBase extends 
HoodieClientTestHarness {
 
   protected HoodieWriteClient getHoodieWriteClient(HoodieWriteConfig cfg, 
boolean rollbackInflightCommit,
       HoodieIndex index) {
-    closeWriteClient();
-    writeClient = new HoodieWriteClient(jsc, cfg, rollbackInflightCommit, 
index);
-    return writeClient;
+    return new HoodieWriteClient(jsc, cfg, rollbackInflightCommit, index);
   }
 
   protected HoodieReadClient getHoodieReadClient(String basePath) {
-    closeReadClient();
-    readClient = new HoodieReadClient(jsc, basePath);
-    return readClient;
-  }
-
-  private void closeWriteClient() {
-    if (null != writeClient) {
-      writeClient.close();
-      writeClient = null;
-    }
-  }
-
-  private void closeReadClient() {
-    if (null != readClient) {
-      readClient.close();
-      readClient = null;
-    }
+    return new HoodieReadClient(jsc, basePath);
   }
 
   /**
diff --git a/hudi-client/src/test/java/org/apache/hudi/TestMultiFS.java 
b/hudi-client/src/test/java/org/apache/hudi/TestMultiFS.java
index 14f4b72..08d0a03 100644
--- a/hudi-client/src/test/java/org/apache/hudi/TestMultiFS.java
+++ b/hudi-client/src/test/java/org/apache/hudi/TestMultiFS.java
@@ -49,7 +49,6 @@ public class TestMultiFS extends HoodieClientTestHarness {
   private static Logger logger = LogManager.getLogger(TestMultiFS.class);
   private String tablePath = "file:///tmp/hoodie/sample-table";
   protected String tableName = "hoodie_rt";
-  private HoodieWriteClient hdfsWriteClient;
   private String tableType = HoodieTableType.COPY_ON_WRITE.name();
 
   @Before
@@ -67,11 +66,7 @@ public class TestMultiFS extends HoodieClientTestHarness {
   }
 
   private HoodieWriteClient getHoodieWriteClient(HoodieWriteConfig config) 
throws Exception {
-    if (null != hdfsWriteClient) {
-      hdfsWriteClient.close();
-    }
-    hdfsWriteClient = new HoodieWriteClient(jsc, config);
-    return hdfsWriteClient;
+    return new HoodieWriteClient(jsc, config);
   }
 
   protected HoodieWriteConfig getHoodieWriteConfig(String basePath) {
diff --git 
a/hudi-client/src/test/java/org/apache/hudi/index/TestHbaseIndex.java 
b/hudi-client/src/test/java/org/apache/hudi/index/TestHbaseIndex.java
index 02f94cd..9d89186 100644
--- a/hudi-client/src/test/java/org/apache/hudi/index/TestHbaseIndex.java
+++ b/hudi-client/src/test/java/org/apache/hudi/index/TestHbaseIndex.java
@@ -76,7 +76,6 @@ public class TestHbaseIndex extends HoodieClientTestHarness {
   private static HBaseTestingUtility utility;
   private static Configuration hbaseConfig;
   private static String tableName = "test_table";
-  private HoodieWriteClient writeClient;
 
   public TestHbaseIndex() throws Exception {
   }
@@ -112,22 +111,13 @@ public class TestHbaseIndex extends 
HoodieClientTestHarness {
 
   @After
   public void tearDown() throws Exception {
-    if (null != writeClient) {
-      writeClient.close();
-      writeClient = null;
-    }
-
     cleanupSparkContexts();
     cleanupTempFolderAndPath();
     cleanupTestDataGenerator();
   }
 
   private HoodieWriteClient getWriteClient(HoodieWriteConfig config) throws 
Exception {
-    if (null != writeClient) {
-      writeClient.close();
-    }
-    writeClient = new HoodieWriteClient(jsc, config);
-    return writeClient;
+    return new HoodieWriteClient(jsc, config);
   }
 
   @Test
diff --git 
a/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCommitArchiveLog.java 
b/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCommitArchiveLog.java
index 9f5c268..5dee6f6 100644
--- 
a/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCommitArchiveLog.java
+++ 
b/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCommitArchiveLog.java
@@ -58,7 +58,6 @@ public class TestHoodieCommitArchiveLog extends 
HoodieClientTestHarness {
   @Before
   public void init() throws Exception {
     initDFS();
-
     initTempFolderAndPath();
     initSparkContexts("TestHoodieCommitArchiveLog");
     hadoopConf = dfs.getConf();
@@ -70,7 +69,6 @@ public class TestHoodieCommitArchiveLog extends 
HoodieClientTestHarness {
   @After
   public void clean() throws IOException {
     cleanupDFS();
-
     cleanupTempFolderAndPath();
     cleanupSparkContexts();
   }
diff --git 
a/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCompactor.java 
b/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCompactor.java
index d685242..ac1fbad 100644
--- a/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCompactor.java
+++ b/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieCompactor.java
@@ -51,7 +51,6 @@ import org.junit.Test;
 
 public class TestHoodieCompactor extends HoodieClientTestHarness {
 
-  private HoodieWriteClient writeClient;
   private Configuration hadoopConf;
 
   @Before
@@ -76,11 +75,7 @@ public class TestHoodieCompactor extends 
HoodieClientTestHarness {
   }
 
   private HoodieWriteClient getWriteClient(HoodieWriteConfig config) throws 
Exception {
-    if (null != writeClient) {
-      writeClient.close();
-    }
-    writeClient = new HoodieWriteClient(jsc, config);
-    return writeClient;
+    return new HoodieWriteClient(jsc, config);
   }
 
   private HoodieWriteConfig getConfig() {
diff --git 
a/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieMergeHandle.java 
b/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieMergeHandle.java
index 76e0d1c..288e04c 100644
--- a/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieMergeHandle.java
+++ b/hudi-client/src/test/java/org/apache/hudi/io/TestHoodieMergeHandle.java
@@ -53,8 +53,6 @@ import org.junit.Test;
 @SuppressWarnings("unchecked")
 public class TestHoodieMergeHandle extends HoodieClientTestHarness {
 
-  private HoodieWriteClient writeClient;
-
   @Before
   public void setUp() throws Exception {
     initSparkContexts("TestHoodieMergeHandle");
@@ -74,11 +72,7 @@ public class TestHoodieMergeHandle extends 
HoodieClientTestHarness {
   }
 
   private HoodieWriteClient getWriteClient(HoodieWriteConfig config) throws 
Exception {
-    if (null != writeClient) {
-      writeClient.close();
-    }
-    writeClient = new HoodieWriteClient(jsc, config);
-    return writeClient;
+    return new HoodieWriteClient(jsc, config);
   }
 
   @Test
diff --git 
a/hudi-client/src/test/java/org/apache/hudi/table/TestMergeOnReadTable.java 
b/hudi-client/src/test/java/org/apache/hudi/table/TestMergeOnReadTable.java
index ce13fb0..2b80125 100644
--- a/hudi-client/src/test/java/org/apache/hudi/table/TestMergeOnReadTable.java
+++ b/hudi-client/src/test/java/org/apache/hudi/table/TestMergeOnReadTable.java
@@ -78,8 +78,6 @@ import org.junit.rules.TemporaryFolder;
 
 public class TestMergeOnReadTable extends HoodieClientTestHarness {
 
-  private HoodieWriteClient writeClient;
-
   @Before
   public void init() throws IOException {
     initDFS();
@@ -100,11 +98,7 @@ public class TestMergeOnReadTable extends 
HoodieClientTestHarness {
   }
 
   private HoodieWriteClient getWriteClient(HoodieWriteConfig config) throws 
Exception {
-    if (null != writeClient) {
-      writeClient.close();
-    }
-    writeClient = new HoodieWriteClient(jsc, config);
-    return writeClient;
+    return new HoodieWriteClient(jsc, config);
   }
 
   @Test

Reply via email to