cap counters at capacity

slidingwindow.go

@@ -3,6 +3,7 @@ package limiters
 import (
 	"context"
 	"fmt"
+	"math"
 	"strconv"
 	"sync"
 	"time"
@@ -57,6 +58,9 @@ func (s *SlidingWindow) Limit(ctx context.Context) (time.Duration, error) {
 		return 0, err
 	}
 
+	prev = int64(math.Min(float64(prev), float64(s.capacity+1)))
+	curr = int64(math.Min(float64(curr), float64(s.capacity+1)))
+
 	total := float64(prev*int64(ttl))/float64(s.rate) + float64(curr)
 	if total-float64(s.capacity) >= s.epsilon {
 		var wait time.Duration

slidingwindow_test.go

@@ -308,6 +308,30 @@ func (s *LimitersTestSuite) TestSlidingWindowOverflowAndWait() {
 	}
 }
 
+func (s *LimitersTestSuite) TestSlidingWindowOverflowAndNoWait() {
+	capacity := int64(10)
+	clock := newFakeClock()
+	for name, bucket := range s.slidingWindows(capacity, time.Second, clock, 1e-9) {
+		s.Run(name, func() {
+			clock.reset()
+			for i := int64(0); i < capacity; i++ {
+				w, err := bucket.Limit(context.TODO())
+				s.Require().NoError(err)
+				s.Require().Equal(time.Duration(0), w)
+			}
+			w, err := bucket.Limit(context.TODO())
+			s.Require().Equal(l.ErrLimitExhausted, err)
+			expected := clock.Now().Add(w)
+			for i := int64(0); i < capacity; i++ {
+				w, err = bucket.Limit(context.TODO())
+				s.Require().Equal(l.ErrLimitExhausted, err)
+				actual := clock.Now().Add(w)
+				s.Require().Equal(expected, actual, i)
+			}
+		})
+	}
+}
+
 func BenchmarkSlidingWindows(b *testing.B) {
 	s := new(LimitersTestSuite)
 	s.SetT(&testing.T{})