feat: support in synctools

4paradigm · Nov 7, 2023 · d627094 · d627094
1 parent ff249f9
commit d627094
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/java/openmldb-synctool/src/main/java/com/_4paradigm/openmldb/synctool/SyncToolConfig.java b/java/openmldb-synctool/src/main/java/com/_4paradigm/openmldb/synctool/SyncToolConfig.java
@@ -37,6 +37,7 @@ public class SyncToolConfig {
     // public static int CHANNEL_KEEP_ALIVE_TIME;
     public static String ZK_CLUSTER;
     public static String ZK_ROOT_PATH;
+    public static String ZK_CERT;
     public static String SYNC_TASK_PROGRESS_PATH;
 
     public static String HADOOP_CONF_DIR;
@@ -86,6 +87,7 @@ private static void parseFromProperties(Properties prop) {
         if (ZK_ROOT_PATH.isEmpty()) {
             throw new RuntimeException("zookeeper.root_path should not be empty");
         }
+        ZK_CERT = prop.getProperty("zookeeper.cert", "");
 
         HADOOP_CONF_DIR = prop.getProperty("hadoop.conf.dir", "");
         if (HADOOP_CONF_DIR.isEmpty()) {

diff --git a/java/openmldb-synctool/src/main/java/com/_4paradigm/openmldb/synctool/SyncToolImpl.java b/java/openmldb-synctool/src/main/java/com/_4paradigm/openmldb/synctool/SyncToolImpl.java
@@ -85,11 +85,13 @@ public SyncToolImpl(String endpoint) throws SqlException, InterruptedException {
         this.zkClient = new ZKClient(ZKConfig.builder()
                 .cluster(SyncToolConfig.ZK_CLUSTER)
                 .namespace(SyncToolConfig.ZK_ROOT_PATH)
+                .cert(SyncToolConfig.ZK_CERT)
                 .build());
         Preconditions.checkState(zkClient.connect(), "zk connect failed");
         SdkOption option = new SdkOption();
         option.setZkCluster(SyncToolConfig.ZK_CLUSTER);
         option.setZkPath(SyncToolConfig.ZK_ROOT_PATH);
+        option.setZkCert(SyncToolConfig.ZK_CERT);
         this.router = new SqlClusterExecutor(option);
         this.zkCollectorPath = SyncToolConfig.ZK_ROOT_PATH + "/sync_tool/collector";