From 303d4c444d2233d97f4f7cefaecd3eda21afe7bb Mon Sep 17 00:00:00 2001
From: Sebastian Winkler <sebastian.winkler@slmn.de>
Date: Mon, 25 Feb 2019 20:16:26 +0100
Subject: [PATCH] adds Reset to bucket package

---
 bucket/bucket.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/bucket/bucket.go b/bucket/bucket.go
index c3e581a..c0b2b27 100644
--- a/bucket/bucket.go
+++ b/bucket/bucket.go
@@ -60,3 +60,14 @@ func Add(
 ) ([]redis.Z, error) {
 	return AddWithValue(client, keySuffix, time.Now().UnixNano(), decay)
 }
+
+// Reset resets the given leaky bucket
+func Reset(
+	client *redis.Client,
+	keySuffix string,
+) error {
+	key := key(keySuffix)
+
+	_, err := client.Del(key).Result()
+	return err
+}