Golang中如何在DB Handler内创建Mongo会话并在主程延迟关闭

要实现把Session的创建封装在db.go的DBOps中，同时在main函数里确保应用关闭时延迟关闭连接，你可以通过以下步骤调整代码：

1. 修正DBOps的Init方法并添加关闭方法 #

首先在db.go里完善Init逻辑，把创建好的Session赋值给结构体字段，同时新增Close方法来封装Session的关闭操作：

package db

import "gopkg.in/mgo.v2"

type DBOps struct { 
    session *mgo.Session 
} 

// Init 初始化MongoDB连接并绑定到DBOps实例
func (m *DBOps) Init(connectionstr string) error { 
    session, err := mgo.Dial(connectionstr) 
    if err != nil { 
        return err // 返回错误而非panic，让上层更灵活处理异常
    } 
    // 可按需添加Session配置，比如设置一致性模式、超时等
    session.SetMode(mgo.Monotonic, true)
    m.session = session
    return nil
}

// Close 安全关闭MongoDB连接
func (m *DBOps) Close() {
    if m.session != nil {
        m.session.Close()
    }
}

2. 在main中初始化并延迟关闭 #

在main函数里，实例化DBOps、完成连接初始化后，通过defer调用它的Close方法，就能确保应用退出时自动关闭连接。之后把这个DBOps实例传递给UserController即可：

package main

import (
    "log"
    "your-project-path/db" // 替换为你的db包实际路径
)

type UserController struct { 
    DB db.DBOps 
}

func main(){ 
    var dbOps db.DBOps
    connectionStr := "mongodb://localhost:27017" // 你的MongoDB连接字符串

    // 初始化数据库连接
    err := dbOps.Init(connectionStr)
    if err != nil {
        log.Fatalf("Failed to connect to MongoDB: %v", err)
    }
    // 延迟关闭连接，保证应用退出时执行
    defer dbOps.Close()

    // 将DBOps实例传递给UserController
    userController := UserController{DB: dbOps}

    // 后续业务逻辑处理，比如启动服务、处理请求等
    // ...
}

补充说明 #

• 封装性保障：通过Init和Close方法封装Session的生命周期，避免外部直接操作session字段，降低误操作风险。
• 多请求场景优化：如果你的应用是HTTP服务这类多请求场景，推荐在每个请求中使用session.Copy()创建临时Session，用完后关闭，能更好地利用mgo的连接池特性；如果是单任务脚本，全局一个Session并在结束时关闭就足够了。

内容的提问来源于stack exchange，提问作者R.Roshan