Use standard lock

PostHog · Dec 16, 2024 · c81ee05 · c81ee05
1 parent 0d0f64b
commit c81ee05
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/util.go b/util.go
@@ -7,7 +7,7 @@ import (
 type SizeLimitedMap struct {
 	ids  map[string][]string
 	size int
-	mu   sync.RWMutex
+	mu   sync.Mutex
 }
 
 func newSizeLimitedMap(size int) *SizeLimitedMap {
@@ -35,8 +35,8 @@ func (sizeLimitedMap *SizeLimitedMap) add(key string, element string) {
 }
 
 func (sizeLimitedMap *SizeLimitedMap) contains(key string, element string) bool {
-	sizeLimitedMap.mu.RLock()
-	defer sizeLimitedMap.mu.RUnlock()
+	sizeLimitedMap.mu.Lock()
+	defer sizeLimitedMap.mu.Unlock()
 
 	if val, ok := sizeLimitedMap.ids[key]; ok {
 		for _, v := range val {
@@ -50,8 +50,8 @@ func (sizeLimitedMap *SizeLimitedMap) contains(key string, element string) bool
 }
 
 func (sizeLimitedMap *SizeLimitedMap) count() int {
-	sizeLimitedMap.mu.RLock()
-	defer sizeLimitedMap.mu.RUnlock()
+	sizeLimitedMap.mu.Lock()
+	defer sizeLimitedMap.mu.Unlock()
 
 	return len(sizeLimitedMap.ids)
 }