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类型可以保证每条日志消息通过一次Writer.Write()调用传递到目标io.Writer：

每个日志记录操作只需调用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数据库

1 个答案: