Go 언어로 동시성 높은 메시지 미들웨어를 구현하는 방법

type MessageQueue struct {
    messages chan interface{}
}

func NewMessageQueue(size int) *MessageQueue {
    return &MessageQueue{
        messages: make(chan interface{}, size),
    }
}

func (mq *MessageQueue) Push(msg interface{}) {
    mq.messages <- msg
}

func (mq *MessageQueue) Pop() interface{} {
    return <-mq.messages
}

type MessageHandler struct {
    queue *MessageQueue
    stop  chan bool
}

func NewMessageHandler(queue *MessageQueue) *MessageHandler {
    return &MessageHandler{
        queue: queue,
        stop:  make(chan bool),
    }
}

func (mh *MessageHandler) Start() {
    go func() {
        for {
            select {
            case msg := <-mh.queue.messages:
                // 处理消息
                fmt.Println("Handle message:", msg)
            case <-mh.stop:
                return
            }
        }
    }()
}

func (mh *MessageHandler) Stop() {
    mh.stop <- true
}

type Router struct {
    handlers map[string]*MessageHandler
}

func NewRouter() *Router {
    return &Router{
        handlers: make(map[string]*MessageHandler),
    }
}

func (r *Router) RegisterHandler(topic string, handler *MessageHandler) {
    r.handlers[topic] = handler
}

func (r *Router) Dispatch(topic string, msg interface{}) {
    handler, ok := r.handlers[topic]
    if ok {
        handler.queue.Push(msg)
    }
}

func main() {
    queue := NewMessageQueue(100)
    handler := NewMessageHandler(queue)
    router := NewRouter()

    // 注册消息处理器到路由分发器
    router.RegisterHandler("topic1", handler)
    router.RegisterHandler("topic2", handler)

    // 启动消息处理器
    handler.Start()

    // 分发消息到对应的处理器
    router.Dispatch("topic1", "message1")
    router.Dispatch("topic2", "message2")

    // 停止消息处理器
    handler.Stop()
}