mirror of
https://github.com/nitezs/sub2clash.git
synced 2024-12-25 02:54:42 -05:00
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
|
package logger
|
||
|
|
||
|
import (
|
||
|
"go.uber.org/zap"
|
||
|
"go.uber.org/zap/zapcore"
|
||
|
"path/filepath"
|
||
|
"sub2clash/config"
|
||
|
"sub2clash/utils"
|
||
|
"sync"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
Logger *zap.Logger
|
||
|
lock sync.Mutex
|
||
|
)
|
||
|
|
||
|
func init() {
|
||
|
buildLogger()
|
||
|
go rotateLogs()
|
||
|
}
|
||
|
|
||
|
func buildLogger() {
|
||
|
lock.Lock()
|
||
|
defer lock.Unlock()
|
||
|
var level zapcore.Level
|
||
|
switch config.Default.LogLevel {
|
||
|
case "error":
|
||
|
level = zap.ErrorLevel
|
||
|
case "debug":
|
||
|
level = zap.DebugLevel
|
||
|
case "warn":
|
||
|
level = zap.WarnLevel
|
||
|
case "info":
|
||
|
level = zap.InfoLevel
|
||
|
default:
|
||
|
level = zap.InfoLevel
|
||
|
}
|
||
|
err := utils.MKDir("logs")
|
||
|
if err != nil {
|
||
|
panic("创建日志失败" + err.Error())
|
||
|
}
|
||
|
zapConfig := zap.NewProductionConfig()
|
||
|
zapConfig.Encoding = "console"
|
||
|
zapConfig.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
|
||
|
zapConfig.EncoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
|
||
|
zapConfig.OutputPaths = []string{"stdout", getLogFileName("info")}
|
||
|
zapConfig.ErrorOutputPaths = []string{"stderr", getLogFileName("error")}
|
||
|
zapConfig.Level = zap.NewAtomicLevelAt(level)
|
||
|
Logger, err = zapConfig.Build()
|
||
|
if err != nil {
|
||
|
panic("创建日志失败" + err.Error())
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// 根据日期获得日志文件
|
||
|
func getLogFileName(name string) string {
|
||
|
return filepath.Join("logs", time.Now().Format("2006-01-02")+"-"+name+".log")
|
||
|
}
|
||
|
|
||
|
func rotateLogs() {
|
||
|
for {
|
||
|
now := time.Now()
|
||
|
nextMidnight := time.Date(
|
||
|
now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location(),
|
||
|
).Add(24 * time.Hour)
|
||
|
durationUntilMidnight := nextMidnight.Sub(now)
|
||
|
|
||
|
time.Sleep(durationUntilMidnight)
|
||
|
buildLogger()
|
||
|
}
|
||
|
}
|