66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package bus
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
)
|
|
|
|
type MessageBus struct {
|
|
inbound chan InboundMessage
|
|
outbound chan OutboundMessage
|
|
handlers map[string]MessageHandler
|
|
mu sync.RWMutex
|
|
}
|
|
|
|
func NewMessageBus() *MessageBus {
|
|
return &MessageBus{
|
|
inbound: make(chan InboundMessage, 100),
|
|
outbound: make(chan OutboundMessage, 100),
|
|
handlers: make(map[string]MessageHandler),
|
|
}
|
|
}
|
|
|
|
func (mb *MessageBus) PublishInbound(msg InboundMessage) {
|
|
mb.inbound <- msg
|
|
}
|
|
|
|
func (mb *MessageBus) ConsumeInbound(ctx context.Context) (InboundMessage, bool) {
|
|
select {
|
|
case msg := <-mb.inbound:
|
|
return msg, true
|
|
case <-ctx.Done():
|
|
return InboundMessage{}, false
|
|
}
|
|
}
|
|
|
|
func (mb *MessageBus) PublishOutbound(msg OutboundMessage) {
|
|
mb.outbound <- msg
|
|
}
|
|
|
|
func (mb *MessageBus) SubscribeOutbound(ctx context.Context) (OutboundMessage, bool) {
|
|
select {
|
|
case msg := <-mb.outbound:
|
|
return msg, true
|
|
case <-ctx.Done():
|
|
return OutboundMessage{}, false
|
|
}
|
|
}
|
|
|
|
func (mb *MessageBus) RegisterHandler(channel string, handler MessageHandler) {
|
|
mb.mu.Lock()
|
|
defer mb.mu.Unlock()
|
|
mb.handlers[channel] = handler
|
|
}
|
|
|
|
func (mb *MessageBus) GetHandler(channel string) (MessageHandler, bool) {
|
|
mb.mu.RLock()
|
|
defer mb.mu.RUnlock()
|
|
handler, ok := mb.handlers[channel]
|
|
return handler, ok
|
|
}
|
|
|
|
func (mb *MessageBus) Close() {
|
|
close(mb.inbound)
|
|
close(mb.outbound)
|
|
}
|