fix concurrency issue in SumComputeTime (#2030)

ten-protocol · Aug 19, 2024 · 6d47054 · 6d47054
1 parent f9c3978
commit 6d47054
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/walletextension/ratelimiter/rate_limiter.go b/tools/walletextension/ratelimiter/rate_limiter.go
@@ -104,6 +104,9 @@ func (rl *RateLimiter) SumComputeTime(userID common.Address) time.Duration {
 
 	var totalComputeTime time.Duration
 	if user, exists := rl.users[userID]; exists {
+		user.mu.RLock() // lock the user to prevent changes while reading
+		defer user.mu.RUnlock()
+
 		cutoff := time.Now().Add(-rl.window)
 		for _, interval := range user.CurrentRequests {
 			// if the request has ended and it's within the window, add the compute time