Merge pull request #31 from mahboubii/patch-1

Fix: race in memory provider

Author: arbarlow

providers/memory/memory.go

@@ -3,16 +3,21 @@ package memory
 import (
 	"context"
 	"fmt"
+	"sync"
 
 	"github.com/lileio/pubsub/v2"
 )
 
 type MemoryProvider struct {
+	mutex        sync.RWMutex
 	Msgs         map[string][]*pubsub.Msg
 	ErrorHandler func(err error)
 }
 
 func (mp *MemoryProvider) Publish(ctx context.Context, topic string, m *pubsub.Msg) error {
+	mp.mutex.Lock()
+	defer mp.mutex.Unlock()
+
 	if mp.Msgs == nil {
 		mp.Msgs = make(map[string][]*pubsub.Msg, 0)
 	}
@@ -23,6 +28,9 @@ func (mp *MemoryProvider) Publish(ctx context.Context, topic string, m *pubsub.M
 }
 
 func (mp *MemoryProvider) Subscribe(opts pubsub.HandlerOptions, h pubsub.MsgHandler) {
+	mp.mutex.RLock()
+	defer mp.mutex.RUnlock()
+
 	for _, v := range mp.Msgs[opts.Topic] {
 		err := h(context.Background(), *v)