go - 使用外部包中的公共(public)成员处理不同结构的通用函数？

Question

我想编写一个可以将某些字段添加到 Firebase 消息结构的函数。有两种不同类型的消息，Message和MulticastMessage，它们都包含相同的Android和APNS字段类型，但消息类型之间没有明确声明的关系。

我认为我应该能够做到这一点:

type firebaseMessage interface {
    *messaging.Message | *messaging.MulticastMessage
}

func highPriority[T firebaseMessage](message T) T {
    message.Android = &messaging.AndroidConfig{...}
    ....
    return message
}

但它给出错误 message.Android undefined (type T has no field or method Android)。而且我也不能写switch m := message.(type) .

我可以编写 switch m := any(message).(type)，但我仍然不确定这是否会满足我的要求。

我从对联合和类型约束感到困惑的人那里发现了一些其他 SO 问题，但我看不到任何有助于解释为什么这不起作用的答案(可能是因为我试图将它与结构一起使用而不是接口(interface)？)或者联合类型约束实际上有什么用处。

Answer 1

在 Go 1.18 中，您不能访问类型参数的公共(public)字段¹，也不能访问公共(public)方法²。这些特性之所以不能工作，仅仅是因为它们还没有在该语言中可用。如链接线程所示，常见的解决方案是为接口(interface)约束指定方法。

但是 *messaging.Message 和 *messaging.MulticastMessage 类型没有通用的访问器方法，它们是在您无法控制的库包中声明的。

方案一:类型开关

如果联合中的类型较少，这会很好地工作。

func highPriority[T firebaseMessage](message T) T {
    switch m := any(message).(type) {
    case *messaging.Message:
        setConfig(m.Android)
    case *messaging.MulticastMessage:
        setConfig(m.Android)
    }
    return message
}

func setConfig(cfg *messaging.AndroidConfig) {
    // just assuming the config is always non-nil
    *cfg = &messaging.AndroidConfig{}
}

Playground :https://go.dev/play/p/9iG0eSep6Qo

解决方案 2:带方法的包装器

归结为 How to add new methods to an existing type in Go?然后将该方法添加到约束中。如果您有很多结构，它仍然不太理想，但代码生成可能会有所帮助:

type wrappedMessage interface {
    *MessageWrapper | *MultiCastMessageWrapper
    SetConfig(c foo.Config)
}

type MessageWrapper struct {
    messaging.Message
}

func (w *MessageWrapper) SetConfig(cfg messaging.Android) {
    *w.Android = cfg
}

// same for MulticastMessageWrapper

func highPriority[T wrappedMessage](message T) T {
    // now you can call this common method 
    message.SetConfig(messaging.Android{"some-value"})
    return message
}

Playground :https://go.dev/play/p/JUHp9Fu27Yt

方案三:反射

如果你有很多结构，你可能最好使用反射。在这种情况下，类型参数不是严格需要的，但有助于提供额外的类型安全。请注意，结构和字段必须是 addressable为了这个工作。

func highPriority[T firebaseMessage](message T) T {
    cfg := &messaging.Android{} 
    reflect.ValueOf(message).Elem().FieldByName("Android").Set(reflect.ValueOf(cfg))
    return message
}

Playground :https://go.dev/play/p/3DbIADhiWdO

注意事项:

1. How can I define a struct field in my interface as a type constraint (type T has no field or method)?
2. In Go generics, how to use a common method for types in a union constraint?