From 4a7cfab5436782752127fe460d46843162620e30 Mon Sep 17 00:00:00 2001
From: wangchuxiao <wangchuxiao97@outlook.com>
Date: Fri, 15 Apr 2022 11:56:36 +0800
Subject: [PATCH] getui

---
 config/config.yaml     | 2 +-
 pkg/common/db/model.go | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/config/config.yaml b/config/config.yaml
index c78611634..f0e25250c 100644
--- a/config/config.yaml
+++ b/config/config.yaml
@@ -167,7 +167,7 @@ longconnsvr:
   websocketMaxMsgLen: 4096
   websocketTimeOut: 10
 
-## 推送只能开启一个
+## 推送只能开启一个 enable=true代表开启
 push:
   tpns: #腾讯推送，暂未测试 暂不要使用
     ios:
diff --git a/pkg/common/db/model.go b/pkg/common/db/model.go
index f9550e7b3..08b6412d7 100644
--- a/pkg/common/db/model.go
+++ b/pkg/common/db/model.go
@@ -46,11 +46,14 @@ func init() {
 		// example: mongodb://$user:$password@mongo1.mongo:27017,mongo2.mongo:27017,mongo3.mongo:27017/$DBDatabase/?replicaSet=rs0&readPreference=secondary&authSource=admin&maxPoolSize=$DBMaxPoolSize
 		uri = config.Config.Mongo.DBUri
 	} else {
+		if config.Config.Mongo.DBPassword != "" && config.Config.Mongo.DBUserName != "" {
+			uri = fmt.Sprintf("mongodb://%s:%s@%s/%s/?maxPoolSize=%d", config.Config.Mongo.DBUserName, config.Config.Mongo.DBPassword, config.Config.Mongo.DBAddress[0],
+				config.Config.Mongo.DBDatabase, config.Config.Mongo.DBMaxPoolSize)
+		}
 		uri = fmt.Sprintf("mongodb://%s/%s/?maxPoolSize=%d",
 			config.Config.Mongo.DBAddress[0], config.Config.Mongo.DBDatabase,
 			config.Config.Mongo.DBMaxPoolSize)
 	}
-
 	mongoClient, err := mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
 	if err != nil {
 		fmt.Println(" mongo.Connect  failed, try ", utils.GetSelfFuncName(), err.Error(), uri)