wlw/ai-agent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: simplify cors and maxBodySize middleware by removing parameters

Browse Source
This commit is contained in:
mlogclub
2026-05-27 22:06:32 +08:00
parent 4d816182ee
commit 498932de61
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/bootstrap/server.go
+6 -4
View File
@@ -29,10 +29,10 @@ func NewServer() (*gin.Engine, error) {
printBanner() printBanner()
app := gin.New() app := gin.New()
app.Use(corsMiddleware(cfg.Server.CORS.AllowedOrigins)) app.Use(corsMiddleware())
app.Use(gin.Recovery()) app.Use(gin.Recovery())
app.Use(requestLogMiddleware()) app.Use(requestLogMiddleware())
app.Use(maxBodySizeMiddleware(cfg.Storage.MaxRequestBodySizeBytes())) app.Use(maxBodySizeMiddleware())
app.Use(i18nx.Middleware()) app.Use(i18nx.Middleware())
addRouter(app) addRouter(app)
@@ -60,7 +60,8 @@ func NewServer() (*gin.Engine, error) {
return app, nil return app, nil
} }
func corsMiddleware(allowedOrigins []string) gin.HandlerFunc { func corsMiddleware() gin.HandlerFunc {
allowedOrigins := config.Current().Server.CORS.AllowedOrigins
allowHeaders := "Origin, Content-Type, Accept, Authorization, X-Requested-With, X-Guest-Id, X-Channel-Id, X-External-Id, X-External-Name, X-Customer-Session-Token, X-Customer-Session-Expires-At" allowHeaders := "Origin, Content-Type, Accept, Authorization, X-Requested-With, X-Guest-Id, X-Channel-Id, X-External-Id, X-External-Name, X-Customer-Session-Token, X-Customer-Session-Expires-At"
exposeHeaders := "Content-Length, Content-Type, Authorization, X-Guest-Id, X-Channel-Id, X-External-Id, X-External-Name, X-Customer-Session-Token, X-Customer-Session-Expires-At" exposeHeaders := "Content-Length, Content-Type, Authorization, X-Guest-Id, X-Channel-Id, X-External-Id, X-External-Name, X-Customer-Session-Token, X-Customer-Session-Expires-At"
allowMethods := "GET, POST, PUT, PATCH, DELETE, OPTIONS" allowMethods := "GET, POST, PUT, PATCH, DELETE, OPTIONS"
@@ -119,7 +120,8 @@ func requestLogMiddleware() gin.HandlerFunc {
} }
} }
func maxBodySizeMiddleware(limit int64) gin.HandlerFunc { func maxBodySizeMiddleware() gin.HandlerFunc {
limit := config.Current().Storage.MaxRequestBodySizeBytes()
return func(ctx *gin.Context) { return func(ctx *gin.Context) {
ctx.Request.Body = http.MaxBytesReader(ctx.Writer, ctx.Request.Body, limit) ctx.Request.Body = http.MaxBytesReader(ctx.Writer, ctx.Request.Body, limit)
ctx.Next() ctx.Next()