fix: migrate 工具按主配置结构读取 Database 段

This commit is contained in:
SunYF
2026-08-28 09:25:21 +08:00
parent 450a094f0e
commit a185247ab1
+8 -3
View File
@@ -25,16 +25,21 @@ import (
var ( var (
client *ent.Client client *ent.Client
ctx = context.Background() ctx = context.Background()
configFile = flag.String("config", "etc/bj_power_mes-api.yaml", "配置文件路径") configFile = flag.String("config", "etc/bj_power_mes-api.yaml", "配置文件路径")
) )
// migrateConf 与主配置结构一致,只取 Database 段
type migrateConf struct {
Database db.DatabaseConf
}
func setUp() { func setUp() {
var c db.DatabaseConf var c migrateConf
conf.MustLoad(*configFile, &c) conf.MustLoad(*configFile, &c)
dsn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", 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) db, err := sql.Open("pgx", dsn)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)