golang 在 log.Logger 中使用代理 io.Writer 时的行为有所不同

在 Go 中使用 log.Logger 时，通过代理 io.Writer 输出的行为有所不同。直接调用代理的 write() 方法时，输出正常。然而，当将代理插入 log.Logger 实例中时，输出仅显示最后一条日志，且重复多次。原因在于 Logger.Print 在内部重复使用切片，导致对切片的追加操作影响了后续的打印。为了解决此问题，需要在使用切片之前复制它。

我正在尝试实现一个满足 io.writer 的代理，这样我就可以将其插入记录器中。这个想法是，它将像平常一样打印输出，但也会保留数据的副本以供稍后读取。

下面代码中的 proxyio 结构应该执行此操作，实际上只要我直接调用它的 write() 方法，它就会执行此操作。但是，当我将其插入 log.logger 实例时，输出是意外的。

（这是精简代码，我想使用的原始实现是使用映射和循环指针，而不是示例代码中使用的 [][]byte buf。此外，我还删除了所有锁定。）

package main

import (
    "fmt"
    "io"
    "io/ioutil"
    "log"
)

type proxyio struct {
    out io.writer // the io we are proxying
    buf [][]byte
}

func newproxyio(out io.writer) *proxyio {
    return &proxyio{
        out: out,
        buf: [][]byte{},
    }
}

func (r *proxyio) write(s []byte) (int, error) {
    r.out.write(s)
    r.buf = append(r.buf, s)
    return len(s), nil
}

func main() {
    p := newproxyio(ioutil.discard)
    p.write([]byte("test1\n"))
    p.write([]byte("test2\n"))
    p.write([]byte("test3\n"))
    l := log.new(p, "", 0)
    l.print("test4")
    l.print("test5")
    l.print("test6")
    for i, e := range p.buf {
        fmt.printf("%d: %s", i, e)
    }
}

（这是演示文稿上的代码 https://play.golang.org/p/uooq4nd-rmi）

我期望这段代码的输出如下：

0: test1
1: test2
2: test3
3: test4
4: test5
5: test6

但是，它总是会打印：

0: test1
1: test2
2: test3
3: test6
4: test6
5: test6

我的地图实现的行为是相同的。我还尝试使用 container/list 中的双向链表作为存储，它总是相同的。所以我一定在这里错过了一些实质性的东西。

为什么我在缓冲区中看到最后一个日志输出三次，而不是最后三行日志输出？

解决方案

如果您查看 Logger.Print 的源代码，您会看到它调用 logger.Output。您会注意到它如何将字符串的值设置为 l.buf，然后调用 write

如果您阅读 this answer，您会发现即使所有内容都是按值传递的

当您将切片传递给函数时，将从中创建一个副本 标头，包括指针，该指针将指向相同的支持 数组。

所以当你这样做时：

l.print("test4")
l.print("test5")
l.print("test6")

logger 有效地重复使用同一切片，并且您 appending 对该同一切片的引用三次，因此在打印时自然会使用最近设置的值三次。

要解决此问题，您可以在使用 []byte 之前复制它，如下所示：

func (r *ProxyIO) Write(s []byte) (int, error) {
    c := make([]byte, len(s))
    copy(c, s)
    r.out.Write(c)
    r.buf = append(r.buf, c)
    return len(c), nil
}

更新演示：https://play.golang.org/p/DIWC1Xa6w0R

本篇关于《golang 在 log.Logger 中使用代理 io.Writer 时的行为有所不同》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！