Expose the current Limit() on existing ratelimiters

common/persistence/wrappers/ratelimited/utils_test.go

@@ -25,6 +25,8 @@ package ratelimited
 import (
 	"context"
 
+	"golang.org/x/time/rate"
+
 	"github.com/uber/cadence/common/clock"
 	"github.com/uber/cadence/common/quotas"
 )
@@ -43,6 +45,10 @@ func (l limiterAlwaysAllow) Reserve() clock.Reservation {
 	return &reservationAlwaysAllow{}
 }
 
+func (l limiterAlwaysAllow) Limit() rate.Limit {
+	return rate.Inf
+}
+
 type limiterNeverAllow struct{}
 
 func (l limiterNeverAllow) Allow() bool {
@@ -58,6 +64,10 @@ func (l limiterNeverAllow) Reserve() clock.Reservation {
 	return &reservationNeverAllow{}
 }
 
+func (l limiterNeverAllow) Limit() rate.Limit {
+	return 0
+}
+
 type reservationAlwaysAllow struct{}
 type reservationNeverAllow struct{}
 

common/quotas/dynamicratelimiter.go

@@ -25,6 +25,8 @@
 import (
 	"context"
 
+	"golang.org/x/time/rate"
+
 	"github.com/uber/cadence/common/clock"
 )
 
@@ -63,3 +65,7 @@
 	d.rl.UpdateMaxDispatch(&rps)
 	return d.rl.Reserve()
 }
+
+func (d *DynamicRateLimiter) Limit() rate.Limit {
+	return rate.Limit(d.rps())
+}

common/quotas/global/collection/internal/counted.go

@@ -26,6 +26,7 @@
 	"context"
 
 	"go.uber.org/atomic"
+	"golang.org/x/time/rate"
 
 	"github.com/uber/cadence/common/clock"
 	"github.com/uber/cadence/common/quotas"
@@ -80,6 +81,10 @@
 	}
 }
 
+func (c CountedLimiter) Limit() rate.Limit {
+	return c.wrapped.Limit()
+}
+
 func (c CountedLimiter) Collect() UsageMetrics {
 	return c.usage.Collect()
 }

common/quotas/global/collection/internal/fallback.go

@@ -209,6 +209,13 @@
 	}
 }
 
+func (b *FallbackLimiter) Limit() rate.Limit {
+	if b.useFallback() {
+		return b.fallback.Limit()
+	}
+	return b.primary.Limit()
+}
+
 func (b *FallbackLimiter) both() quotas.Limiter {
 	if b.useFallback() {
 		return NewShadowedLimiter(b.fallback, b.primary)

common/quotas/global/collection/internal/fallback_test.go

@@ -187,6 +187,7 @@ type allowres struct{}
 func (allowlimiter) Allow() bool                  { return true }
 func (a allowlimiter) Wait(context.Context) error { return nil }
 func (a allowlimiter) Reserve() clock.Reservation { return allowres{} }
+func (a allowlimiter) Limit() rate.Limit          { return rate.Inf }
 
 func (a allowres) Allow() bool { return true }
 func (a allowres) Used(bool)   {}

common/quotas/global/collection/internal/shadowed.go

@@ -25,6 +25,8 @@
 import (
 	"context"
 
+	"golang.org/x/time/rate"
+
 	"github.com/uber/cadence/common/clock"
 	"github.com/uber/cadence/common/quotas"
 )
@@ -79,6 +81,10 @@
 	}
 }
 
+func (s shadowedLimiter) Limit() rate.Limit {
+	return s.primary.Limit()
+}
+
 func (s shadowedReservation) Allow() bool {
 	_ = s.shadow.Allow()
 	return s.primary.Allow()

common/quotas/interfaces.go

@@ -25,6 +25,8 @@ package quotas
 import (
 	"context"
 
+	"golang.org/x/time/rate"
+
 	"github.com/uber/cadence/common/clock"
 )
 
@@ -56,6 +58,12 @@ type Limiter interface {
 
 	// Reserve reserves a rate limit token
 	Reserve() clock.Reservation
+
+	// Limit returns the current configured ratelimit.
+	//
+	// If this Limiter wraps multiple values, this is generally the "most relevant" one,
+	// i.e. the one that is most likely to apply to the next request
+	Limit() rate.Limit
 }
 
 // Policy corresponds to a quota policy. A policy allows implementing layered

common/quotas/ratelimiter.go

@@ -22,9 +22,9 @@
 
 import (
 	"context"
-	"math"
 	"sync"
 	"sync/atomic"
+	"testing"
 	"time"
 
 	"golang.org/x/time/rate"
@@ -58,8 +58,9 @@
 }
 
 // NewSimpleRateLimiter returns a new rate limiter backed by the golang rate
-// limiter
-func NewSimpleRateLimiter(rps int) *RateLimiter {
+// limiter.  This is currently only used in tests.
+func NewSimpleRateLimiter(t *testing.T, rps int) *RateLimiter {
+	t.Helper() // ensure a T has been passed
 	initialRps := float64(rps)
 	return NewRateLimiter(&initialRps, _defaultRPSTTL, _burstSize)
 }
@@ -107,13 +108,13 @@
 }
 
 // Limit returns the current rate per second limit for this ratelimiter
-func (rl *RateLimiter) Limit() float64 {
+func (rl *RateLimiter) Limit() rate.Limit {
 	rl.RLock()
 	defer rl.RUnlock()
 	if rl.maxDispatchPerSecond != nil {
-		return *rl.maxDispatchPerSecond
+		return rate.Limit(*rl.maxDispatchPerSecond)
 	}
-	return math.MaxFloat64
+	return rate.Inf
 }
 
 func (rl *RateLimiter) storeLimiter(maxDispatchPerSecond *float64) {

service/matching/tasklist/matcher.go

@@ -436,7 +436,7 @@ func (tm *TaskMatcher) UpdateRatelimit(rps *float64) {
 
 // Rate returns the current rate at which tasks are dispatched
 func (tm *TaskMatcher) Rate() float64 {
-	return tm.limiter.Limit()
+	return float64(tm.limiter.Limit())
 }
 
 func (tm *TaskMatcher) pollOrForward(

service/worker/archiver/client_test.go

@@ -70,9 +70,9 @@ func (s *clientSuite) SetupTest() {
 		log.NewNoop(),
 		nil,
 		dynamicconfig.GetIntPropertyFn(1000),
-		quotas.NewSimpleRateLimiter(1000),
-		quotas.NewSimpleRateLimiter(1),
-		quotas.NewSimpleRateLimiter(1),
+		quotas.NewSimpleRateLimiter(s.T(), 1000),
+		quotas.NewSimpleRateLimiter(s.T(), 1),
+		quotas.NewSimpleRateLimiter(s.T(), 1),
 		s.archiverProvider,
 		dynamicconfig.GetBoolPropertyFn(false),
 	).(*client)