(iotdb) branch master updated: [AINode] Add cluster_ingress_port for AINodeWrapper (#16043)

[yongzao]

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

yongzao pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/iotdb.git


The following commit(s) were added to refs/heads/master by this push:
     new ac8642fa22f [AINode] Add cluster_ingress_port for AINodeWrapper 
(#16043)
ac8642fa22f is described below

commit ac8642fa22f3923e5fdc9d1aa572fad0a96bd114
Author: Yongzao <yong...@apache.org>
AuthorDate: Mon Jul 28 12:33:08 2025 +0800

    [AINode] Add cluster_ingress_port for AINodeWrapper (#16043)
---
 .../java/org/apache/iotdb/it/env/cluster/env/AbstractEnv.java |  6 ++++--
 .../org/apache/iotdb/it/env/cluster/node/AINodeWrapper.java   | 11 +++++++++--
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git 
a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/env/AbstractEnv.java
 
b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/env/AbstractEnv.java
index e68a6d550c2..625280358e0 100644
--- 
a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/env/AbstractEnv.java
+++ 
b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/env/AbstractEnv.java
@@ -258,7 +258,7 @@ public abstract class AbstractEnv implements BaseEnv {
 
     if (addAINode) {
       this.aiNodeWrapperList = new ArrayList<>();
-      startAINode(seedConfigNode, testClassName);
+      startAINode(seedConfigNode, this.dataNodeWrapperList.get(0).getPort(), 
testClassName);
     }
 
     checkClusterStatusWithoutUnknown();
@@ -307,11 +307,13 @@ public abstract class AbstractEnv implements BaseEnv {
     return dataNodeWrapper;
   }
 
-  private void startAINode(final String seedConfigNode, final String 
testClassName) {
+  private void startAINode(
+      final String seedConfigNode, final int clusterIngressPort, final String 
testClassName) {
     final String aiNodeEndPoint;
     final AINodeWrapper aiNodeWrapper =
         new AINodeWrapper(
             seedConfigNode,
+            clusterIngressPort,
             testClassName,
             testMethodName,
             index,
diff --git 
a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/node/AINodeWrapper.java
 
b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/node/AINodeWrapper.java
index c4a85555e76..608b799f7a8 100644
--- 
a/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/node/AINodeWrapper.java
+++ 
b/integration-test/src/main/java/org/apache/iotdb/it/env/cluster/node/AINodeWrapper.java
@@ -43,6 +43,7 @@ public class AINodeWrapper extends AbstractNodeWrapper {
   private static final Logger logger = IoTDBTestLogger.logger;
   private final long startTime;
   private final String seedConfigNode;
+  private final int clusterIngressPort;
 
   private static final String SCRIPT_FILE = "start-ainode.sh";
 
@@ -67,6 +68,7 @@ public class AINodeWrapper extends AbstractNodeWrapper {
 
   public AINodeWrapper(
       String seedConfigNode,
+      int clusterIngressPort,
       String testClassName,
       String testMethodName,
       int clusterIndex,
@@ -74,6 +76,7 @@ public class AINodeWrapper extends AbstractNodeWrapper {
       long startTime) {
     super(testClassName, testMethodName, port, clusterIndex, false, startTime);
     this.seedConfigNode = seedConfigNode;
+    this.clusterIngressPort = clusterIngressPort;
     this.startTime = startTime;
   }
 
@@ -105,8 +108,12 @@ public class AINodeWrapper extends AbstractNodeWrapper {
 
       // set attribute
       replaceAttribute(
-          new String[] {"ain_seed_config_node", "ain_rpc_port"},
-          new String[] {this.seedConfigNode, Integer.toString(getPort())},
+          new String[] {"ain_seed_config_node", "ain_rpc_port", 
"ain_cluster_ingress_port"},
+          new String[] {
+            this.seedConfigNode,
+            Integer.toString(getPort()),
+            Integer.toString(this.clusterIngressPort)
+          },
           propertiesFile);
 
       // start AINode