[Fix][Zeta] improve the local mode hazelcast connection (#6521)

apache · Mar 17, 2024 · cb3db75 · cb3db75
1 parent c0487a9
commit cb3db75
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/...c/main/java/org/apache/seatunnel/core/starter/seatunnel/command/ClientExecuteCommand.java b/...c/main/java/org/apache/seatunnel/core/starter/seatunnel/command/ClientExecuteCommand.java
@@ -48,6 +48,7 @@
 import java.nio.file.Path;
 import java.time.Duration;
 import java.time.LocalDateTime;
+import java.util.Collections;
 import java.util.Random;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.Executors;
@@ -79,19 +80,21 @@ public void execute() throws CommandExecuteException {
         SeaTunnelConfig seaTunnelConfig = ConfigProvider.locateAndGetSeaTunnelConfig();
         try {
             String clusterName = clientCommandArgs.getClusterName();
+            ClientConfig clientConfig = ConfigProvider.locateAndGetClientConfig();
             if (clientCommandArgs.getMasterType().equals(MasterType.LOCAL)) {
                 clusterName =
                         creatRandomClusterName(
                                 StringUtils.isNotEmpty(clusterName)
                                         ? clusterName
                                         : Constant.DEFAULT_SEATUNNEL_CLUSTER_NAME);
                 instance = createServerInLocal(clusterName, seaTunnelConfig);
+                int port = instance.getCluster().getLocalMember().getSocketAddress().getPort();
+                clientConfig
+                        .getNetworkConfig()
+                        .setAddresses(Collections.singletonList("localhost:" + port));
             }
             if (StringUtils.isNotEmpty(clusterName)) {
                 seaTunnelConfig.getHazelcastConfig().setClusterName(clusterName);
-            }
-            ClientConfig clientConfig = ConfigProvider.locateAndGetClientConfig();
-            if (StringUtils.isNotEmpty(clusterName)) {
                 clientConfig.setClusterName(clusterName);
             }
             engineClient = new SeaTunnelClient(clientConfig);