使用外部包中的公共成员处理不同结构的通用函数？-ICode联盟

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

IT小君 2022-11-12T11:57:12

我想编写一个可以将某些字段添加到 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)（cannot use type switch on type parameter value message (variable of type T constrained by firebaseMessage)）。

我可以写switch m := any(message).(type)，但我仍然不确定这是否会达到我想要的效果。

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

点击广告，支持我们为你提供更好的服务

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{} }

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 }

IT小君

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

但是，这些类型*messaging.Message并*messaging.MulticastMessage没有公共访问器方法，而是在您无法控制的库包中声明。

解决方案一：类型开关

如果联合中有少量类型，这很好用。

游乐场：https ://go.dev/play/p/9iG0eSep6Qo

解决方案 2：包装方法

这归结为如何在 Go 中向现有类型添加新方法？然后将该方法添加到约束中。如果您有许多结构，它仍然不太理想，但代码生成可能会有所帮助：

游乐场：https ://go.dev/play/p/JUHp9Fu27Yt

方案三：反射

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

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

游乐场：https ://go.dev/play/p/3DbIADhiWdO

笔记：

2022-11-12T11:57:12 回复