mirror of
https://github.com/NotAShelf/watchdog.git
synced 2026-04-17 23:58:19 +00:00
watchdog: make metrics rate limit configurable; document env vars
Signed-off-by: NotAShelf <raf@notashelf.dev> Change-Id: I01033406c32bd4e31a76e676be97af046a6a6964
This commit is contained in:
parent
ac24734e8f
commit
9c8f91ef27
2 changed files with 21 additions and 9 deletions
|
|
@ -90,9 +90,15 @@ func Run(cfg *config.Config) error {
|
|||
)
|
||||
}
|
||||
|
||||
// Add rate limiting to metrics endpoint (30 requests per minute)
|
||||
metricsRateLimiter := ratelimit.NewTokenBucket(30, 30, time.Minute)
|
||||
metricsHandler = rateLimitMiddleware(metricsHandler, metricsRateLimiter)
|
||||
// Add rate limiting to metrics endpoint
|
||||
if cfg.Limits.MaxMetricsPerMinute > 0 {
|
||||
metricsRateLimiter := ratelimit.NewTokenBucket(
|
||||
cfg.Limits.MaxMetricsPerMinute,
|
||||
cfg.Limits.MaxMetricsPerMinute,
|
||||
time.Minute,
|
||||
)
|
||||
metricsHandler = rateLimitMiddleware(metricsHandler, metricsRateLimiter)
|
||||
}
|
||||
|
||||
mux.Handle(cfg.Server.MetricsPath, metricsHandler)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue