golang multiwriter

在Go语言中，有一个非常实用的标准库`io`包中的`MultiWriter`类型，它可以同时将数据写入多个`Writer`中。这个功能对于需要同时将数据输出到多个不同位置的场景非常有用。比如说，当我们需要将日志同时输出到终端和文件中时，就可以使用`MultiWriter`来简化代码。

如何使用MultiWriter

使用`MultiWriter`非常简单，只需要将需要同时输出的`Writer`作为参数传递给`io.MultiWriter`函数即可。例如，我们可以创建一个终端输出的`Writer`对象和一个文件输出的`Writer`对象：

import (
    "os"
    "io"
    "log"
)

func main() {
    file, err := os.Create("log.txt")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    logger := log.New(io.MultiWriter(file, os.Stdout), "log: ", log.Lshortfile)
    logger.Println("Hello, World!")
}

在上面的代码中，我们先通过`os.Create`函数创建一个文件对象，然后使用`io.MultiWriter`函数将文件对象和终端的`Stdout`对象传递给`log.New`函数。最后，我们创建了一个带有文件输出和终端输出的`Logger`对象。当我们调用`logger.Println`函数输出日志时，日志信息会同时被写入到文件和终端中。

使用MultiWriter进行批量写入

除了上面的例子中同时向两个`Writer`中写入数据外，`MultiWriter`还可以很方便地进行批量写入。我们可以将多个`Writer`对象放在一个切片中，然后使用`io.MultiWriter`函数进行批量处理。例如，我们可以将日志分别写入文件和数据库：

import (
    "os"
    "io"
    "log"
    "database/sql"
)

func main() {
    file, err := os.Create("log.txt")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/test")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    writers := []io.Writer{file, db}
    logger := log.New(io.MultiWriter(writers...), "", log.LstdFlags)
    logger.Println("Hello, World!")
}

在上面的代码中，我们将文件对象和数据库对象放在一个切片`writers`中。然后，我们使用`io.MultiWriter`将切片中的所有`Writer`对象作为参数传递给`log.New`函数，创建了一个带有批量写入功能的`Logger`对象。当我们调用`logger.Println`函数输出日志时，日志信息会同时被写入到文件和数据库中。

自定义MultiWriter

尽管标准库中的`io.MultiWriter`非常方便，但有时候我们可能希望定制一些特殊的写入逻辑。这时，我们可以自己实现一个类似于`MultiWriter`的类型。下面是一个简单的示例：

type CustomMultiWriter struct {
    writers []io.Writer
}

func (w *CustomMultiWriter) Write(p []byte) (int, error) {
    var n int
    var err error
    for _, writer := range w.writers {
        n, err = writer.Write(p)
        if err != nil {
            return n, err
        }
    }
    return n, nil
}

func main() {
    file, err := os.Create("log.txt")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    stdout := os.Stdout

    writer := &CustomMultiWriter{
        writers: []io.Writer{file, stdout},
    }

    logger := log.New(writer, "", log.LstdFlags)
    logger.Println("Hello, World!")
}

在上面的代码中，我们定义了一个`CustomMultiWriter`类型，并实现了`Write`方法，该方法会遍历`writers`切片中的所有`Writer`对象，分别将数据写入到它们中。然后，我们创建了一个`CustomMultiWriter`对象，将文件对象和终端的`Stdout`对象传递给它。最后，我们创建了一个带有自定义多重写入功能的`Logger`对象。当我们调用`logger.Println`函数输出日志时，日志信息会同时被写入到文件和终端中。

总的来说，Go语言中的`MultiWriter`类型提供了一种方便的方式，可以同时将数据写入多个`Writer`中。它大大简化了多个输出位置的处理逻辑。无论是同时输出日志到终端和文件，还是进行批量写入，甚至进行自定义的多重写入，`MultiWriter`都可以轻松应对。希望本文对你理解和使用`MultiWriter`有所帮助。