refactor(store): store requires a context.Context

iandotmartin · Aug 16, 2017 · af247d0 · af247d0
1 parent 4c093d4
commit af247d0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 301 deletions.
diff --git a/store.go b/store.go
@@ -1,11 +1,16 @@
 package limiter
 
-import "time"
+import (
+	"context"
+	"time"
+)
 
 // Store is the common interface for limiter stores.
 type Store interface {
-	Get(key string, rate Rate) (Context, error)
-	Peek(key string, rate Rate) (Context, error)
+	// Get returns the limit for given identifier.
+	Get(ctx context.Context, key string, rate Rate) (Context, error)
+	// Peek returns the limit for given identifier, without modification on current values.
+	Peek(ctx context.Context, key string, rate Rate) (Context, error)
 }
 
 // StoreOptions are options for store.

diff --git a/store_memory.go b/store_memory.go
diff --git a/store_redis.go b/store_redis.go