Skip to content

Commit 1c5864f

Browse files
committed
feat: add rate limit middlewares for job
Signed-off-by: Gaius <gaius.qi@gmail.com>
1 parent 75d6242 commit 1c5864f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

manager/config/constants.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,10 +91,10 @@ const (
9191
DefaultJobRateLimitFillInterval = 1 * time.Minute
9292

9393
// DefaultJobRateLimitCapacity is the default capacity for job rate limit.
94-
DefaultJobRateLimitCapacity = 100
94+
DefaultJobRateLimitCapacity = 10
9595

9696
// DefaultJobRateLimitQuantum is the default quantum for job rate limit.
97-
DefaultJobRateLimitQuantum = 100
97+
DefaultJobRateLimitQuantum = 10
9898

9999
// DefaultJobPreheatRegistryTimeout is the default timeout for requesting registry to get token and manifest.
100100
DefaultJobPreheatRegistryTimeout = 1 * time.Minute

manager/router/router.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ func Init(cfg *config.Config, logDir string, service service.Service, database *
202202
// TODO Add auth to the following routes and fix the tests.
203203
// Job.
204204
job := apiv1.Group("/jobs")
205-
job.POST("", h.CreateJob)
205+
job.POST("", middlewares.RateLimit(cfg.Job.RateLimit.FillInterval, cfg.Job.RateLimit.Capacity, cfg.Job.RateLimit.Quantum), h.CreateJob)
206206
job.DELETE(":id", h.DestroyJob)
207207
job.PATCH(":id", h.UpdateJob)
208208
job.GET(":id", h.GetJob)

0 commit comments

Comments
 (0)