support nodeurls in session mode

thulab · Oct 18, 2023 · 80e1b98 · 80e1b98
1 parent 3d60fe0
commit 80e1b98
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/iotdb-1.1/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb110/IoTDBSession.java b/iotdb-1.1/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb110/IoTDBSession.java
@@ -32,6 +32,7 @@
 import cn.edu.tsinghua.iot.benchmark.tsdb.TsdbException;
 import org.slf4j.LoggerFactory;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.Executors;
 
@@ -159,11 +160,14 @@ public SessionDataSet.DataIterator iterator() {
   public IoTDBSession(DBConfig dbConfig) {
     super(dbConfig);
     LOGGER = LoggerFactory.getLogger(IoTDBSession.class);
+    List<String> hostUrls = new ArrayList<>(dbConfig.getHOST().size());
+    for (int i = 0; i < dbConfig.getHOST().size(); i++) {
+      hostUrls.add(dbConfig.getHOST().get(i) + ":" + dbConfig.getPORT().get(i));
+    }
     sessionWrapper =
         new BenchmarkSession(
             new Session.Builder()
-                .host(dbConfig.getHOST().get(0))
-                .port(Integer.parseInt(dbConfig.getPORT().get(0)))
+                .nodeUrls(hostUrls)
                 .username(dbConfig.getUSERNAME())
                 .password(dbConfig.getPASSWORD())
                 .enableRedirection(true)