You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.6 KiB
74 lines
1.6 KiB
package dispatcher
|
|
|
|
import (
|
|
"github.com/wagslane/go-rabbitmq/internal/logger"
|
|
"math"
|
|
"math/rand"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// Dispatcher -
|
|
type Dispatcher struct {
|
|
subscribers map[int]dispatchSubscriber
|
|
subscribersMux *sync.Mutex
|
|
logger logger.Logger
|
|
}
|
|
|
|
type dispatchSubscriber struct {
|
|
notifyCancelOrCloseChan chan error
|
|
closeCh <-chan struct{}
|
|
}
|
|
|
|
// NewDispatcher -
|
|
func NewDispatcher(logger logger.Logger) *Dispatcher {
|
|
return &Dispatcher{
|
|
subscribers: make(map[int]dispatchSubscriber),
|
|
subscribersMux: &sync.Mutex{},
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// Dispatch -
|
|
func (d *Dispatcher) Dispatch(err error) error {
|
|
d.subscribersMux.Lock()
|
|
defer d.subscribersMux.Unlock()
|
|
for _, subscriber := range d.subscribers {
|
|
select {
|
|
case <-time.After(time.Second * 5):
|
|
d.logger.Errorf("Unexpected rabbitmq error: timeout in dispatch")
|
|
case subscriber.notifyCancelOrCloseChan <- err:
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// AddSubscriber -
|
|
func (d *Dispatcher) AddSubscriber() (<-chan error, chan<- struct{}) {
|
|
const maxRand = math.MaxInt
|
|
const minRand = 0
|
|
id := rand.Intn(maxRand-minRand) + minRand
|
|
|
|
closeCh := make(chan struct{})
|
|
notifyCancelOrCloseChan := make(chan error)
|
|
|
|
d.subscribersMux.Lock()
|
|
d.subscribers[id] = dispatchSubscriber{
|
|
notifyCancelOrCloseChan: notifyCancelOrCloseChan,
|
|
closeCh: closeCh,
|
|
}
|
|
d.subscribersMux.Unlock()
|
|
|
|
go func(id int) {
|
|
<-closeCh
|
|
d.subscribersMux.Lock()
|
|
defer d.subscribersMux.Unlock()
|
|
sub, ok := d.subscribers[id]
|
|
if !ok {
|
|
return
|
|
}
|
|
close(sub.notifyCancelOrCloseChan)
|
|
delete(d.subscribers, id)
|
|
}(id)
|
|
return notifyCancelOrCloseChan, closeCh
|
|
}
|