Merge pull request #43 from sashati/v9

Add Reset function to remove history of a key

Author: vmihailenco

rate.go

@@ -16,6 +16,7 @@ type rediser interface {
 	EvalSha(ctx context.Context, sha1 string, keys []string, args ...interface{}) *redis.Cmd
 	ScriptExists(ctx context.Context, hashes ...string) *redis.BoolSliceCmd
 	ScriptLoad(ctx context.Context, script string) *redis.StringCmd
+	Del(ctx context.Context, keys ...string) *redis.IntCmd
 }
 
 type Limit struct {
@@ -158,6 +159,11 @@ func (l Limiter) AllowAtMost(
 	return res, nil
 }
 
+// Reset gets a key and reset all limitations and previous usages
+func (l *Limiter) Reset(ctx context.Context, key string) error {
+	return l.rdb.Del(ctx, redisPrefix+key).Err()
+}
+
 func dur(f float64) time.Duration {
 	if f == -1 {
 		return -1

rate_test.go

@@ -37,6 +37,15 @@ func TestAllow(t *testing.T) {
 	assert.Equal(t, res.RetryAfter, time.Duration(-1))
 	assert.InDelta(t, res.ResetAfter, 100*time.Millisecond, float64(10*time.Millisecond))
 
+	err = l.Reset(ctx, "test_id")
+	assert.Nil(t, err)
+	res, err = l.Allow(ctx, "test_id", limit)
+	assert.Nil(t, err)
+	assert.Equal(t, res.Allowed, 1)
+	assert.Equal(t, res.Remaining, 9)
+	assert.Equal(t, res.RetryAfter, time.Duration(-1))
+	assert.InDelta(t, res.ResetAfter, 100*time.Millisecond, float64(10*time.Millisecond))
+
 	res, err = l.AllowN(ctx, "test_id", limit, 2)
 	assert.Nil(t, err)
 	assert.Equal(t, res.Allowed, 2)
@@ -57,6 +66,7 @@ func TestAllow(t *testing.T) {
 	assert.Equal(t, res.Remaining, 0)
 	assert.InDelta(t, res.RetryAfter, 99*time.Second, float64(time.Second))
 	assert.InDelta(t, res.ResetAfter, 999*time.Millisecond, float64(10*time.Millisecond))
+
 }
 
 func TestRetryAfter(t *testing.T) {