[CELEBORN-1263] Fix Master HA mode without internal port error

### What changes were proposed in this pull request? Master can not start up with ha mode ### Why are the changes needed? ![image](https://github.com/apache/incubator-celeborn/assets/3350718/2856f89e-70bf-4e8e-b05d-5c37cfc6d6b8) ### Does this PR introduce _any_ user-facing change? no ### How was this patch tested? local tested Closes apache#2283 from wxplovecc/fix-ha-internal-port. Lead-authored-by: 吴祥平 <[email protected]> Co-authored-by: 吴祥平 <[email protected]> Signed-off-by: Cheng Pan <[email protected]>
SteNicholas · Feb 4, 2024 · a8a6e1f · a8a6e1f
1 parent fe09e46
commit a8a6e1f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/master/src/main/scala/org/apache/celeborn/service/deploy/master/MasterArguments.scala b/master/src/main/scala/org/apache/celeborn/service/deploy/master/MasterArguments.scala
@@ -44,7 +44,9 @@ class MasterArguments(args: Array[String], conf: CelebornConf) {
     val localNode = clusterInfo.localNode
     _host = _host.orElse(Some(conf.haMasterNodeHost(localNode.nodeId)))
     _port = _port.orElse(Some(conf.haMasterNodePort(localNode.nodeId)))
-    _internalPort = _internalPort.orElse(Some(conf.haMasterNodeInternalPort(localNode.nodeId)))
+    _internalPort = _internalPort.orElse {
+      if (conf.internalPortEnabled) Some(conf.haMasterNodeInternalPort(localNode.nodeId)) else None
+    }
     _masterClusterInfo = Some(clusterInfo)
   } else {
     _host = _host.orElse(Some(conf.masterHost))