Fix rate limiter panic (#125)

jaegertracing · Mar 23, 2017 · 8cacc0e · 8cacc0e
1 parent 54d9b0c
commit 8cacc0e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/utils/rate_limiter.go b/utils/rate_limiter.go
@@ -20,14 +20,19 @@
 
 package utils
 
-import "time"
+import (
+	"sync"
+	"time"
+)
 
 // RateLimiter is a filter used to check if a message that is worth itemCost units is within the rate limits.
 type RateLimiter interface {
 	CheckCredit(itemCost float64) bool
 }
 
 type rateLimiter struct {
+	sync.Mutex
+
 	creditsPerSecond float64
 	balance          float64
 	maxBalance       float64
@@ -58,6 +63,8 @@ func NewRateLimiter(creditsPerSecond, maxBalance float64) RateLimiter {
 }
 
 func (b *rateLimiter) CheckCredit(itemCost float64) bool {
+	b.Lock()
+	defer b.Unlock()
 	// calculate how much time passed since the last tick, and update current tick
 	currentTime := b.timeNow()
 	elapsedTime := currentTime.Sub(b.lastTick)