add key field to result

Shareed2k · Mar 23, 2020 · 89af7b2 · 89af7b2
1 parent 917bb18
commit 89af7b2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/gcra.go b/gcra.go
@@ -41,6 +41,7 @@ func (c *gcra) AllowN(n int) (*Result, error) {
 
 	res := &Result{
 		Limit:      limit,
+		Key:        c.key,
 		Allowed:    values[0].(int64) == 0,
 		Remaining:  values[1].(int64),
 		RetryAfter: dur(retryAfter),

diff --git a/rate.go b/rate.go
@@ -28,6 +28,9 @@ type (
 		// Limit is the limit that was used to obtain this result.
 		Limit *Limit
 
+		// Key is the key of limit
+		Key string
+
 		// Allowed reports whether event may happen at time now.
 		Allowed bool
 

diff --git a/simple.go b/simple.go
@@ -28,6 +28,7 @@ func (c *simple) Allow() (r *Result, err error) {
 
 	return &Result{
 		Limit:      limit,
+		Key:        c.key,
 		Allowed:    values[0].(int64) == 1,
 		Remaining:  values[1].(int64),
 		RetryAfter: dur(retryAfter),