如何在Go语言中为共享字段实现联合类型？

Go 本身不支持原生联合类型，但可以通过以下几种无第三方依赖的方式实现你的需求：

方案1：类型断言实现通用Setter函数 #

通过类型断言枚举目标结构体，编写一个通用的设置函数：

package bar

import (
    "fmt"
    "foo"
)

func SetSharedVal(f any, val int) error {
    switch v := f.(type) {
    case *foo.FooA:
        v.SharedVal = val
    case *foo.FooB:
        v.SharedVal = val
    case *foo.FooC:
        v.SharedVal = val
    default:
        return fmt.Errorf("unsupported type: %T", f)
    }
    return nil
}

使用示例：

var a foo.FooA
SetSharedVal(&a, 10) // 直接传入结构体指针

var b foo.FooB
SetSharedVal(&b, 20)

方案2：包装结构体+面向对象方法 #

如果希望用方法调用的形式，可以创建一个包装结构体，为其实现SetSharedVal方法：

package bar

import (
    "fmt"
    "foo"
)

type WrappedFoo struct {
    obj any
}

// NewWrappedFoo 仅允许包装指定的Foo类型指针
func NewWrappedFoo(obj any) (*WrappedFoo, error) {
    switch obj.(type) {
    case *foo.FooA, *foo.FooB, *foo.FooC:
        return &WrappedFoo{obj: obj}, nil
    default:
        return nil, fmt.Errorf("unsupported type: %T", obj)
    }
}

func (w *WrappedFoo) SetSharedVal(val int) error {
    switch v := w.obj.(type) {
    case *foo.FooA:
        v.SharedVal = val
    case *foo.FooB:
        v.SharedVal = val
    case *foo.FooC:
        v.SharedVal = val
    default:
        return fmt.Errorf("unsupported type: %T", w.obj)
    }
    return nil
}

// Get 获取原始结构体指针，方便后续业务操作
func (w *WrappedFoo) Get() any {
    return w.obj
}

使用示例：

a := &foo.FooA{}
wrapped, _ := NewWrappedFoo(a)
wrapped.SetSharedVal(10)

// 如需使用原始对象，可通过类型断言转换
originalA := wrapped.Get().(*foo.FooA)

方案3：反射实现通用处理（适合扩展场景） #

如果后续可能新增带SharedVal字段的结构体，用反射可以避免逐个枚举类型：

package bar

import (
    "fmt"
    "reflect"
)

func SetSharedVal(f any, val int) error {
    valPtr := reflect.ValueOf(f)
    // 必须传入结构体指针
    if valPtr.Kind() != reflect.Ptr {
        return fmt.Errorf("must pass a pointer to struct")
    }
    valElem := valPtr.Elem()
    field := valElem.FieldByName("SharedVal")
    if !field.IsValid() {
        return fmt.Errorf("struct %T has no SharedVal field", f)
    }
    // 校验字段类型是否为int
    if field.Kind() != reflect.Int {
        return fmt.Errorf("SharedVal in %T is not int type", f)
    }
    field.SetInt(int64(val))
    return nil
}

这种方式无需修改代码即可支持所有带int类型SharedVal字段的结构体，但反射存在轻微性能损耗，且编译阶段无法做类型校验。

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