diff --git a/global/global.go b/global/global.go index 236ffa6..bb7b3c2 100644 --- a/global/global.go +++ b/global/global.go @@ -7,6 +7,8 @@ import ( ) var TimezoneLocation *time.Location +var DatabaseDsn string +var MongoDsn string type TaskRecord struct { TaskMap map[int]time.Time diff --git a/infra/db/mongo.go b/infra/db/mongo.go index be10ac9..b1c6efe 100644 --- a/infra/db/mongo.go +++ b/infra/db/mongo.go @@ -1,6 +1,7 @@ package db import ( + "LAPP_ETL/global" "LAPP_ETL/infra/config" "context" "fmt" @@ -14,7 +15,9 @@ var AppMongo *mongo.Client func InitMongo() error { //1.建立链接 ctx, _ := context.WithTimeout(context.Background(), 10*time.Second) - client, err := mongo.Connect(ctx, options.Client().ApplyURI(fmt.Sprintf("mongodb://%s:%d", config.AppConfig.Mongo.Host, config.AppConfig.Mongo.Port))) + dsn := fmt.Sprintf("mongodb://%s:%d", config.AppConfig.Mongo.Host, config.AppConfig.Mongo.Port) + global.MongoDsn = dsn + client, err := mongo.Connect(ctx, options.Client().ApplyURI(dsn)) if err != nil { return err } diff --git a/infra/db/mssql.go b/infra/db/mssql.go index a089a7d..784ac2d 100644 --- a/infra/db/mssql.go +++ b/infra/db/mssql.go @@ -1,6 +1,7 @@ package db import ( + "LAPP_ETL/global" "LAPP_ETL/infra/config" model "LAPP_ETL/models/etl" "fmt" @@ -17,6 +18,7 @@ func InitDB() error { config.AppConfig.DB.User, config.AppConfig.DB.Password, config.AppConfig.DB.Port) + global.DatabaseDsn = dsn engine, err := xorm.NewEngine("mssql", dsn) if err != nil { return err