feat: safe websocket server implementation

internal/mutex.go

@@ -0,0 +1,33 @@
+package internal
+
+import (
+	"sync"
+)
+
+type CustomRwMutex struct {
+	mu *sync.RWMutex
+}
+
+func NewCustomRwMutex() *CustomRwMutex {
+	return &CustomRwMutex{
+		mu: &sync.RWMutex{},
+	}
+}
+
+func (rwMu *CustomRwMutex) WriteHandler(fn func() error) error {
+	rwMu.mu.Lock()
+	defer rwMu.mu.Unlock()
+	if err := fn(); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (rwMu *CustomRwMutex) ReadHandler(fn func() error) error {
+	rwMu.mu.RLock()
+	defer rwMu.mu.RUnlock()
+	if err := fn(); err != nil {
+		return err
+	}
+	return nil
+}