From a64002217d946814c62f031ef2bcde9494df4239 Mon Sep 17 00:00:00 2001
From: zhangxin <Xin.zhang@le-it.com.cn>
Date: Mon, 2 Aug 2021 16:14:53 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=85=A8=E5=B1=80=E5=8F=98?=
 =?UTF-8?q?=E9=87=8F=E4=BF=9D=E5=AD=98=E4=BD=BF=E7=94=A8=E7=9A=84=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=BA=93=E8=BF=9E=E6=8E=A5=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 global/global.go  | 2 ++
 infra/db/mongo.go | 5 ++++-
 infra/db/mssql.go | 2 ++
 3 files changed, 8 insertions(+), 1 deletion(-)

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