You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.5 KiB
62 lines
1.5 KiB
package logger
|
|
|
|
import (
|
|
"net/url"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/shiguanghuxian/etcd-manage/program/common"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
// 日志对象
|
|
var (
|
|
Log *zap.SugaredLogger
|
|
)
|
|
|
|
// InitLogger 日志初始化，用于记录操作日志
|
|
func InitLogger(logPath string, isDebug bool) (*zap.SugaredLogger, error) {
|
|
infoLogPath := ""
|
|
// errorLogPath := ""
|
|
if logPath == "" {
|
|
logRoot := common.GetRootDir() + "logs" + string(os.PathSeparator)
|
|
if isExt, _ := common.PathExists(logRoot); isExt == false {
|
|
os.MkdirAll(logRoot, os.ModePerm)
|
|
}
|
|
infoLogPath = logRoot + "access.log"
|
|
} else {
|
|
logPath = strings.TrimRight(logPath, string(os.PathSeparator))
|
|
infoLogPath = logPath + string(os.PathSeparator) + "access.log"
|
|
}
|
|
|
|
// 兼容win根完整路径问题
|
|
zap.RegisterSink("winfile", func(u *url.URL) (zap.Sink, error) {
|
|
return os.OpenFile(u.Path[1:], os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
|
|
})
|
|
|
|
cfg := &zap.Config{
|
|
Encoding: "json",
|
|
}
|
|
cfg.EncoderConfig = zap.NewProductionEncoderConfig()
|
|
atom := zap.NewAtomicLevel()
|
|
if isDebug == true {
|
|
atom.SetLevel(zapcore.DebugLevel)
|
|
cfg.OutputPaths = []string{"stdout"}
|
|
} else {
|
|
atom.SetLevel(zapcore.InfoLevel)
|
|
if runtime.GOOS == "windows" {
|
|
cfg.OutputPaths = []string{"winfile:///" + infoLogPath}
|
|
} else {
|
|
cfg.OutputPaths = []string{infoLogPath}
|
|
}
|
|
}
|
|
cfg.Level = atom
|
|
logger, err := cfg.Build()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
Log = logger.Sugar()
|
|
return Log, nil
|
|
}
|