Go：创建 io.Writer 接口以记录到 mongodb 数据库

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《Go：创建 io.Writer 接口以记录到 mongodb 数据库》带大家来了解一下Go：创建 io.Writer 接口以记录到 mongodb 数据库，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

问题内容

使用 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 接口，我认为这解决了一个不同的问题。

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

正确答案

这很容易实现，因为该log.Logger类型保证每条日志消息都通过io.Writer一次Writer.Write()调用传递到目标：

每个日志记录操作都会调用 Writer 的 Write 方法。一个 Logger 可以同时从多个 goroutine 中使用；它保证序列化对 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)
}

今天关于《Go：创建 io.Writer 接口以记录到 mongodb 数据库》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于golang的内容请关注golang学习网公众号！