From 2b1ed8e8a0fa6fc131124b0d92745fb2b2a4a974 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=96=8C?= <259278618@qq.com>
Date: Fri, 19 Dec 2025 14:58:00 +0800
Subject: [PATCH] =?UTF-8?q?mongodb=E5=9F=BA=E7=A1=80=E6=93=8D=E4=BD=9C?=
 =?UTF-8?q?=E6=96=B9=E6=B3=95=E5=A2=9E=E5=8A=A0redis=E7=BC=93=E5=AD=98?=
 =?UTF-8?q?=E7=AD=96=E7=95=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mongo/mongo.go | 156 +++++--------------------------------------------
 1 file changed, 13 insertions(+), 143 deletions(-)

diff --git a/mongo/mongo.go b/mongo/mongo.go
index 8bb52c8..394cec0 100644
--- a/mongo/mongo.go
+++ b/mongo/mongo.go
@@ -5,7 +5,6 @@ import (
 	"errors"
 	"fmt"
 	"strings"
-	"sync"
 	"time"
 
 	"gitee.com/red-future---jilin-g/common/consts"
@@ -23,166 +22,37 @@ import (
 	"go.mongodb.org/mongo-driver/v2/mongo/options"
 )
 
-var (
-	db           *mongo.Database
-	client       *mongo.Client
-	isConnected  bool
-	mu           sync.RWMutex
-	mongoAddr    string
-	dbName       string
-	healthCtx    context.Context
-	healthCancel context.CancelFunc
-)
+var db *mongo.Database
 
 // GetDB 获取 MongoDB 数据库实例
 func GetDB() *mongo.Database {
-	mu.RLock()
-	defer mu.RUnlock()
 	return db
 }
 
-// connect 建立MongoDB连接
-func connect() error {
-	mu.Lock()
-	defer mu.Unlock()
-
-	if client != nil {
-		err := client.Disconnect(context.Background())
-		if err != nil {
-			return err
-		}
-	}
-
-	// 创建连接选项
-	opt := options.Client().
-		ApplyURI(mongoAddr).
-		SetMaxPoolSize(100).
-		SetMinPoolSize(10).
-		SetMaxConnecting(10).
-		SetConnectTimeout(10 * time.Second)
-
-	var err error
-	client, err = mongo.Connect(opt)
-	if err != nil {
-		isConnected = false
-		glog.Error(context.Background(), "MongoDB连接失败", err)
-		return err
-	}
-
-	// 测试连接
-	testCtx, testCancel := context.WithTimeout(context.Background(), 5*time.Second)
-	defer testCancel()
-
-	err = client.Ping(testCtx, nil)
-	if err != nil {
-		isConnected = false
-		glog.Error(testCtx, "MongoDB连接测试失败", err)
-		return err
-	}
-
-	db = client.Database(dbName)
-	isConnected = true
-	glog.Info(context.Background(), "✅ MongoDB连接成功")
-	return nil
-}
-
-// healthCheck 健康检查协程
-func healthCheck() {
-	ticker := time.NewTicker(30 * time.Second)
-	defer ticker.Stop()
-
-	for {
-		select {
-		case <-healthCtx.Done():
-			return
-		case <-ticker.C:
-			mu.RLock()
-			currentConnected := isConnected
-			currentClient := client
-			mu.RUnlock()
-
-			if !currentConnected || currentClient == nil {
-				glog.Warning(context.Background(), "MongoDB连接断开，尝试重连")
-				if err := reconnect(); err != nil {
-					glog.Error(context.Background(), "MongoDB重连失败", err)
-				}
-				continue
-			}
-
-			// 测试连接状态
-			testCtx, testCancel := context.WithTimeout(context.Background(), 5*time.Second)
-			err := currentClient.Ping(testCtx, nil)
-			testCancel()
-
-			if err != nil {
-				mu.Lock()
-				isConnected = false
-				mu.Unlock()
-				glog.Warning(context.Background(), "MongoDB连接健康检查失败", err)
-
-				// 尝试重连
-				if err := reconnect(); err != nil {
-					glog.Error(context.Background(), "MongoDB重连失败", err)
-				}
-			} else {
-				glog.Debug(context.Background(), "MongoDB连接健康检查通过")
-			}
-		}
-	}
-}
-
-// reconnect 重连函数
-func reconnect() error {
-	maxRetries := 3
-	retryDelay := 2 * time.Second
-
-	for i := 0; i < maxRetries; i++ {
-		glog.Info(context.Background(), fmt.Sprintf("尝试第%d次重连MongoDB", i+1))
-
-		if err := connect(); err == nil {
-			glog.Info(context.Background(), "MongoDB重连成功")
-			return nil
-		}
-
-		if i < maxRetries-1 {
-			time.Sleep(retryDelay)
-			retryDelay *= 2 // 指数退避
-		}
-	}
-
-	return gerror.New("MongoDB重连失败，已达到最大重试次数")
-}
-
-// init 初始化MongoDB连接
 func init() {
 	// 按需初始化：没有配置 mongo.address 则跳过
-	mongoAddr = g.Cfg().MustGet(context.Background(), "mongo.address").String()
+	mongoAddr := g.Cfg().MustGet(context.Background(), "mongo.address").String()
 	if mongoAddr == "" {
 		return
 	}
 
-	// 创建健康检查上下文
-	healthCtx, healthCancel = context.WithCancel(context.Background())
-
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	opt := options.Client().ApplyURI(mongoAddr)
+	client, err := mongo.Connect(opt)
+	if err != nil {
+		glog.Error(ctx, "mongodb连接失败", err)
+		return
+	}
 	// 从连接串中解析数据库名
-	dbName = gstr.SubStr(mongoAddr, strings.LastIndex(mongoAddr, "/")+1, len(mongoAddr))
+	dbName := gstr.SubStr(mongoAddr, strings.LastIndex(mongoAddr, "/")+1, len(mongoAddr))
 	// 如果连接串带有参数（如 ?retryWrites=true），需要去掉参数部分
 	if strings.Contains(dbName, "?") {
 		dbName = gstr.SubStr(dbName, 0, strings.Index(dbName, "?"))
 	}
-
-	// 初始连接
-	if err := connect(); err != nil {
-		glog.Error(context.Background(), "MongoDB初始连接失败", err)
-		return
-	}
-
-	// 启动健康检查协程
-	go healthCheck()
-
-	glog.Info(context.Background(), "✅ MongoDB初始化完成，连接健康检查已启动")
+	db = client.Database(dbName)
+	glog.Info(ctx, "✅ MongoDB 初始化成功")
 }
-
 func listOptionsToMap(ctx context.Context, opts ...options.Lister[options.FindOptions]) (m map[string]interface{}) {
 	// 输出opts参数中的值
 	m = make(map[string]interface{})