fix: 统一flag处理

This commit is contained in:
Mmx
2021-11-25 08:03:17 +08:00
parent 9ddc4950bf
commit 1509e5f848
2 changed files with 5 additions and 5 deletions

View File

@@ -1,8 +1,8 @@
package controllers package controllers
import ( import (
"flag"
"fmt" "fmt"
"github.com/Mmx233/BitSrunLoginGo/global"
"github.com/Mmx233/tool" "github.com/Mmx233/tool"
"time" "time"
) )
@@ -18,9 +18,7 @@ var Daemon = daemon{
} }
func init() { func init() {
goDaemon := flag.Bool("daemon", false, "") if global.Flags.Daemon {
flag.Parse()
if *goDaemon {
Guardian(false) Guardian(false)
} }
} }

View File

@@ -3,10 +3,12 @@ package global
import "flag" import "flag"
var Flags struct { var Flags struct {
Path string Path string
Daemon bool
} }
func initFlags() { func initFlags() {
flag.StringVar(&Flags.Path, "config", "", "config path") flag.StringVar(&Flags.Path, "config", "", "config path")
flag.BoolVar(&Flags.Daemon, "daemon", false, "")
flag.Parse() flag.Parse()
} }