如何使用String()字段打印struct?

时间:2015-10-15 07:37:19

标签: string go struct

此代码:

type A struct {
    t time.Time
}

func main() {
    a := A{time.Now()}
    fmt.Println(a)
    fmt.Println(a.t)
}

打印:

{{63393490800 0 0x206da0}}
2009-11-10 23:00:00 +0000 UTC

A没有实现String(),因此它不是fmt.Stringer并打印其原生代表。但是对于我想要打印的每个结构实现String()非常繁琐。更糟糕的是,如果我添加或删除某些字段,我必须更新String()。是否有一种更简单的方法来打印结构,其字段是' String() S'

1 个答案:

答案 0 :(得分:4)

这就是fmt软件包的实现方式,因此您无法对其进行更改。

但是你可以编写一个帮助函数,它使用反射(reflect包)迭代结构的字段,如果有这样的方法,可以在字段上调用String()方法。

示例实施:

func PrintStruct(s interface{}, names bool) string {
    v := reflect.ValueOf(s)
    t := v.Type()
    // To avoid panic if s is not a struct:
    if t.Kind() != reflect.Struct {
        return fmt.Sprint(s)
    }

    b := &bytes.Buffer{}
    b.WriteString("{")
    for i := 0; i < v.NumField(); i++ {
        if i > 0 {
            b.WriteString(" ")
        }
        v2 := v.Field(i)
        if names {
            b.WriteString(t.Field(i).Name)
            b.WriteString(":")
        }
        if v2.CanInterface() {
            if st, ok := v2.Interface().(fmt.Stringer); ok {
                b.WriteString(st.String())
                continue
            }
        }
        fmt.Fprint(b, v2)
    }
    b.WriteString("}")
    return b.String()
}

现在,当您想要打印struct时,您可以执行以下操作:

fmt.Println(PrintStruct(a, true))

您也可以选择在您的结构中添加String()方法,只需调用我们的PrintStruct()函数:

func (a A) String() string {
    return PrintStruct(a, true)
}

每当您更改结构时,您都不必对String()方法执行任何操作,因为它使用反射来动态遍历所有字段。

备注:

由于我们正在使用反射,您必须导出t time.Time字段才能使用(还添加了一些额外的字段用于测试目的):

type A struct {
    T          time.Time
    I          int
    unexported string
}

测试它:

a := A{time.Now(), 2, "hi!"}
fmt.Println(a)
fmt.Println(PrintStruct(a, true))
fmt.Println(PrintStruct(a, false))
fmt.Println(PrintStruct("I'm not a struct", true))

输出(在Go Playground上尝试):

{T:2009-11-10 23:00:00 +0000 UTC I:2 unexported:hi!}
{T:2009-11-10 23:00:00 +0000 UTC I:2 unexported:hi!}
{2009-11-10 23:00:00 +0000 UTC 2 hi!}
I'm not a struct