From b3658898cc44e1300597cc181cde574caf2be7ac Mon Sep 17 00:00:00 2001
From: djshow832 <873581766@qq.com>
Date: Wed, 27 Mar 2024 17:43:42 +0800
Subject: [PATCH] add tiproxy advertise-addr

---
 components/playground/instance/tiproxy.go | 1 +
 pkg/cluster/spec/tiproxy.go               | 1 +
 2 files changed, 2 insertions(+)

diff --git a/components/playground/instance/tiproxy.go b/components/playground/instance/tiproxy.go
index 5895219888..732670e716 100644
--- a/components/playground/instance/tiproxy.go
+++ b/components/playground/instance/tiproxy.go
@@ -125,6 +125,7 @@ func (c *TiProxy) Start(ctx context.Context, version utils.Version) error {
 	if err := enc.Encode(spec.MergeConfig(userConfig, map[string]any{
 		"proxy.pd-addrs":        strings.Join(endpoints, ","),
 		"proxy.addr":            utils.JoinHostPort(c.Host, c.Port),
+		"proxy.advertise-addr":  AdvertiseHost(c.Host),
 		"api.addr":              utils.JoinHostPort(c.Host, c.StatusPort),
 		"log.log-file.filename": c.LogFile(),
 	})); err != nil {
diff --git a/pkg/cluster/spec/tiproxy.go b/pkg/cluster/spec/tiproxy.go
index dd34af3442..f56ebeb89b 100644
--- a/pkg/cluster/spec/tiproxy.go
+++ b/pkg/cluster/spec/tiproxy.go
@@ -231,6 +231,7 @@ func (i *TiProxyInstance) checkConfig(
 	}
 	cfg["proxy.pd-addrs"] = strings.Join(pds, ",")
 	cfg["proxy.addr"] = utils.JoinHostPort(i.GetListenHost(), i.GetPort())
+	cfg["proxy.advertise-addr"] = spec.Host
 	cfg["api.addr"] = utils.JoinHostPort(i.GetListenHost(), spec.StatusPort)
 	cfg["log.log-file.filename"] = filepath.Join(paths.Log, "tiproxy.log")