diff options
| author | 2023-08-28 06:59:08 +0000 | |
|---|---|---|
| committer | 2023-08-28 06:59:08 +0000 | |
| commit | e6407ec95c1c0b134866d7bd4b2d17f4b0c3bf39 (patch) | |
| tree | 4085dc4a80680a2f480e0466a984e3097f4c58b4 /vendor/github.com/ulule | |
| parent | [chore/frontend] Make line-height a wee little bit bigger (#2159) (diff) | |
| download | gotosocial-e6407ec95c1c0b134866d7bd4b2d17f4b0c3bf39.tar.xz | |
[chore]: Bump github.com/KimMachineGun/automemlimit from 0.2.6 to 0.3.0 (#2165)
Diffstat (limited to 'vendor/github.com/ulule')
| -rw-r--r-- | vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go | 65 | ||||
| -rw-r--r-- | vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go | 71 |
2 files changed, 0 insertions, 136 deletions
diff --git a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go b/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go deleted file mode 100644 index 23bad417a..000000000 --- a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go +++ /dev/null @@ -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() -} diff --git a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go b/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go deleted file mode 100644 index 604c6bc68..000000000 --- a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go +++ /dev/null @@ -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 - }) -} |
