配置文件,新增运行模式配置项

hoysoft · Mar 7, 2021 · 44b5ed6 · 44b5ed6
1 parent 48c5bf0
commit 44b5ed6
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 3 deletions.
diff --git a/app/global/global.go b/app/global/global.go
@@ -12,6 +12,8 @@ var (
 	SessionDrive            string
 	SessionOption           = make(map[string]interface{})
 	SessionDriveRedisConfig = make(map[string]interface{})
+	GinRunMode              string
+	GinWriteLog             bool
 )
 
 func ReadConfig() {
@@ -50,4 +52,9 @@ func ReadConfig() {
 		SessionDriveRedisConfig["password"], _ = config.GetConfig("session_driver_redis.password")
 	}
 
+	// 运行模式
+	GinRunMode, err = config.GetConfig("site.gin_run_mode")
+
+	// 是否记录运行日志
+	GinWriteLog, err = config.GetConfigToBool("site.gin_write_log")
 }
diff --git a/app/routers/routers.go b/app/routers/routers.go
@@ -70,8 +70,13 @@ func roleMiddleWare(role int) gin.HandlerFunc {
 }
 
 func InitRouter() *gin.Engine {
-	gin.SetMode(gin.ReleaseMode)
-	gin.DefaultWriter = ioutil.Discard
+	// 运行模式 release,debug,test
+	gin.SetMode(global.GinRunMode)
+
+	// 是否记录运行日志
+	if !global.GinWriteLog {
+		gin.DefaultWriter = ioutil.Discard
+	}
 
 	r := gin.Default()
 	r.Delims("[[", "]]")

diff --git a/config/config.ini b/config/config.ini
@@ -1,6 +1,13 @@
 # 站点配置
 [site]
-    http_port = 8080
+    ; web 端口号
+    http_port = "8080"
+
+    ; gin 运行模式 release,debug,test
+    gin_run_mode = "release"
+
+    ; 是否记录运行日志
+    gin_write_log = "1"
 
 # 数据库配置
 [mysql]