通过注入类型查找切片元素的模式

时间:2018-08-05 12:58:14

标签: go interface type-assertion

我尝试使用对象的类型在接口切片内查找对象。我目前的解决方案如下:

package main

import (
    "errors"
    "fmt"
)

type Entity struct {
    children []Childable
}

func (e *Entity) ChildByInterface(l interface{}) (Childable, error) {
    for _, c := range e.children {
        if fmt.Sprintf("%T", c) == fmt.Sprintf("%T", l) {
            return c, nil
        }
    }
    return nil, errors.New("child doesn't exist")
}

type Childable interface {
    GetName() string
}

func main() {
    ent := &Entity{
        []Childable{
            &Apple{name: "Appy"},
            &Orange{name: "Orry"},
            // more types can by introduced based on build tags
        },
    }

    appy, err := ent.ChildByInterface(&Apple{})
    if err != nil {
        fmt.Println(err)
    } else {
        appy.(*Apple).IsRed()
        fmt.Printf("%+v", appy)
    }
}

type Apple struct {
    name string
    red  bool
}

func (a *Apple) GetName() string {
    return a.name
}

func (a *Apple) IsRed() {
    a.red = true
}

type Orange struct {
    name   string
    yellow bool
}

func (o *Orange) GetName() string {
    return o.name
}

func (o *Orange) IsYellow() {
    o.yellow = true
}

https://play.golang.org/p/FmkWILBqqA-

更多可使用的类型(Apple,Orange等)可以使用构建标记注入。因此,为了确保查找类型的安全并避免错误,我将interface{}传递给查找函数。 Childable接口还确保新注入的类型实现正确的功能。

这是事情开始变得混乱的地方。目前,我正在对接口的类型和Childable对象的类型进行字符串比较,以查看它们是否匹配: fmt.Sprintf("%T", c) == fmt.Sprintf("%T", l)

然后,我仍然只能返回Childable接口。因此,我必须使用类型断言来获取正确的类型:appy.(*Apple)

为了使孩子保持其正确的类型而进行的电镀非常繁琐,而查找匹配项的字符串比较会对性能产生重大影响。我可以使用哪种更好的解决方案将两个接口相互匹配以避免性能下降?

1 个答案:

答案 0 :(得分:1)

到目前为止,fmt.Sprintf("%T", c)在幕后使用reflect并没有暗示它的优点-最好直接使用reflect。 您可以将引用参数用作结果的占位符,而不是返回值。

func (e *Entity) ChildByInterface(l Childable) error {
    for _, c := range e.children {
        if reflect.TypeOf(l) == reflect.TypeOf(c) {
            fmt.Println(c)
            reflect.ValueOf(l).Elem().Set(reflect.ValueOf(c).Elem())
            return nil
        }
    }
    return errors.New("child doesn't exist")
}

现在通过一个占位符

apple := &Apple{}
err := ent.ChildByInterface(apple)
//and use it
apple.IsRed()

Working code