From 95b487c51ef98054b8dfc5a10c3ea074a129dee4 Mon Sep 17 00:00:00 2001
From: creamlike1024 <creamlike1024@outlook.com>
Date: Wed, 27 Aug 2025 01:30:01 +0800
Subject: [PATCH] feat: add disable cache middleware

---
 middleware/disable-cache.go | 12 ++++++++++++
 router/api-router.go        |  2 +-
 2 files changed, 13 insertions(+), 1 deletion(-)
 create mode 100644 middleware/disable-cache.go

diff --git a/middleware/disable-cache.go b/middleware/disable-cache.go
new file mode 100644
index 00000000..3076e90a
--- /dev/null
+++ b/middleware/disable-cache.go
@@ -0,0 +1,12 @@
+package middleware
+
+import "github.com/gin-gonic/gin"
+
+func DisableCache() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		c.Header("Cache-Control", "no-store, no-cache, must-revalidate, private, max-age=0")
+		c.Header("Pragma", "no-cache")
+		c.Header("Expires", "0")
+		c.Next()
+	}
+}
diff --git a/router/api-router.go b/router/api-router.go
index f8d87d64..311bb0a4 100644
--- a/router/api-router.go
+++ b/router/api-router.go
@@ -114,7 +114,7 @@ func SetApiRouter(router *gin.Engine) {
 			channelRoute.GET("/models", controller.ChannelListModels)
 			channelRoute.GET("/models_enabled", controller.EnabledListModels)
 			channelRoute.GET("/:id", controller.GetChannel)
-			channelRoute.POST("/:id/key", middleware.CriticalRateLimit(), controller.GetChannelKey)
+			channelRoute.POST("/:id/key", middleware.CriticalRateLimit(), middleware.DisableCache(), controller.GetChannelKey)
 			channelRoute.GET("/test", controller.TestAllChannels)
 			channelRoute.GET("/test/:id", controller.TestChannel)
 			channelRoute.GET("/update_balance", controller.UpdateAllChannelsBalance)