Edited Info structure in redis.go file. Edited the example in README.…

…md file.
JGLTechnologies · Feb 22, 2023 · b089895 · b089895
1 parent 2c1ce27
commit b089895
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -117,12 +117,14 @@ type CustomStore struct {
 func (s *CustomStore) Limit(key string, c *gin.Context) Info {
 	if UserWentOverLimit {
 		return Info{
+			Limit:         100,
 			RateLimited:   true,
 			ResetTime:     reset,
 			RemainingHits: 0,
 		}
 	}
 	return Info{
+		Limit:         100,
 		RateLimited:   false,
 		ResetTime:     reset,
 		RemainingHits: remaining,

diff --git a/redis.go b/redis.go
@@ -38,6 +38,7 @@ func (s *redisStoreType) Limit(key string, c *gin.Context) Info {
 	}
 	if s.skip != nil && s.skip(c) {
 		return Info{
+			Limit:         s.limit,
 			RateLimited:   false,
 			ResetTime:     time.Now().Add(time.Duration((s.rate - (time.Now().Unix() - ts)) * time.Second.Nanoseconds())),
 			RemainingHits: s.limit - uint(hits),
@@ -50,13 +51,15 @@ func (s *redisStoreType) Limit(key string, c *gin.Context) Info {
 				panic(err)
 			} else {
 				return Info{
+					Limit:         s.limit,
 					RateLimited:   false,
 					ResetTime:     time.Now().Add(time.Duration((s.rate - (time.Now().Unix() - ts)) * time.Second.Nanoseconds())),
 					RemainingHits: 0,
 				}
 			}
 		}
 		return Info{
+			Limit:         s.limit,
 			RateLimited:   true,
 			ResetTime:     time.Now().Add(time.Duration((s.rate - (time.Now().Unix() - ts)) * time.Second.Nanoseconds())),
 			RemainingHits: 0,
@@ -74,13 +77,15 @@ func (s *redisStoreType) Limit(key string, c *gin.Context) Info {
 			panic(err)
 		} else {
 			return Info{
+				Limit:         s.limit,
 				RateLimited:   false,
 				ResetTime:     time.Now().Add(time.Duration((s.rate - (time.Now().Unix() - ts)) * time.Second.Nanoseconds())),
 				RemainingHits: s.limit - uint(hits),
 			}
 		}
 	}
 	return Info{
+		Limit:         s.limit,
 		RateLimited:   false,
 		ResetTime:     time.Now().Add(time.Duration((s.rate - (time.Now().Unix() - ts)) * time.Second.Nanoseconds())),
 		RemainingHits: s.limit - uint(hits),