Fixing SessionStrategy's close method

thulab · Dec 27, 2024 · ccff547 · ccff547
1 parent 882c89b
commit ccff547
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/...2.0/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb200/DMLStrategy/SessionStrategy.java b/...2.0/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb200/DMLStrategy/SessionStrategy.java
@@ -453,7 +453,7 @@ Status waitWriteTaskToFinishAndGetStatus() {
 
   @Override
   public void init() {
-    sessionManager.init();
+    sessionManager.open();
     this.service = Executors.newSingleThreadExecutor();
   }
 
@@ -470,9 +470,12 @@ public void cleanup() {
 
   @Override
   public void close() throws TsdbException {
-    if (sessionManager != null) {
-      sessionManager.close();
+    try {
+      if (sessionManager != null) {
+        sessionManager.close();
+      }
+    } finally {
+      service.shutdown();
     }
-    service.shutdown();
   }
 }