playground: add tiproxy to dsn

Signed-off-by: xhe <[email protected]>
pingcap · Nov 12, 2024 · 78fc06d · 78fc06d
1 parent 775384c
commit 78fc06d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/components/playground/main.go b/components/playground/main.go
@@ -553,11 +553,14 @@ func loadPort(dir string) (port int, err error) {
 	return
 }
 
-func dumpDSN(fname string, dbs []*instance.TiDBInstance) {
+func dumpDSN(fname string, dbs []*instance.TiDBInstance, tdbs []*instance.TiProxy) {
 	var dsn []string
 	for _, db := range dbs {
 		dsn = append(dsn, fmt.Sprintf("mysql://root@%s", db.Addr()))
 	}
+	for _, tdb := range tdbs {
+		dsn = append(dsn, fmt.Sprintf("mysql://root@%s", tdb.Addr()))
+	}
 	_ = utils.WriteFile(fname, []byte(strings.Join(dsn, "\n")), 0644)
 }
 

diff --git a/components/playground/playground.go b/components/playground/playground.go
@@ -1312,7 +1312,7 @@ func (p *Playground) bootCluster(ctx context.Context, env *environment.Environme
 		p.updateMonitorTopology(spec.ComponentPrometheus, *monitorInfo)
 	}
 
-	dumpDSN(filepath.Join(p.dataDir, "dsn"), p.tidbs)
+	dumpDSN(filepath.Join(p.dataDir, "dsn"), p.tidbs, p.tiproxys)
 
 	go func() {
 		// fmt.Printf("serve at :%d\n", p.port)