go - 结构类型作为映射键

问题描述

我们有以下功能：

func (h *Handler) Handle(message interface{}) error {
    //here there is a switch for different messages
    switch m := message.(type) {
    }
}

此签名已给出且无法更改。处理程序处理大约 20 种不同的消息类型。

现在，其中一些消息（大约 4 条）需要特殊的后处理。在不同的包装中。

因此，我想这样做：

 func (h *Handler) Handle(message interface{}) error {
        //here there is a switch for different messages

        switch m := message.(type) {
        }
        //only post-process if original message processing succeeds
        postProcessorPkg.Process(message)
    }

现在，在Process函数中，我想快速查找消息类型是否确实是我们需要后处理的类型。我不想在switch这里再做一次。有许多处理程序，在不同的包中，具有不同数量的消息类型，它应该是通用的。

所以我想在后处理器中注册消息类型，然后进行查找：

func (p *Postprocessor) Register(msgtype interface{}) {
     registeredTypes[msgtype] = msgtype
}

接着

func (p *Postprocessor) Process(msgtype interface{}) error {
     if ok := registeredTypes[msgtype]; !ok {
        return errors.New("Unsupported message type")
     }
     prop := GetProp(registeredTypes[msgtype])
     doSmthWithProp(prop)
}

现在这一切都行不通了，因为据我所知，我只能“注册”消息的实例，而不是消息类型本身。因此，地图只会匹配消息的特定实例，而不是它的类型，这正是我所需要的。

所以我想这需要重新设计。我可以完全放弃注册和地图查找，但是

我无法将Handle函数更改为特定类型（需要保留签名message interface{}
我想避免不得不使用reflect，因为我将很难与一些同事捍卫这样的解决方案。

标签： gotypesinterfacego-map

由于无法将类型设置为映射键，因此我最终决定实现以下解决方案，该解决方案基于 @Chrono Kitsune 的解决方案：

type Postprocess interface {
    NeedsPostprocess() bool
}

type MsgWithPostProcess struct {}

func (p *MsgWithPostProcess) NeedsPostprocess() bool {
  return true
}

type Msg1 struct {
   MsgWithPostProcess
   //other stuff
}

type Msg2 struct {
    MsgWithPostProcess
    //other stuff
}

type Msg3 struct {
    //no postprocessing needed
}

func (p *Postprocessor) Process(msgtype interface{}) error {
     if _, ok := msgtype.(Postprocess); ok {
        //do postprocessing
     }         
}

在我的简单测试中，我只做了Msg1并且Msg2将被后处理，但不是Msg3，这正是我想要的。

go - 结构类型作为映射键

问题描述

解决方案

推荐阅读