cache certain GET endpoints (#27)

prettyirrelevant · Jun 1, 2023 · 7510901 · 7510901
1 parent 31a6610
commit 7510901
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/kilishi/main.go b/kilishi/main.go
@@ -56,10 +56,15 @@ func setupMiddlewares(app *fiber.App, cfg *config.Config) {
 	app.Use(recover.New())
 	app.Use(cache.New(cache.Config{
 		Expiration: 24 * time.Hour,
-		Methods:    []string{fiber.MethodPost},
+		Methods:    []string{fiber.MethodPost, fiber.MethodGet},
 		Next: func(c *fiber.Ctx) bool {
-			noCacheEndpoints := map[string]bool{"/api/v1/playlists": true, "/api/v1/auth/spotify/refresh": true}
-			if _, ok := noCacheEndpoints[c.Path()]; ok && c.Method() == fiber.MethodPost {
+			noCachePOSTEndpoints := map[string]bool{"/api/v1/playlists": true, "/api/v1/auth/spotify/refresh": true}
+			if _, ok := noCachePOSTEndpoints[c.Path()]; ok && c.Method() == fiber.MethodPost {
+				return true
+			}
+
+			noCacheGETEndpoints := map[string]bool{"/api/v1/auth/deezer/callback": true, "/api/v1/auth/spotify/callback": true, "/api/v1/ping": true}
+			if _, ok := noCacheGETEndpoints[c.Path()]; ok && c.Method() == fiber.MethodGet {
 				return true
 			}