Go语言Mgo应用中的连接管理与TCP超时处理指南

package main

import (
    "log"
    "time"

    "gopkg.in/mgo.v2"
)

var globalSession *mgo.Session

func init() {
    dialInfo := &mgo.DialInfo{
        Addrs:    []string{"localhost:27017"},
        Timeout:  10 * time.Second, // 设置连接和操作超时，例如10秒
        Database: "yourdb",
        Username: "user",
        Password: "password",
        // Source:   "admin", // 如果用户名/密码在非admin数据库，可能需要指定认证源
    }

    var err error
    globalSession, err = mgo.DialWithInfo(dialInfo)
    if err != nil {
        log.Fatalf("无法连接到MongoDB: %v", err)
    }

    // 可选：设置会话模式，确保读写一致性
    // mgo.Monotonic 表示读操作可能从不同的服务器读取数据，但会保持操作的顺序
    // mgo.Strong 表示读写操作都必须在主节点上进行，提供最强一致性
    globalSession.SetMode(mgo.Monotonic, true)
    log.Println("成功连接到MongoDB。")
}

func main() {
    // 应用程序生命周期结束时关闭全局会话
    defer func() {
        if globalSession != nil {
            globalSession.Close()
            log.Println("MongoDB全局会话已关闭。")
        }
    }()

    // 应用程序的其他逻辑
    // ...
}

// getCollection 辅助函数：获取集合并返回一个复制的会话
func getCollection(collectionName string) (*mgo.Collection, *mgo.Session) {
    session := globalSession.Copy() // 从全局主会话复制一个新的会话
    // 调用者负责 defer session.Close()
    return session.DB("yourdb").C(collectionName), session
}

// 示例：在一个HTTP请求处理函数或服务方法中如何使用
func handleGetUser(userID string) (interface{}, error) {
    coll, session := getCollection("users")
    defer session.Close() // 确保会话返回连接池

    var user struct {
        ID   string `bson:"_id"`
        Name string `bson:"name"`
        Email string `bson:"email"`
    }

    err := coll.FindId(userID).One(&user)
    if err != nil {
        if err == mgo.ErrNotFound {
            return nil, fmt.Errorf("用户 %s 未找到", userID)
        }
        // 记录详细错误，例如包含原始错误信息
        log.Printf("查询用户 %s 失败: %v", userID, err)
        return nil, fmt.Errorf("数据库操作失败: %w", err)
    }
    return user, nil
}

// 示例：处理会话可能出现的瞬时错误
func processDataWithRetry(documentID string) error {
    coll, session := getCollection("documents")
    defer session.Close()

    // 尝试执行操作
    err := coll.UpdateId(documentID, map[string]interface{}{"$set": {"status": "processed"}}).Err()
    if err != nil {
        // 如果遇到瞬时网络错误或超时，可以尝试刷新会话
        // mgo.Session的Refresh()方法会尝试重新建立底层连接（如果需要）并清除会话的错误状态
        if mgo.Is
            log.Printf("文档 %s 更新失败，尝试刷新会话: %v", documentID, err)
            session.Refresh() // 刷新会话，清除错误状态
            // 再次尝试，或直接返回错误让上层处理
            // err = coll.UpdateId(documentID, map[string]interface{}{"$set": {"status": "processed"}}).Err()
            // if err != nil {
            //  return fmt.Errorf("刷新后再次更新文档 %s 失败: %w", documentID, err)
            // }
        }
        return fmt.Errorf("更新文档 %s 失败: %w", documentID, err)
    }
    return nil
}