本文介绍如何通过实现标准库的 io.Writer 接口，构建一个可直接用于 log.New() 或 log.SetOutput() 的 MongoDB 日志写入器，使日志消息自动持久化到 MongoDB 集合中。
本文介绍如何通过实现标准库的 io.Writer 接口，构建一个可直接用于 log.New() 或 log.SetOutput() 的 MongoDB 日志写入器，使日志消息自动持久化到 MongoDB 集合中。
Go 标准库的 log.Logger 设计精巧：它严格保证每次日志输出仅调用一次 io.Writer.Write() 方法，并且线程安全。这使得我们能轻松将任意后端（如文件、网络、数据库）接入日志系统——只需实现一个满足 io.Writer 签名的类型即可。
下面是一个基于 mgo.v2（适用于旧版 MongoDB 驱动）的 MongoWriter 实现：
import (
"io"
"time"
"gopkg.in/mgo.v2"
"gopkg.in/mgo.v2/bson"
)
type MongoWriter struct {
sess *mgo.Session
}
func (mw *MongoWriter) Write(p []byte) (n int, err error) {
// 安全截断末尾换行符（log 会自动添加，通常无需存储）
origLen := len(p)
if origLen > 0 && p[origLen-1] == '\n' {
p = p[:origLen-1]
}
// 获取日志集合（建议使用专用 DB，如 "logs"）
c := mw.sess.DB("logs").C("entries")
doc := bson.M{
"created": time.Now().UTC(),
"msg": string(p),
"level": "info", // 可扩展为支持 debug/warn/error 字段
}
err = c.Insert(doc)
if err != nil {
return 0, err // 返回 0 表示写入失败，log 会记录错误（若配置了 error output）
}
return origLen, nil // 必须返回原始字节数，符合 io.Writer 合约
}
✅ 使用方式（两种等效方式）：
// 方式 1：替换全局 logger 输出（适合简单场景）
sess, _ := mgo.Dial("mongodb://localhost:27017")
defer sess.Close()
mw := &MongoWriter{sess}
log.SetOutput(mw)
log.Println("这条日志将写入 MongoDB logs.entries 集合")
// 方式 2：创建独立 logger 实例（推荐，更可控）
myLogger := log.New(mw, "[DB] ", log.LstdFlags|log.Lshortfile)
myLogger.Println("自定义前缀 + 文件位置信息")
⚠️ 关键注意事项：
-
性能与可靠性：MongoDB 写入是同步 I/O 操作，高频日志可能成为瓶颈。生产环境建议：
- 添加连接池复用（
mw.sess.Copy()避免共享 session）； - 使用 goroutine 异步写入 + channel 缓冲（需自行实现缓冲队列与错误重试）；
- 考虑引入日志中间件（如
zap+ 自定义 sink）替代标准log。
- 添加连接池复用（
-
Session 生命周期：确保
*mgo.Session在Write()调用期间有效；避免在Write()中调用Copy()后未Close()导致连接泄漏。 -
错误处理：
io.Writer.Write()返回错误时，log.Logger会尝试写入其errorOutput（默认为os.Stderr）。建议在MongoWriter中记录驱动错误（如log.Printf("MongoWriter error: %v", err)），便于调试。 -
兼容性提示：
mgo.v2已归档，新项目请迁移到官方驱动go.mongodb.org/mongo-driver/mongo，其Collection.InsertOne()接口需适配为异步（context.Context参数），此时Write()应使用context.Background()或传入上下文字段。
? 进阶建议：
可扩展 MongoWriter 支持结构化日志（例如解析 log.Lshortfile 生成 file, line, func 字段），或集成 logrus/zerolog 的 Hook 机制，实现更灵活的日志路由与过滤能力。
总之，利用 io.Writer 的抽象能力，Go 让数据库日志变得简洁而强大——核心在于尊重接口契约，兼顾健壮性与可观测性。
