fix request limiter.

This commit is contained in:
2025-07-09 17:33:10 +08:00
parent b1d395eac6
commit 9399ade2df
4 changed files with 7 additions and 7 deletions

View File

@@ -13,8 +13,8 @@ import (
func AddCommentRoutes(router fiber.Router) { func AddCommentRoutes(router fiber.Router) {
api := router.Group("/comments") api := router.Group("/comments")
api.Use(middleware.NewRequestLimiter(500, 24*time.Hour)).Post("/resource/:resourceID", createResourceComment) api.Post("/resource/:resourceID", createResourceComment, middleware.NewRequestLimiter(500, 24*time.Hour))
api.Use(middleware.NewRequestLimiter(500, 24*time.Hour)).Post("/reply/:commentID", createReplyComment) api.Post("/reply/:commentID", createReplyComment, middleware.NewRequestLimiter(500, 24*time.Hour))
api.Get("/resource/:resourceID", listResourceComments) api.Get("/resource/:resourceID", listResourceComments)
api.Get("/reply/:commentID", listReplyComments) api.Get("/reply/:commentID", listReplyComments)
api.Get("/user/:username", listCommentsByUser) api.Get("/user/:username", listCommentsByUser)

View File

@@ -19,17 +19,17 @@ import (
func AddFileRoutes(router fiber.Router) { func AddFileRoutes(router fiber.Router) {
fileGroup := router.Group("/files") 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/block/:id/:index", uploadBlock)
fileGroup.Post("/upload/finish/:id", finishUpload) fileGroup.Post("/upload/finish/:id", finishUpload)
fileGroup.Post("/upload/cancel/:id", cancelUpload) 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.Post("/upload/url", createServerDownloadTask)
fileGroup.Get("/:id", getFile) fileGroup.Get("/:id", getFile)
fileGroup.Put("/:id", updateFile) fileGroup.Put("/:id", updateFile)
fileGroup.Delete("/:id", deleteFile) fileGroup.Delete("/:id", deleteFile)
fileGroup.Get("/download/local", downloadLocalFile) 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))
} }
} }

View File

@@ -97,7 +97,7 @@ func handleGetResampledImage(c fiber.Ctx) error {
func AddImageRoutes(api fiber.Router) { func AddImageRoutes(api fiber.Router) {
image := api.Group("/image") 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("/resampled/:id", handleGetResampledImage)
image.Get("/:id", handleGetImage) image.Get("/:id", handleGetImage)
image.Delete("/:id", handleDeleteImage) image.Delete("/:id", handleDeleteImage)

View File

@@ -344,7 +344,7 @@ func handleGetMe(c fiber.Ctx) error {
func AddUserRoutes(r fiber.Router) { func AddUserRoutes(r fiber.Router) {
u := r.Group("user") 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.Post("/login", handleUserLogin)
u.Put("/avatar", handleUserChangeAvatar) u.Put("/avatar", handleUserChangeAvatar)
u.Post("/password", handleUserChangePassword) u.Post("/password", handleUserChangePassword)