chore(pkg/database): use atomic for waitingCount in instrumentedRWMutex

Signed-off-by: Artur Melanchyk <[email protected]>

pkg/database/instrumented_rwmutex.go

@@ -18,34 +18,30 @@ package database
 
 import (
 	"sync"
+	"sync/atomic"
 	"time"
 )
 
 type instrumentedRWMutex struct {
 	rwmutex sync.RWMutex
 
 	trwmutex      sync.RWMutex
-	waitingCount  int
+	waitingCount  int64
 	lastReleaseAt time.Time
 }
 
 func (imux *instrumentedRWMutex) State() (waitingCount int, lastReleaseAt time.Time) {
 	imux.trwmutex.RLock()
 	defer imux.trwmutex.RUnlock()
 
-	return imux.waitingCount, imux.lastReleaseAt
+	return int(atomic.LoadInt64(&imux.waitingCount)), imux.lastReleaseAt
 }
 
 func (imux *instrumentedRWMutex) Lock() {
-	imux.trwmutex.Lock()
-	imux.waitingCount++
-	imux.trwmutex.Unlock()
+	atomic.AddInt64(&imux.waitingCount, 1)
+	defer atomic.AddInt64(&imux.waitingCount, -1)
 
 	imux.rwmutex.Lock()
-
-	imux.trwmutex.Lock()
-	imux.waitingCount--
-	imux.trwmutex.Unlock()
 }
 
 func (imux *instrumentedRWMutex) Unlock() {
@@ -58,15 +54,10 @@ func (imux *instrumentedRWMutex) Unlock() {
 }
 
 func (imux *instrumentedRWMutex) RLock() {
-	imux.trwmutex.Lock()
-	imux.waitingCount++
-	imux.trwmutex.Unlock()
+	atomic.AddInt64(&imux.waitingCount, 1)
+	defer atomic.AddInt64(&imux.waitingCount, -1)
 
 	imux.rwmutex.RLock()
-
-	imux.trwmutex.Lock()
-	imux.waitingCount--
-	imux.trwmutex.Unlock()
 }
 
 func (imux *instrumentedRWMutex) RUnlock() {