From 9399ade2df5b8ca0661c280bbb01c06ad9262556 Mon Sep 17 00:00:00 2001 From: nyne Date: Wed, 9 Jul 2025 17:33:10 +0800 Subject: [PATCH] fix request limiter. --- server/api/comment.go | 4 ++-- server/api/file.go | 6 +++--- server/api/image.go | 2 +- server/api/user.go | 2 +- 4 files changed, 7 insertions(+), 7 deletions(-) diff --git a/server/api/comment.go b/server/api/comment.go index e1829bf..88be0c0 100644 --- a/server/api/comment.go +++ b/server/api/comment.go @@ -13,8 +13,8 @@ import ( func AddCommentRoutes(router fiber.Router) { api := router.Group("/comments") - api.Use(middleware.NewRequestLimiter(500, 24*time.Hour)).Post("/resource/:resourceID", createResourceComment) - api.Use(middleware.NewRequestLimiter(500, 24*time.Hour)).Post("/reply/:commentID", createReplyComment) + api.Post("/resource/:resourceID", createResourceComment, middleware.NewRequestLimiter(500, 24*time.Hour)) + api.Post("/reply/:commentID", createReplyComment, middleware.NewRequestLimiter(500, 24*time.Hour)) api.Get("/resource/:resourceID", listResourceComments) api.Get("/reply/:commentID", listReplyComments) api.Get("/user/:username", listCommentsByUser) diff --git a/server/api/file.go b/server/api/file.go index b6d3c4a..80857f0 100644 --- a/server/api/file.go +++ b/server/api/file.go @@ -19,17 +19,17 @@ import ( func AddFileRoutes(router fiber.Router) { fileGroup := router.Group("/files") { - fileGroup.Use(middleware.NewRequestLimiter(10, time.Hour)).Post("/upload/init", initUpload) + fileGroup.Post("/upload/init", initUpload, middleware.NewRequestLimiter(10, time.Hour)) fileGroup.Post("/upload/block/:id/:index", uploadBlock) fileGroup.Post("/upload/finish/:id", finishUpload) fileGroup.Post("/upload/cancel/:id", cancelUpload) - fileGroup.Use(middleware.NewRequestLimiter(50, time.Hour)).Post("/redirect", createRedirectFile) + fileGroup.Post("/redirect", createRedirectFile, middleware.NewRequestLimiter(50, time.Hour)) fileGroup.Post("/upload/url", createServerDownloadTask) fileGroup.Get("/:id", getFile) fileGroup.Put("/:id", updateFile) fileGroup.Delete("/:id", deleteFile) fileGroup.Get("/download/local", downloadLocalFile) - fileGroup.Use(middleware.NewDynamicRequestLimiter(config.MaxDownloadsPerDayForSingleIP, 24*time.Hour)).Get("/download/:id", downloadFile) + fileGroup.Get("/download/:id", downloadFile, middleware.NewDynamicRequestLimiter(config.MaxDownloadsPerDayForSingleIP, 24*time.Hour)) } } diff --git a/server/api/image.go b/server/api/image.go index 8d0cf9f..a1e24d0 100644 --- a/server/api/image.go +++ b/server/api/image.go @@ -97,7 +97,7 @@ func handleGetResampledImage(c fiber.Ctx) error { func AddImageRoutes(api fiber.Router) { image := api.Group("/image") { - image.Use(middleware.NewRequestLimiter(50, time.Hour)).Put("/", handleUploadImage) + image.Put("/", handleUploadImage, middleware.NewRequestLimiter(50, time.Hour)) image.Get("/resampled/:id", handleGetResampledImage) image.Get("/:id", handleGetImage) image.Delete("/:id", handleDeleteImage) diff --git a/server/api/user.go b/server/api/user.go index c46bb8c..66e99df 100644 --- a/server/api/user.go +++ b/server/api/user.go @@ -344,7 +344,7 @@ func handleGetMe(c fiber.Ctx) error { func AddUserRoutes(r fiber.Router) { u := r.Group("user") - u.Use(middleware.NewRequestLimiter(5, time.Hour)).Post("/register", handleUserRegister) + u.Post("/register", handleUserRegister, middleware.NewRequestLimiter(5, time.Hour)) u.Post("/login", handleUserLogin) u.Put("/avatar", handleUserChangeAvatar) u.Post("/password", handleUserChangePassword)