From 97dfd18eb49cda83f10c73d2c18cac0ae358365f Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Tue, 20 May 2025 17:36:54 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0mongodb=E4=BD=BF=E7=94=A8?=
 =?UTF-8?q?=EF=BC=8C=E6=B7=BB=E5=8A=A0=E7=94=A8=E6=88=B7=E5=89=8D=E7=AB=AF?=
 =?UTF-8?q?=E9=85=8D=E7=BD=AE=E7=9A=84=E4=B8=8A=E4=BC=A0=E5=8F=8A=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dao/db.go | 20 ++++++++++++--------
 main.go   |  4 ++++
 2 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/dao/db.go b/dao/db.go
index 9141633..58aa13c 100644
--- a/dao/db.go
+++ b/dao/db.go
@@ -148,21 +148,25 @@ var mongoClient *mongo.Client
 
 //var collection *mongo.Collection
 
-func InitMongoDB() {
+func InitMongoDB() error {
 	// 设置 MongoDB 客户端选项
 	clientOptions := options.Client().ApplyURI(proto.Config.MONGO_URI)
 	// 连接到 MongoDB
 	var err error
 	mongoClient, err = mongo.Connect(context.TODO(), clientOptions)
 	if err != nil {
-		log.Fatal(err)
+		log.Println("Error connecting to MongoDB:", err)
+	} else {
+		// 检查连接
+		err = mongoClient.Ping(context.TODO(), nil)
+		if err != nil {
+			log.Println("Error pinging MongoDB:", err)
+		} else {
+			log.Println("Connected to MongoDB!")
+		}
 	}
-	// 检查连接
-	err = mongoClient.Ping(context.TODO(), nil)
-	if err != nil {
-		log.Fatal(err)
-	}
-	log.Println("Connected to MongoDB!")
+
+	return err
 }
 
 func CloseMongoDB() {
diff --git a/main.go b/main.go
index 652b79f..7e0be58 100644
--- a/main.go
+++ b/main.go
@@ -27,6 +27,10 @@ func main() {
 	if err != nil {
 		panic("failed to connect database:" + err.Error())
 	}
+	err = dao.InitMongoDB()
+	if err != nil {
+		panic("failed to connect mongodb:" + err.Error())
+	}
 	err = worker.InitRedis()
 	if err != nil {
 		panic("failed to connect redis:" + err.Error())