Add Limit to Result

vmihailenco · vmihailenco · commit 5f72b3658503 · 2019-12-29T11:13:18.000+02:00
diff --git a/rate.go b/rate.go
@@ -86,6 +86,7 @@ func (l *Limiter) AllowN(key string, limit *Limit, n int) (*Result, error) {
 	}
 
 	res := &Result{
+		Limit:      limit,
 		Allowed:    values[0].(int64) == 0,
 		Remaining:  int(values[1].(int64)),
 		RetryAfter: dur(retryAfter),
@@ -102,6 +103,9 @@ func dur(f float64) time.Duration {
 }
 
 type Result struct {
+	// Limit is the limit that was used to obtain this result.
+	Limit *Limit
+
 	// Allowed reports whether event may happen at time now.
 	Allowed bool
 

Original file line number	Diff line number	Diff line change
`@@ -86,6 +86,7 @@ func (l Limiter) AllowN(key string, limit Limit, n int) (*Result, error) {`
`86`	`86`	`}`
`87`	`87`
`88`	`88`	`res := &Result{`
	`89`	`+ Limit: limit,`
`89`	`90`	`Allowed: values[0].(int64) == 0,`
`90`	`91`	`Remaining: int(values[1].(int64)),`
`91`	`92`	`RetryAfter: dur(retryAfter),`
`@@ -102,6 +103,9 @@ func dur(f float64) time.Duration {`
`102`	`103`	`}`
`103`	`104`
`104`	`105`	`type Result struct {`
	`106`	`+ // Limit is the limit that was used to obtain this result.`
	`107`	`+ Limit *Limit`
	`108`	`+`
`105`	`109`	`// Allowed reports whether event may happen at time now.`
`106`	`110`	`Allowed bool`
`107`	`111`