max multipart memory

liuzhaomax · Jul 27, 2024 · 5da4766 · 5da4766
1 parent 737c12b
commit 5da4766
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/environment/config/dev.yaml b/environment/config/dev.yaml
@@ -25,6 +25,7 @@ lib:
     interval: 15
   gin:
     run_mode: debug # debug test release
+    max_multipart_memory: 10 # MB
   db:
     type: mysql
     debug: true

diff --git a/environment/config/st.yaml b/environment/config/st.yaml
@@ -25,6 +25,7 @@ lib:
     interval: 15
   gin:
     run_mode: debug # debug test release
+    max_multipart_memory: 10 # MB
   db:
     type: mysql
     debug: true

diff --git a/internal/core/gin.go b/internal/core/gin.go
@@ -11,12 +11,14 @@ func init() {
 }
 
 type Gin struct {
-	RunMode string `mapstructure:"run_mode"`
+	RunMode            string `mapstructure:"run_mode"`
+	MaxMultipartMemory int64  `mapstructure:"max_multipart_memory"`
 }
 
 // InitGinEngine Gin引擎的provider
 func InitGinEngine() *gin.Engine {
-	gin.SetMode(GetConfig().Lib.Gin.RunMode) // debug, test, release
+	gin.SetMode(cfg.Lib.Gin.RunMode) // debug, test, release
 	app := gin.Default()
+	app.MaxMultipartMemory = cfg.Lib.Gin.MaxMultipartMemory << 20
 	return app
 }