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

我想编写一个可以将某些字段添加到 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 问题,但我看不到任何有助于解释为什么这不起作用的答案(可能是因为我试图将它与结构而不是接口一起使用?)或者什么联合类型约束实际上有用。


繁星淼淼
浏览 91回答 1
1回答

汪汪一只猫

在 Go 1.18 中,您不能访问公共字段1,也不能访问类型参数的公共方法2。这些特性之所以不能工作,仅仅是因为它们还没有在该语言中可用。如链接线程所示,常见的解决方案是为接口约束指定方法。但是,类型*messaging.Message和*messaging.MulticastMessage没有通用的访问器方法,并且在您无法控制的库包中声明。解决方案1:类型开关如果联合中的类型较少,则此方法可以正常工作。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{}}游乐场:https ://go.dev/play/p/9iG0eSep6Qo解决方案 2:包装方法这归结为如何在 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 MulticastMessageWrapperfunc highPriority[T wrappedMessage](message T) T {    // now you can call this common method     message.SetConfig(messaging.Android{"some-value"})    return message}游乐场: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
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Go