如何用反射实现结构体接口字段的注册赋值以消除switch分支？

完全可以通过Go的反射机制实现相同功能，替代switch分支逻辑。不过需要先注意一个前提：必须将Aggregation结构体的字段改为导出字段（大写开头）——因为Go的反射无法访问未导出的小写字段，这是语言的访问控制限制。

实现思路 #

1. 通过反射获取传入参数的类型，确认其为接口类型；
2. 遍历Aggregation结构体的所有字段，对比字段类型与传入参数的类型；
3. 找到匹配的字段后，利用反射为该字段赋值。

代码实现 #

首先修改结构体（将字段改为导出）：

type Aggregation struct {
    A InterfaceA
    B InterfaceB
    // ...
    N InterfaceN
}

然后实现反射版的Register方法：

func (a *Aggregation) Register(i interface{}) *Aggregation {
    inputType := reflect.TypeOf(i)
    // 仅处理接口类型的输入
    if inputType.Kind() != reflect.Interface {
        return a
    }

    aggVal := reflect.ValueOf(a).Elem()
    // 遍历结构体所有字段
    for j := 0; j < aggVal.NumField(); j++ {
        field := aggVal.Field(j)
        // 匹配字段类型与输入接口类型
        if field.Type() == inputType {
            field.Set(reflect.ValueOf(i))
            break // 找到匹配项后退出循环
        }
    }
    return a
}

注意事项 #

• 如果传入的参数不是接口类型，方法会直接返回，不做任何赋值操作；
• 若存在多个同类型的接口字段，代码会为第一个匹配的字段赋值；
• 反射操作带来的性能开销极小，在大多数场景下可以忽略，但对极致性能要求的场景，原switch分支会更高效。

内容的提问来源于stack exchange，提问作者Andrey Dunaev