logprefix flag 추가

This commit is contained in:
2024-07-24 17:49:45 +09:00
parent f0a97c4701
commit 2dafadf949

View File

@ -8,6 +8,7 @@ import (
"path" "path"
"runtime" "runtime"
"runtime/debug" "runtime/debug"
"strconv"
"strings" "strings"
"repositories.action2quare.com/ayo/gocommon/flagx" "repositories.action2quare.com/ayo/gocommon/flagx"
@ -15,7 +16,7 @@ import (
var stdlogger *log.Logger var stdlogger *log.Logger
var UseLogFile = flagx.Bool("logfile", false, "") var UseLogFile = flagx.Bool("logfile", false, "")
var logprefix = flagx.Int("logprefix", 3, "0 : no_prefix, 1 : date, 2 : time, 3 : datetime") var _ = flagx.Int("logprefix", 3, "0 : no_prefix, 1 : date, 2 : time, 3 : datetime")
func init() { func init() {
binpath, _ := os.Executable() binpath, _ := os.Executable()
@ -24,7 +25,29 @@ func init() {
var outWriter io.Writer var outWriter io.Writer
outWriter = os.Stdout outWriter = os.Stdout
if *UseLogFile { args := os.Args
useLogFile := false
for _, arg := range args {
if strings.HasPrefix(arg, "-logfile=") {
useLogFile, _ = strconv.ParseBool(arg[9:])
break
}
if arg == "-logfile" {
useLogFile = true
break
}
}
logprefix := 3
for _, arg := range args {
if strings.HasPrefix(arg, "-logprefix=") {
logprefix, _ = strconv.Atoi(arg[11:])
break
}
}
if useLogFile {
ext := path.Ext(binname) ext := path.Ext(binname)
if len(ext) > 0 { if len(ext) > 0 {
binname = binname[:len(binname)-len(ext)] binname = binname[:len(binname)-len(ext)]
@ -39,8 +62,8 @@ func init() {
outWriter = io.MultiWriter(outWriter, logFile) outWriter = io.MultiWriter(outWriter, logFile)
} }
if *logprefix < 4 { if logprefix < 4 {
stdlogger = log.New(outWriter, "", *logprefix) stdlogger = log.New(outWriter, "", logprefix)
} else { } else {
stdlogger = log.New(outWriter, "", log.LstdFlags) stdlogger = log.New(outWriter, "", log.LstdFlags)
} }