fix: Prevent ring index overflow

zeromicro · Nov 23, 2023 · c579f6f · c579f6f
1 parent 2281aab
commit c579f6f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/core/collection/ring.go b/core/collection/ring.go
@@ -1,6 +1,9 @@
 package collection
 
-import "sync"
+import (
+	"math"
+	"sync"
+)
 
 // A Ring can be used as fixed size ring.
 type Ring struct {
@@ -26,7 +29,7 @@ func (r *Ring) Add(v any) {
 	defer r.lock.Unlock()
 
 	r.elements[r.index%len(r.elements)] = v
-	r.index++
+	r.index = (r.index + 1) % math.MaxInt32 // prevent ring index overflow
 }
 
 // Take takes all items from r.