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) {
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)

View File

@@ -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))
}
}

View File

@@ -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)

View File

@@ -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)