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())