From dbd81eed2b347d16e9cf7f487563d08e58ccdaf9 Mon Sep 17 00:00:00 2001 From: UlricQin Date: Fri, 3 Jul 2020 12:38:20 +0800 Subject: [PATCH 1/2] code refactor --- src/modules/judge/judge.go | 4 ++++ src/toolkits/http/http.go | 1 - 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/modules/judge/judge.go b/src/modules/judge/judge.go index 26bcebca..7eb5592d 100644 --- a/src/modules/judge/judge.go +++ b/src/modules/judge/judge.go @@ -80,6 +80,10 @@ func main() { go judge.NodataJudge(cfg.NodataConcurrency) go report.Init(cfg.Report, "monapi") + if cfg.Logger.Level != "DEBUG" { + gin.SetMode(gin.ReleaseMode) + } + r := gin.New() routes.Config(r) go http.Start(r, "judge", cfg.Logger.Level) diff --git a/src/toolkits/http/http.go b/src/toolkits/http/http.go index a15c2f33..445b49f2 100644 --- a/src/toolkits/http/http.go +++ b/src/toolkits/http/http.go @@ -25,7 +25,6 @@ func Start(r *gin.Engine, mod string, level string) { recoveryMid := middleware.Recovery() if level != "DEBUG" { - gin.SetMode(gin.ReleaseMode) middleware.DisableConsoleColor() } else { srv.WriteTimeout = 120 * time.Second From 420b61ab5221c94cc5edd5c95728524c828ad676 Mon Sep 17 00:00:00 2001 From: UlricQin Date: Mon, 6 Jul 2020 14:46:39 +0800 Subject: [PATCH 2/2] use release mode in collector --- src/modules/collector/collector.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/modules/collector/collector.go b/src/modules/collector/collector.go index 2271656b..fa64136e 100644 --- a/src/modules/collector/collector.go +++ b/src/modules/collector/collector.go @@ -97,6 +97,10 @@ func main() { go worker.PusherStart() go worker.Zeroize() + if cfg.Logger.Level != "DEBUG" { + gin.SetMode(gin.ReleaseMode) + } + r := gin.New() routes.Config(r) http.Start(r, "collector", cfg.Logger.Level)