转到：创建io.Writer接口以登录到mongodb数据库

小编典典

转到：创建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接口，我认为这解决了一个不同的问题。

我仍然对这种语言有所了解，因此我可能在上面不正确地使用了一些术语。任何更正都非常欢迎。

阅读 222

2020-07-02

共1个答案

小编典典

这很容易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)
}

2020-07-02