添加全局变量保存使用的数据库连接数据

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

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
 	}

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