woodpecker/cncd/pubsub/topic.go

46 lines
627 B
Go
Raw Normal View History

2017-03-05 07:56:08 +00:00
package pubsub
import "sync"
type topic struct {
sync.Mutex
name string
2019-04-06 13:44:04 +00:00
done chan struct{}
2017-03-05 07:56:08 +00:00
subs map[*subscriber]struct{}
}
func newTopic(dest string) *topic {
return &topic{
name: dest,
2019-04-06 13:44:04 +00:00
done: make(chan struct{}),
2017-03-05 07:56:08 +00:00
subs: make(map[*subscriber]struct{}),
}
}
func (t *topic) subscribe(s *subscriber) {
t.Lock()
t.subs[s] = struct{}{}
t.Unlock()
}
func (t *topic) unsubscribe(s *subscriber) {
t.Lock()
delete(t.subs, s)
t.Unlock()
}
func (t *topic) publish(m Message) {
t.Lock()
for s := range t.subs {
go s.receiver(m)
}
t.Unlock()
}
func (t *topic) close() {
t.Lock()
close(t.done)
t.Unlock()
}