adds Reset to bucket package

ProjectCacophony · Feb 25, 2019 · 303d4c4 · 303d4c4
1 parent 8cdfc0d
commit 303d4c4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git 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
+}