mirror of
https://github.com/grafana/grafana.git
synced 2024-11-30 12:44:10 -06:00
25 lines
676 B
Go
25 lines
676 B
Go
package middleware
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/web"
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
type getTimeFn func() time.Time
|
|
|
|
// RateLimit is a very basic rate limiter.
|
|
// Will allow average of "rps" requests per second over an extended period of time, with max "burst" requests at the same time.
|
|
// getTime should return the current time. For non-testing purposes use time.Now
|
|
func RateLimit(rps, burst int, getTime getTimeFn) web.Handler {
|
|
l := rate.NewLimiter(rate.Limit(rps), burst)
|
|
return func(c *models.ReqContext) {
|
|
if !l.AllowN(getTime(), 1) {
|
|
c.JsonApiErr(429, "Rate limit reached", nil)
|
|
return
|
|
}
|
|
}
|
|
}
|