mirror of https://github.com/v2ray/v2ray-core
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.
47 lines
663 B
47 lines
663 B
package event
|
|
|
|
import "sync"
|
|
|
|
type Event uint16
|
|
|
|
type Handler func(data interface{}) error
|
|
|
|
type Registry interface {
|
|
On(Event, Handler)
|
|
}
|
|
|
|
type Listener struct {
|
|
sync.RWMutex
|
|
events map[Event][]Handler
|
|
}
|
|
|
|
func (l *Listener) On(e Event, h Handler) {
|
|
l.Lock()
|
|
defer l.Unlock()
|
|
|
|
if l.events == nil {
|
|
l.events = make(map[Event][]Handler)
|
|
}
|
|
|
|
handlers := l.events[e]
|
|
handlers = append(handlers, h)
|
|
l.events[e] = handlers
|
|
}
|
|
|
|
func (l *Listener) Fire(e Event, data interface{}) error {
|
|
l.RLock()
|
|
defer l.RUnlock()
|
|
|
|
if l.events == nil {
|
|
return nil
|
|
}
|
|
|
|
for _, h := range l.events[e] {
|
|
if err := h(data); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|