Add listenAddress support in raft address agent option

logbn · Jan 4, 2025 · 4781a6c · 4781a6c
1 parent a9d6db0
commit 4781a6c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/agent_option.go b/agent_option.go
@@ -33,9 +33,14 @@ func WithGossipAddress(advertiseAddress string, bindAddress ...string) AgentOpti
 	}
 }
 
-func WithRaftAddress(raftAddress string) AgentOption {
+func WithRaftAddress(raftAddress string, listenAddress ...string) AgentOption {
 	return func(a *Agent) error {
 		a.hostConfig.RaftAddress = raftAddress
+		if len(listenAddress) > 0 {
+			a.hostConfig.ListenAddress = listenAddress[0]
+		} else {
+			a.hostConfig.ListenAddress = fmt.Sprintf("0.0.0.0:%s", strings.Split(raftAddress, ":")[1])
+		}
 		return nil
 	}
 }