本文详解如何在 go 中正确管理 mongodb 连接，解决因每次请求新建会话导致的连接数激增问题，通过复用客户端、配置连接池参数及使用现代驱动（mongo-go-driver）实现稳定、高效、可伸缩的数据访问。
本文详解如何在 go 中正确管理 mongodb 连接，解决因每次请求新建会话导致的连接数激增问题，通过复用客户端、配置连接池参数及使用现代驱动（mongo-go-driver）实现稳定、高效、可伸缩的数据访问。
你提供的 Go 代码使用了已停止维护的 mgo 驱动（最后更新于 2018 年），且存在严重连接管理缺陷：每个 HTTP 请求都调用 mgo.Dial("localhost") 创建全新会话，却仅在 SendData 中 defer closeSession(session) —— 而 ReceiveData 函数完全遗漏了 session.Close()，导致连接持续累积、无法复用，最终触发 MongoDB 连接数飙升（日志中从 #60 涨至 #81+），甚至可能耗尽系统文件描述符或触发 Atlas 连接限制（如 M10 集群单节点上限 1500 连接）。
这不是“限制为 1 个连接”的问题，而是架构级错误：MongoDB 客户端本应是长生命周期的单例，而非按请求创建/销毁。现代最佳实践要求：
✅ 正确做法：全局复用 Client，而非 Session
MongoDB 官方推荐的 mongo-go-driver（自 2019 年起取代 mgo）将连接池内置于 *mongo.Client 中。一个 Client 实例可安全并发使用，自动复用连接、负载均衡、故障转移。
✅ 示例：基于 mongo-go-driver 的生产级初始化
package main
import (
"context"
"log"
"net/http"
"time"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/mongo/readpref"
)
var client *mongo.Client // 全局单例
func initDB() {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
// 配置连接池：关键参数控制并发连接行为
clientOptions := options.Client().
ApplyURI("mongodb://localhost:27017").
SetMaxPoolSize(20). // 最大并发连接数（默认100，按负载调整）
SetMinPoolSize(5). // 保持的最小空闲连接数（防冷启动延迟）
SetMaxConnIdleTime(30 * time.Second). // 空闲连接最大存活时间
SetConnectTimeout(5 * time.Second). // 建连超时
SetReadPreference(readpref.Primary()) // 优先读主节点
var err error
client, err = mongo.Connect(ctx, clientOptions)
if err != nil {
log.Fatal("Failed to connect to MongoDB:", err)
}
// 验证连接
err = client.Ping(ctx, readpref.Primary())
if err != nil {
log.Fatal("Failed to ping MongoDB:", err)
}
log.Println("MongoDB connected successfully")
}
func ReceiveData(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
body, err := io.ReadAll(r.Body)
if err != nil {
http.Error(w, "Failed to read body", http.StatusBadRequest)
return
}
// 复用全局 client，无需 Dial/Close
collection := client.Database("schedule_calculator").Collection("schedule_save")
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
// 插入文档（结构体需适配 bson 标签）
doc := map[string]interface{}{
"_id": bson.NewObjectId(),
"data": string(body),
"date": time.Now(),
}
_, err = collection.InsertOne(ctx, doc)
if err != nil {
http.Error(w, "Database insert failed", http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(map[string]string{"status": "success"})
}
// SendData 同理，复用 client
func SendData(w http.ResponseWriter, r *http.Request) {
collection := client.Database("schedule_calculator").Collection("schedule_save")
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
count, err := collection.CountDocuments(ctx, bson.M{})
if err != nil || count == 0 {
http.Error(w, "No data found", http.StatusNotFound)
return
}
// 获取最新一条记录（优化：用 sort + limit 替代 Skip）
cursor, err := collection.Find(ctx, bson.M{},
options.Find().SetSort(bson.D{{"date", -1}}).SetLimit(1))
if err != nil {
http.Error(w, "Query failed", http.StatusInternalServerError)
return
}
defer cursor.Close(ctx)
var result struct {
Data string `bson:"data"`
Date time.Time `bson:"date"`
}
if cursor.Next(ctx) {
if err = cursor.Decode(&result); err != nil {
http.Error(w, "Decode failed", http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(result.Data)
} else {
http.Error(w, "No document returned", http.StatusInternalServerError)
}
}
func main() {
initDB()
defer func() {
if client != nil {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
if err := client.Disconnect(ctx); err != nil {
log.Printf("Failed to disconnect from MongoDB: %v", err)
}
}
}()
http.HandleFunc("/receive", ReceiveData)
http.HandleFunc("/send", SendData)
log.Println("Server starting on :8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}
⚠️ 关键注意事项
- 绝不使用 mgo：该库无 TLS 1.3 支持、不兼容 MongoDB 4.4+ 新特性，且存在已知竞态与内存泄漏。
- Client 是线程安全的：整个应用只需一个 *mongo.Client 实例，由驱动自动管理底层连接池。
- SetMaxPoolSize 是核心调优项：设为 20 可覆盖大多数中等负载场景；过高（如 >100）易触发 Atlas 连接上限；过低则导致请求排队阻塞。
- 避免 Skip(n) 分页：Skip(dbSize-1) 在大数据集下性能极差（需扫描前 N-1 条），应改用 _id 或时间戳索引 + Sort({date: -1}).Limit(1)。
- 务必设置 context 超时：防止数据库慢查询拖垮整个服务。
- 生产环境必须启用认证与 TLS：连接字符串示例：mongodb+srv://user:pass@cluster.mongodb.net/?tls=true&authSource=admin。
? 连接数验证方式
- MongoDB Shell 执行 db.currentOp({"secs_running": {"$gt": 0}}).itcount() 查看活跃操作；
- 监控 serverStatus.connections 字段；
- Atlas 控制台实时查看 “Connections” 指标（按节点/Router 统计）。
遵循以上实践，你的 Go 服务将稳定维持数十个连接（由 MaxPoolSize 控制），彻底告别连接风暴，同时获得官方驱动的全部新特性支持与安全更新。
