Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: enable gzip on api route & web route

Browse Source
This commit is contained in:
JustSong 2023-04-25 21:56:07 +08:00
parent 69ee87c57f
commit 284beed8dc
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
router/api-router.go
View File

@ -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
router/web-router.go
View File

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