MongoDB Atlas Serverless事务引发连接池耗尽问题求助

核心问题原因 #

你遇到的连接池耗尽问题，根源在于事务会话（Session）没有被正确关闭，导致连接被永久占用。结合MongoDB Serverless的连接特性，以及你代码中的上下文处理不当，最终触发了连接池耗尽。

代码中的关键问题 #

1. 独立上下文的误用
   你在函数内部创建了一个10秒超时的上下文，同时用于事务执行和会话关闭。如果回调中的事务操作耗时接近10秒，上下文会提前超时，导致session.EndSession()无法正常完成——Serverless环境下，事务会话必须通过完整的EndSession流程才能释放连接，超时的上下文会打断这个过程，让连接一直被占用。

2. 忽略上下文的cancel函数
   context.WithTimeout返回的cancel函数未被调用，会导致上下文资源泄漏，虽然不是直接诱因，但会加重资源占用问题。

3. Serverless与共享集群的差异
   免费版共享集群的连接池管理相对宽松，即使会话关闭不彻底，可能会被后台回收；但Serverless实例采用按需分配的连接模型，事务独占连接的机制更严格，未释放的连接会直接耗尽池资源。

修正后的代码 #

import "log"

func (m MongoDB) Transaction(ctx context.Context, callback func(ctx mongo.SessionContext) (any, error)) (any, error) {
    session, err := m.DB.Client().StartSession()
    if err != nil {
        return nil, fmt.Errorf("failed creating session for transaction | %w", err)
    }

    // 基于调用者传入的上下文创建事务专用超时上下文，并确保cancel被执行
    txCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
    defer cancel()

    // 单独处理会话关闭，捕获关闭时的错误避免吞掉问题
    defer func() {
        if err := session.EndSession(txCtx); err != nil {
            log.Printf("warning: failed to end transaction session: %v", err)
        }
    }()

    res, err := session.WithTransaction(txCtx, callback)
    if err != nil {
        return nil, err
    }

    return res, nil
}

额外排查建议 #

• 检查回调函数内部是否存在未正确处理的事务操作，比如手动调用AbortTransaction后未返回错误，导致驱动无法正确回收连接。
• 升级MongoDB Go驱动到最新稳定版，旧版本可能存在Serverless环境下的连接池泄漏bug。
• 调整事务超时时间，确保与Serverless实例的事务超时配置匹配（避免事务提前被服务端终止，导致连接挂起）。

内容的提问来源于stack exchange，提问作者raphael.oester