From f4fe9e18bab020911d41fca3c44293c5564604de Mon Sep 17 00:00:00 2001 From: VnPower Date: Tue, 5 Sep 2023 18:58:42 +0700 Subject: [PATCH] Add limiter --- views/routes.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/views/routes.go b/views/routes.go index 004a7fc..53bf63c 100644 --- a/views/routes.go +++ b/views/routes.go @@ -7,6 +7,7 @@ import ( "codeberg.org/vnpower/pixivfe/configs" "codeberg.org/vnpower/pixivfe/handler" "github.com/gofiber/fiber/v2" + "github.com/gofiber/fiber/v2/middleware/limiter" ) var PC *handler.PixivClient @@ -41,8 +42,12 @@ func SetupRoutes(r *fiber.App) { PC.SetUserAgent(configs.UserAgent) PC.AddHeader("Accept-Language", "en-US,en;q=0.5") + limit := limiter.New(limiter.Config{ + Max: 10, + }) + r.Get("/", index_page) - r.Get("artworks/:id/", artwork_page) + r.Get("artworks/:id/", limit, artwork_page) r.Get("users/:id/:category?", user_page) r.Get("newest", newest_artworks_page) r.Get("ranking", ranking_page)