From a185247ab1bbc2f94ad18ffca80992c3af96eae5 Mon Sep 17 00:00:00 2001 From: SunYF <123@hard_man.com> Date: Fri, 28 Aug 2026 09:25:21 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20migrate=20=E5=B7=A5=E5=85=B7=E6=8C=89?= =?UTF-8?q?=E4=B8=BB=E9=85=8D=E7=BD=AE=E7=BB=93=E6=9E=84=E8=AF=BB=E5=8F=96?= =?UTF-8?q?=20Database=20=E6=AE=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bj_power_mes/schema/tools/migrate.go | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/bj_power_mes/schema/tools/migrate.go b/bj_power_mes/schema/tools/migrate.go index 25f3b07..9eb5e5f 100644 --- a/bj_power_mes/schema/tools/migrate.go +++ b/bj_power_mes/schema/tools/migrate.go @@ -25,16 +25,21 @@ import ( var ( client *ent.Client - ctx = context.Background() + ctx = context.Background() configFile = flag.String("config", "etc/bj_power_mes-api.yaml", "配置文件路径") ) +// migrateConf 与主配置结构一致,只取 Database 段 +type migrateConf struct { + Database db.DatabaseConf +} + func setUp() { - var c db.DatabaseConf + var c migrateConf conf.MustLoad(*configFile, &c) dsn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", - c.Host, c.Port, c.User, c.Password, c.Dbname) + c.Database.Host, c.Database.Port, c.Database.User, c.Database.Password, c.Database.Dbname) db, err := sql.Open("pgx", dsn) if err != nil { log.Fatal(err)