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() }