Jelajahi Sumber

漏掉web的日志关闭操作

Heartfilia 1 tahun lalu
induk
melakukan
e19242bd14
2 mengubah file dengan 7 tambahan dan 2 penghapusan
  1. 4 0
      core/api.go
  2. 3 2
      utils/logger.go

+ 4 - 0
core/api.go

@@ -293,6 +293,10 @@ func setupRouters(conf config.ConfStruct) *gin.Engine {
 }
 
 func InitAPI(conf config.ConfStruct) {
+	if conf.CloseWebLog {
+		// 将默认的日志输出器设置为空
+		gin.DefaultWriter = utils.LogWriter{}
+	}
 	gin.SetMode(getGinMode(conf.Mode))
 	router := setupRouters(conf)
 

+ 3 - 2
utils/logger.go

@@ -7,6 +7,7 @@ import (
 var isPrint = true
 
 func InitLogger(closeLog bool) {
+
 	if closeLog {
 		isPrint = false
 	}
@@ -18,9 +19,9 @@ func InitLogger(closeLog bool) {
 	})
 }
 
-type logWriter struct{}
+type LogWriter struct{}
 
-func (w logWriter) Write(p []byte) (n int, err error) {
+func (w LogWriter) Write(p []byte) (n int, err error) {
 	return len(p), nil
 }