v0.21.0 scram password auth, cred and token management, browser tls-terminating proxy, viewer client

internal/tokenbucket/bucket.go

@@ -43,6 +43,14 @@ func (tb *TokenBucket) AllowN(n float64) bool {
 	return false
 }
 
+// Refund returns n tokens, never beyond capacity, for an attempt that turned
+// out not to count against the limit
+func (tb *TokenBucket) Refund(n float64) {
+	tb.mu.Lock()
+	defer tb.mu.Unlock()
+	tb.tokens = min(tb.capacity, tb.tokens+n)
+}
+
 // Tokens returns the current number of available tokens
 func (tb *TokenBucket) Tokens() float64 {
 	tb.mu.Lock()
@@ -84,4 +92,4 @@ func (tb *TokenBucket) Capacity() float64 {
 	tb.mu.Lock()
 	defer tb.mu.Unlock()
 	return tb.capacity
-}
+}