fix: migrate 工具按主配置结构读取 Database 段
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user