-
-
Save millken/b7d0785bbe890076475c09f9ead7e362 to your computer and use it in GitHub Desktop.
Event dispatcher
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package events | |
import ( | |
"sync" | |
"context" | |
) | |
type Dispatcher struct { | |
mutex *sync.RWMutex | |
listeners map[string][]Listener | |
} | |
func NewDispatcher() *Dispatcher { | |
return &Dispatcher{ | |
listeners: make(map[string][]Listener), | |
mutex: new(sync.RWMutex), | |
} | |
} | |
func (this *Dispatcher) AddListener(on string, listener Listener) { | |
this.mutex.Lock() | |
defer this.mutex.Unlock() | |
this.listeners[on] = append(this.listeners[on], listener) | |
} | |
func (this *Dispatcher) Dispatch(ctx context.Context, event Event) error { | |
listeners, ok := this.listeners[event.GetKey()] | |
if !ok { | |
return nil | |
} | |
var wg sync.WaitGroup | |
wg.Add(len(listeners)) | |
errChannel := make(chan error, 1) | |
done := make(chan bool, 1) | |
for _, listener := range listeners { | |
go func() { | |
defer wg.Done() | |
if err := listener.Execute(ctx, event); err != nil { | |
errChannel <- err | |
} | |
}() | |
} | |
go func() { | |
wg.Wait() | |
close(done) | |
}() | |
select { | |
case <-done: | |
case err := <-errChannel: | |
return err | |
} | |
return nil | |
} | |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package events | |
type Event interface { | |
GetKey() string | |
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package events | |
import ( | |
"context" | |
) | |
type Listener interface { | |
Execute(ctx context.Context, event Event) error | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment