gin-rate-limit is a rate limiter for the gin framework. By default, it can only store rate limit info in memory and with redis. If you want to store it somewhere else you can make your own store or use third party stores. The library is new so there are no third party stores yet, so I would appreciate if someone could make one.
Install
go get github.com/JGLTechnologies/gin-rate-limitRedis Example
package main
import (
"github.com/JGLTechnologies/gin-rate-limit"
"github.com/gin-gonic/gin"
"github.com/redis/go-redis/v9"
"time"
)
func keyFunc(c *gin.Context) string {
return c.ClientIP()
}
func errorHandler(c *gin.Context, info ratelimit.Info) {
c.String(429, "Too many requests. Try again in "+time.Until(info.ResetTime).String())
}
func main() {
server := gin.Default()
// This makes it so each ip can only make 5 requests per second
store := ratelimit.RedisStore(&ratelimit.RedisOptions{
RedisClient: redis.NewClient(&redis.Options{
Addr: "localhost:7680",
}),
Rate: time.Second,
Limit: 5,
})
mw := ratelimit.RateLimiter(store, &ratelimit.Options{
ErrorHandler: errorHandler,
KeyFunc: keyFunc,
})
server.GET("/", mw, func(c *gin.Context) {
c.String(200, "Hello World")
})
server.Run(":8080")
}Basic Setup
package main
import (
"github.com/gin-gonic/gin"
"github.com/JGLTechnologies/gin-rate-limit"
"time"
)
func keyFunc(c *gin.Context) string {
return c.ClientIP()
}
func errorHandler(c *gin.Context, info ratelimit.Info) {
c.String(429, "Too many requests. Try again in "+time.Until(info.ResetTime).String())
}
func main() {
server := gin.Default()
// This makes it so each ip can only make 5 requests per second
store := ratelimit.InMemoryStore(&ratelimit.InMemoryOptions{
Rate: time.Second,
Limit: 5,
})
mw := ratelimit.RateLimiter(store, &ratelimit.Options{
ErrorHandler: errorHandler,
KeyFunc: keyFunc,
})
server.GET("/", mw, func(c *gin.Context) {
c.String(200, "Hello World")
})
server.Run(":8080")
}Custom Store Example
package main
import (
"github.com/JGLTechnologies/gin-rate-limit"
"github.com/gin-gonic/gin"
)
type CustomStore struct {
}
// Your store must have a method called Limit that takes a key, *gin.Context and returns ratelimit.Info
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,
}
}
