判断struct的interface-type字段是否设置

PHP小编小新在这里分享一个关于判断struct的interface-type字段是否设置的技巧。在Go语言中，struct类型可以实现多个接口，通过判断interface-type字段是否设置，我们可以轻松判断一个struct是否实现了某个特定的接口。这个技巧非常实用，可以在代码中准确判断对象的类型，从而进行相应的处理。接下来，我们一起来看看具体的实现方法吧！

问题内容

给定一个结构体，其字段属于接口类型：

type a interface { foo() }
type b interface { bar() }

type container struct {
   fielda a
   fieldb b
   ...
}

以及实现这些接口的结构：

type a struct {}
func (*a) foo() {}
func newa() *a { return &a{} }

type b struct {}
func (*b) bar() {}
func newb() *b { return &b{} }

以及创建 container 实例但未显式设置所有字段的代码：

c := &container{}
c.fielda = newa()
// c.fieldb = newb()  // <-- fieldb not explicitly set

使用反射，如何检测未显式设置的字段？

在运行时检查 fieldb 的值，vscode 愉快地报告该值为 nil。同样，尝试调用 c.fieldb.bar() 将会因 nil 指针取消引用而发生恐慌。但反射不允许您测试 isnil，iszero 也不会返回 true：

func validateContainer(c *container) {
    tC := reflect.TypeOf(*c)
    for i := 0; i < tC.NumField(); i++ {
        reflect.ValueOf(tC.Field(i)).IsNil() // panics
        reflect.ValueOf(tC.Field(i)).IsZero() // always false
        reflect.ValueOf(tC.Field(i)).IsValid() // always true
    }
}

解决方法

您应该检查 reflect.valueof(*c) 的字段，而不是 reflect.typeof(*c)。

package main

import (
    "fmt"
    "reflect"
)

func validatecontainer(c *container) {
    tc := reflect.typeof(*c)
    vc := reflect.valueof(*c)
    for i := 0; i < vc.numfield(); i++ {
        if f := vc.field(i); f.kind() == reflect.interface {
            fmt.printf("%v: isnil: %v, iszero: %v, isvalid: %v\n",
                tc.field(i).name,
                f.isnil(),
                f.iszero(),
                f.isvalid(),
            )
        }
    }

    // tc.field(i) returns a reflect.structfield that describes the field.
    // it's not the field itself.
    fmt.printf("%#v\n", reflect.valueof(tc.field(0)))
}

type a interface{ foo() }
type b interface{ bar() }

type container struct {
    fielda a
    fieldb b
}

type a struct{}

func (*a) foo() {}
func newa() *a  { return &a{} }

func main() {
    c := &container{}
    c.fielda = newa()
    validatecontainer(c)
}

输出：

FieldA: IsNil: false, isZero: false, IsValid: true
FieldB: IsNil: true, isZero: true, IsValid: true
reflect.StructField{Name:"FieldA", PkgPath:"", Type:(*reflect.rtype)(0x48de20), Tag:"", Offset:0x0, Index:[]int{0}, Anonymous:false}

以上就是判断struct的interface-type字段是否设置的详细内容，更多请关注编程网其它相关文章！