diff --git a/logger/logger.go b/logger/logger.go index f5d568c..869037a 100644 --- a/logger/logger.go +++ b/logger/logger.go @@ -1,7 +1,6 @@ package logger import ( - "flag" "fmt" "io" "log" @@ -10,10 +9,12 @@ import ( "runtime" "runtime/debug" "strings" + + "repositories.action2quare.com/ayo/gocommon/flagx" ) var stdlogger *log.Logger -var _ = flag.Bool("logfile", false, "") +var UseLogFile = flagx.Bool("logfile", false, "") func init() { binpath, _ := os.Executable() @@ -22,16 +23,7 @@ func init() { var outWriter io.Writer outWriter = os.Stdout - args := os.Args - useLogFile := false - for _, arg := range args { - if arg == "-logfile" { - useLogFile = true - break - } - } - - if useLogFile { + if *UseLogFile { ext := path.Ext(binname) if len(ext) > 0 { binname = binname[:len(binname)-len(ext)]