mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2025-10-29 18:12:25 -05:00
[chore]: Bump github.com/KimMachineGun/automemlimit from 0.2.6 to 0.3.0 (#2165)
This commit is contained in:
parent
13f1c85e70
commit
e6407ec95c
7 changed files with 116 additions and 186 deletions
65
vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go
generated
vendored
65
vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go
generated
vendored
|
|
@ -1,65 +0,0 @@
|
|||
package gin
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"github.com/ulule/limiter/v3"
|
||||
)
|
||||
|
||||
// Middleware is the middleware for gin.
|
||||
type Middleware struct {
|
||||
Limiter *limiter.Limiter
|
||||
OnError ErrorHandler
|
||||
OnLimitReached LimitReachedHandler
|
||||
KeyGetter KeyGetter
|
||||
ExcludedKey func(string) bool
|
||||
}
|
||||
|
||||
// NewMiddleware return a new instance of a gin middleware.
|
||||
func NewMiddleware(limiter *limiter.Limiter, options ...Option) gin.HandlerFunc {
|
||||
middleware := &Middleware{
|
||||
Limiter: limiter,
|
||||
OnError: DefaultErrorHandler,
|
||||
OnLimitReached: DefaultLimitReachedHandler,
|
||||
KeyGetter: DefaultKeyGetter,
|
||||
ExcludedKey: nil,
|
||||
}
|
||||
|
||||
for _, option := range options {
|
||||
option.apply(middleware)
|
||||
}
|
||||
|
||||
return func(ctx *gin.Context) {
|
||||
middleware.Handle(ctx)
|
||||
}
|
||||
}
|
||||
|
||||
// Handle gin request.
|
||||
func (middleware *Middleware) Handle(c *gin.Context) {
|
||||
key := middleware.KeyGetter(c)
|
||||
if middleware.ExcludedKey != nil && middleware.ExcludedKey(key) {
|
||||
c.Next()
|
||||
return
|
||||
}
|
||||
|
||||
context, err := middleware.Limiter.Get(c, key)
|
||||
if err != nil {
|
||||
middleware.OnError(c, err)
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
c.Header("X-RateLimit-Limit", strconv.FormatInt(context.Limit, 10))
|
||||
c.Header("X-RateLimit-Remaining", strconv.FormatInt(context.Remaining, 10))
|
||||
c.Header("X-RateLimit-Reset", strconv.FormatInt(context.Reset, 10))
|
||||
|
||||
if context.Reached {
|
||||
middleware.OnLimitReached(c)
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
c.Next()
|
||||
}
|
||||
71
vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go
generated
vendored
71
vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go
generated
vendored
|
|
@ -1,71 +0,0 @@
|
|||
package gin
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// Option is used to define Middleware configuration.
|
||||
type Option interface {
|
||||
apply(*Middleware)
|
||||
}
|
||||
|
||||
type option func(*Middleware)
|
||||
|
||||
func (o option) apply(middleware *Middleware) {
|
||||
o(middleware)
|
||||
}
|
||||
|
||||
// ErrorHandler is an handler used to inform when an error has occurred.
|
||||
type ErrorHandler func(c *gin.Context, err error)
|
||||
|
||||
// WithErrorHandler will configure the Middleware to use the given ErrorHandler.
|
||||
func WithErrorHandler(handler ErrorHandler) Option {
|
||||
return option(func(middleware *Middleware) {
|
||||
middleware.OnError = handler
|
||||
})
|
||||
}
|
||||
|
||||
// DefaultErrorHandler is the default ErrorHandler used by a new Middleware.
|
||||
func DefaultErrorHandler(c *gin.Context, err error) {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// LimitReachedHandler is an handler used to inform when the limit has exceeded.
|
||||
type LimitReachedHandler func(c *gin.Context)
|
||||
|
||||
// WithLimitReachedHandler will configure the Middleware to use the given LimitReachedHandler.
|
||||
func WithLimitReachedHandler(handler LimitReachedHandler) Option {
|
||||
return option(func(middleware *Middleware) {
|
||||
middleware.OnLimitReached = handler
|
||||
})
|
||||
}
|
||||
|
||||
// DefaultLimitReachedHandler is the default LimitReachedHandler used by a new Middleware.
|
||||
func DefaultLimitReachedHandler(c *gin.Context) {
|
||||
c.String(http.StatusTooManyRequests, "Limit exceeded")
|
||||
}
|
||||
|
||||
// KeyGetter will define the rate limiter key given the gin Context.
|
||||
type KeyGetter func(c *gin.Context) string
|
||||
|
||||
// WithKeyGetter will configure the Middleware to use the given KeyGetter.
|
||||
func WithKeyGetter(handler KeyGetter) Option {
|
||||
return option(func(middleware *Middleware) {
|
||||
middleware.KeyGetter = handler
|
||||
})
|
||||
}
|
||||
|
||||
// DefaultKeyGetter is the default KeyGetter used by a new Middleware.
|
||||
// It returns the Client IP address.
|
||||
func DefaultKeyGetter(c *gin.Context) string {
|
||||
return c.ClientIP()
|
||||
}
|
||||
|
||||
// WithExcludedKey will configure the Middleware to ignore key(s) using the given function.
|
||||
func WithExcludedKey(handler func(string) bool) Option {
|
||||
return option(func(middleware *Middleware) {
|
||||
middleware.ExcludedKey = handler
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue