Sfoglia il codice sorgente

feat: enable gzip on api route & web route

JustSong 2 anni fa
parent
commit
284beed8dc
2 ha cambiato i file con 4 aggiunte e 0 eliminazioni
  1. 2 0
      router/api-router.go
  2. 2 0
      router/web-router.go

+ 2 - 0
router/api-router.go

@@ -1,6 +1,7 @@
 package router
 package router
 
 
 import (
 import (
+	"github.com/gin-contrib/gzip"
 	"github.com/gin-gonic/gin"
 	"github.com/gin-gonic/gin"
 	"one-api/controller"
 	"one-api/controller"
 	"one-api/middleware"
 	"one-api/middleware"
@@ -8,6 +9,7 @@ import (
 
 
 func SetApiRouter(router *gin.Engine) {
 func SetApiRouter(router *gin.Engine) {
 	apiRouter := router.Group("/api")
 	apiRouter := router.Group("/api")
+	apiRouter.Use(gzip.Gzip(gzip.DefaultCompression))
 	apiRouter.Use(middleware.GlobalAPIRateLimit())
 	apiRouter.Use(middleware.GlobalAPIRateLimit())
 	{
 	{
 		apiRouter.GET("/status", controller.GetStatus)
 		apiRouter.GET("/status", controller.GetStatus)

+ 2 - 0
router/web-router.go

@@ -2,6 +2,7 @@ package router
 
 
 import (
 import (
 	"embed"
 	"embed"
+	"github.com/gin-contrib/gzip"
 	"github.com/gin-contrib/static"
 	"github.com/gin-contrib/static"
 	"github.com/gin-gonic/gin"
 	"github.com/gin-gonic/gin"
 	"net/http"
 	"net/http"
@@ -10,6 +11,7 @@ import (
 )
 )
 
 
 func setWebRouter(router *gin.Engine, buildFS embed.FS, indexPage []byte) {
 func setWebRouter(router *gin.Engine, buildFS embed.FS, indexPage []byte) {
+	router.Use(gzip.Gzip(gzip.DefaultCompression))
 	router.Use(middleware.GlobalWebRateLimit())
 	router.Use(middleware.GlobalWebRateLimit())
 	router.Use(middleware.Cache())
 	router.Use(middleware.Cache())
 	router.Use(static.Serve("/", common.EmbedFolder(buildFS, "web/build")))
 	router.Use(static.Serve("/", common.EmbedFolder(buildFS, "web/build")))