harness-drone/eventbus/builtin/bus.go

package builtin

import (
	"sync"

	"github.com/drone/drone/eventbus"
)

type EventBus struct {
	sync.Mutex
	subs map[chan *eventbus.Event]bool
}

// New creates a new EventBus that manages a list of
// subscribers to which events are published.
func New() *EventBus {
	return &EventBus{
		subs: make(map[chan *eventbus.Event]bool),
	}
}

// Subscribe adds the channel to the list of
// subscribers. Each subscriber in the list will
// receive broadcast events.
func (b *EventBus) Subscribe(c chan *eventbus.Event) {
	b.Lock()
	b.subs[c] = true
	b.Unlock()
}

// Unsubscribe removes the channel from the
// list of subscribers.
func (b *EventBus) Unsubscribe(c chan *eventbus.Event) {
	b.Lock()
	delete(b.subs, c)
	b.Unlock()
}

// Send dispatches a message to all subscribers.
func (b *EventBus) Send(event *eventbus.Event) {
	b.Lock()
	defer b.Unlock()

	for s := range b.subs {
		go func(c chan *eventbus.Event) {
			defer recover()
			c <- event
		}(s)
	}
}