Go：创建用于登录到 mongodb 数据库的 io.Writer 接口

Question

使用 go (golang)：

有没有办法创建一个输出到数据库的记录器？

或者更准确地说，我可以实现某种可以作为第一个参数传递给 io.Writer 的接口log.New()吗？

EG：（dbLogger 将接收日志的输出并将其写入数据库）

logger := log.New(dbLogger, "dbLog: ", log.Lshortfile) logger.Print("This message will be stored in the database")

我假设我应该创建自己的数据库日志功能，但我很好奇是否已经有一种方法可以使用该语言中的现有工具来做到这一点。

在某些情况下，我使用mgo.v2来处理我的 mongodb 数据库，但除了在GridFS 中我没有看到任何 io.Writer 接口，我认为它解决了一个不同的问题。

我也仍然对语言有所了解，所以我可能错误地使用了上面的一些术语。非常欢迎任何更正。

Answer 1

这很容易实现，因为该log.Logger类型保证io.Writer通过单个Writer.Write()调用将每条日志消息传送到目的地：

每个日志操作都会对 Writer 的 Write 方法进行一次调用。一个 Logger 可以在多个 goroutines 中同时使用；它保证序列化对 Writer 的访问。

所以基本上你只需要创建一个实现 的类型io.Writer，它的Write()方法用字节片的内容创建一个新文档，并将其保存在 MongoDB 中。

这是一个简单的实现：

type MongoWriter struct {
    sess *mgo.Session
}

func (mw *MongoWriter) Write(p []byte) (n int, err error) {
    c := mw.sess.DB("").C("log")
    err = c.Insert(bson.M{
        "created": time.Now(),
        "msg":     string(p),
    })
    if err != nil {
        return
    }
    return len(p), nil
}

使用它：

sess := ... // Get a MongoDB session

mw := &MongoWriter{sess}
log.SetOutput(mw)

// Now the default Logger of the log package uses our MongoWriter.
// Generate a log message that will be inserted into MongoDB:
log.Println("I'm the first log message.")
log.Println("I'm multi-line,\nbut will still be in a single log message.")

显然，如果您正在使用另一个log.Logger实例，请将其设置MongoWriter为，例如：

mylogger := log.New(mw, "", 0)
mylogger.Println("Custom logger")

请注意，log.Logger即使日志消息本身不以换行符结尾，日志消息也会以换行符结尾。如果您不想记录结束换行符，您可以简单地将其剪切，例如：

func (mw *MongoWriter) Write(p []byte) (n int, err error) {
    origLen := len(p)
    if len(p) > 0 && p[len(p)-1] == '\n' {
        p = p[:len(p)-1] // Cut terminating newline
    }

    c := mw.sess.DB("").C("log")

    // ... the rest is the same

    return origLen, nil // Must return original length (we resliced p)
}