feat: add Reset to the LeakyBucketStateBackend and TokenBucketStateBa…

…ckend
mennanov · Oct 26, 2024 · 3ac06eb · 3ac06eb
1 parent 048d5e1
commit 3ac06eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/leakybucket.go b/leakybucket.go
@@ -38,6 +38,8 @@ type LeakyBucketStateBackend interface {
 	State(ctx context.Context) (LeakyBucketState, error)
 	// SetState sets (persists) the current state of the LeakyBucket.
 	SetState(ctx context.Context, state LeakyBucketState) error
+	// Reset resets (persists) the current state of the LeakyBucket.
+	Reset(ctx context.Context) error
 }
 
 // LeakyBucket implements the https://en.wikipedia.org/wiki/Leaky_bucket#As_a_queue algorithm.

diff --git a/tokenbucket.go b/tokenbucket.go
@@ -40,6 +40,8 @@ type TokenBucketStateBackend interface {
 	State(ctx context.Context) (TokenBucketState, error)
 	// SetState sets (persists) the current state of the TokenBucket.
 	SetState(ctx context.Context, state TokenBucketState) error
+	// Reset resets (persists) the current state of the TokenBucket.
+	Reset(ctx context.Context) error
 }
 
 // TokenBucket implements the https://en.wikipedia.org/wiki/Token_bucket algorithm.