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