feat: use mutually exclusive lock for pub/sub

1 files changed, 19 insertions, 1 deletions

diff --git a/pkg/event/aggregator.go b/pkg/event/aggregator.go
index 996a550..7ff5061 100644
--- a/pkg/event/aggregator.go
+++ b/pkg/event/aggregator.go
@@ -1,11 +1,23 @@
 package event
 
-import "github.com/xlgmokha/x/pkg/x"
+import (
+	"sync"
+
+	"github.com/xlgmokha/x/pkg/x"
+)
 
 type Aggregator struct {
+	mu            sync.RWMutex
 	subscriptions map[Event][]Subscription
 }
 
+func WithDefaults() x.Option[*Aggregator] {
+	return x.With(func(item *Aggregator) {
+		item.mu = sync.RWMutex{}
+		item.subscriptions = map[Event][]Subscription{}
+	})
+}
+
 func WithoutSubscriptions() x.Option[*Aggregator] {
 	return WithSubscriptions(map[Event][]Subscription{})
 }
@@ -17,10 +29,16 @@ func WithSubscriptions(subscriptions map[Event][]Subscription) x.Option[*Aggrega
 }
 
 func (a *Aggregator) Subscribe(event Event, f Subscription) {
+	a.mu.Lock()
+	defer a.mu.Unlock()
+
 	a.subscriptions[event] = append(a.subscriptions[event], f)
 }
 
 func (a *Aggregator) Publish(event Event, message any) {
+	a.mu.RLock()
+	defer a.mu.RUnlock()
+
 	for _, subscription := range a.subscriptions[event] {
 		subscription(message)
 	}