From 1509e5f8485fec79c84c2d0f112c05e1ba4aab1e Mon Sep 17 00:00:00 2001 From: Mmx <1624045573@qq.com> Date: Thu, 25 Nov 2021 08:03:17 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E7=BB=9F=E4=B8=80flag=E5=A4=84=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- controllers/daemon.go | 6 ++---- global/flags.go | 4 +++- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/controllers/daemon.go b/controllers/daemon.go index f31a8bb..3ca444b 100644 --- a/controllers/daemon.go +++ b/controllers/daemon.go @@ -1,8 +1,8 @@ package controllers import ( - "flag" "fmt" + "github.com/Mmx233/BitSrunLoginGo/global" "github.com/Mmx233/tool" "time" ) @@ -18,9 +18,7 @@ var Daemon = daemon{ } func init() { - goDaemon := flag.Bool("daemon", false, "") - flag.Parse() - if *goDaemon { + if global.Flags.Daemon { Guardian(false) } } diff --git a/global/flags.go b/global/flags.go index 50f0b88..7c9667e 100644 --- a/global/flags.go +++ b/global/flags.go @@ -3,10 +3,12 @@ package global import "flag" var Flags struct { - Path string + Path string + Daemon bool } func initFlags() { flag.StringVar(&Flags.Path, "config", "", "config path") + flag.BoolVar(&Flags.Daemon, "daemon", false, "") flag.Parse() }