From a55e8b53a3e1dddf66f35a3fc7cbd34bb9433a7a Mon Sep 17 00:00:00 2001 From: qintang Date: Thu, 12 May 2022 12:43:35 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20DBConf=E5=A2=9E=E5=8A=A0DB=E9=85=8D?= =?UTF-8?q?=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dtmcli/dtmimp/types.go | 1 + dtmcli/dtmimp/utils.go | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/dtmcli/dtmimp/types.go b/dtmcli/dtmimp/types.go index a0738f8..827b1c1 100644 --- a/dtmcli/dtmimp/types.go +++ b/dtmcli/dtmimp/types.go @@ -21,4 +21,5 @@ type DBConf struct { Port int64 `yaml:"Port"` User string `yaml:"User"` Password string `yaml:"Password"` + Db string `yaml:"Db"` } diff --git a/dtmcli/dtmimp/utils.go b/dtmcli/dtmimp/utils.go index 1d08358..7220b30 100644 --- a/dtmcli/dtmimp/utils.go +++ b/dtmcli/dtmimp/utils.go @@ -210,9 +210,9 @@ func GetDsn(conf DBConf) string { driver := conf.Driver dsn := map[string]string{ "mysql": fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=true&loc=Local&interpolateParams=true", - conf.User, conf.Password, host, conf.Port, ""), + conf.User, conf.Password, host, conf.Port, conf.Db), "postgres": fmt.Sprintf("host=%s user=%s password=%s dbname='%s' port=%d sslmode=disable", - host, conf.User, conf.Password, "", conf.Port), + host, conf.User, conf.Password, conf.Db, conf.Port), }[driver] PanicIf(dsn == "", fmt.Errorf("unknow driver: %s", driver)) return dsn