From ede3b07963a82527f438cc610a4350ed2965e123 Mon Sep 17 00:00:00 2001
From: wenxu12345 <44203734@qq.com>
Date: Mon, 21 Feb 2022 15:14:24 +0800
Subject: [PATCH] replace mongo driver

---
 pkg/common/db/model.go | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/pkg/common/db/model.go b/pkg/common/db/model.go
index 729e1323e..f6f074c48 100644
--- a/pkg/common/db/model.go
+++ b/pkg/common/db/model.go
@@ -43,6 +43,16 @@ func init() {
 		config.Config.Mongo.DBMaxPoolSize)
 
 	mongoClient, err := mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
+	if err != nil{
+		log.NewError(" mongo.Connect  failed, try ", err.Error(), uri)
+		time.Sleep(time.Duration(30) * time.Second)
+		mongoClient, err1 = mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
+		if err1 != nil {
+			log.NewError(" mongo.Connect  failed, panic", err.Error(), uri)
+			panic(err1.Error())
+		}
+	}
+
 
 
 	mgoDailInfo := &mgo.DialInfo{
@@ -58,13 +68,7 @@ func init() {
 	mgoSession, err = mgo.DialWithInfo(mgoDailInfo)
 
 	if err != nil {
-		log.NewError(" mongo.Connect  failed, try ", err.Error(), uri)
-		time.Sleep(time.Duration(30) * time.Second)
-		mongoClient, err1 = mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
-		if err1 != nil {
-			log.NewError(" mongo.Connect  failed, panic", err.Error(), uri)
-			panic(err1.Error())
-		}
+
 		mgoSession, err1 = mgo.DialWithInfo(mgoDailInfo)
 		if err1 != nil {
 			log.NewError(" mongo.Connect  failed, panic", err.Error(), uri)