fix dump

Signed-off-by: wjhuang2016 <[email protected]>
PingCAP-QE · Apr 1, 2024 · 6f6587a · 6f6587a
1 parent 39a3941
commit 6f6587a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dump/dumping.go b/dump/dumping.go
@@ -13,8 +13,9 @@ import (
 	"strings"
 )
 
-func DumpToFile(dbName string, tblName []string, fileName string) error {
+func DumpToFile(dbName string, tblName []string, fileName string, host string) error {
 	conf := export.DefaultConfig()
+	conf.Host = strings.Split(host, ":")[0]
 	conf.Port = 4000
 
 	ns := make([]string, len(tblName))

diff --git a/framework/mainloop.go b/framework/mainloop.go
@@ -687,7 +687,7 @@ func (c *testCase) execute(ctx context.Context) error {
 					return err
 				}
 				pwd := os.Getenv("PWD")
-				err = dump.DumpToFile("test", tblNames, fmt.Sprintf("local://%s/bug-%s-%d", pwd, time.Now().Format("2006-01-02-15-04-05"), num))
+				err = dump.DumpToFile("test", tblNames, fmt.Sprintf("local://%s/bug-%s-%d", pwd, time.Now().Format("2006-01-02-15-04-05"), num), c.cfg.dbAddr)
 				if err != nil {
 					return err
 				}