为什么 IO.Writer 没有填充接收器？

Question

我正在尝试测试模板生成工具。为了做到这一点，我认为捕获模板执行输出的最简单方法是使用 io writer 并在测试期间提供它。问题是由于某些原因，接收器没有用模板输出“更新”。希望下面的代码更好地解释了我面临的问题。package mainimport "fmt"import "text/template"type Company struct{    Name string} type Companies []Companyfunc main() {    s := new(stringer)    v := Companies{Company{Name:"Sony"}}    tmp :=  template.Must(template.New("main").Parse(src))    if err := tmp.Execute(s, v); err !=nil{        panic(err)    }    if *s != "this is the header template"{        fmt.Println("expected: \"this is the header template\" received: ", *s)     }else{      fmt.Println("s is %v", *s)    }}type stringer stringfunc (s *stringer)Write(b []byte)(int, error){    *s = stringer(b)    return len(b), nil}var src = ` this is the header template    {{range .}}    {{.Name}}    {{end}}`http://play.golang.org/p/y4zWgyd5G1

Answer 1

您的stringer类型只是输入的“别名” *string。string在 Go 中是不可变的。您不应该使用 astring或指向 a 的指针来string构建模板的输出，因为您不能修改 a string，您只能创建一个新的（并丢弃旧的）。

template.Execute()期待一个io.Writer. Write()输出的方法可能会被多次调用，并且您的stringer.Write()方法总是会丢弃之前写入它的数据。您可以通过始终将新数据连接到旧数据来修复它，如下所示：

*s = *s + stringer(b)

但是这个解决方案非常低效（它生成新string的并丢弃旧的）。

一个更好且随时可用的替代方案是bytes.Buffer. 您可以创建一个字节缓冲区来实现这样的Writer接口：

bytes.NewBuffer(nil)

您不需要创建自己的stringer类型。在Go Playground上试试你修改过的程序。