72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package log
|
||
|
||
import (
|
||
"io"
|
||
"time"
|
||
|
||
rotatelogs "github.com/lestrrat/go-file-rotatelogs"
|
||
)
|
||
|
||
type WriteSyncer interface {
|
||
io.Writer
|
||
Sync() error
|
||
}
|
||
|
||
// 得到LogWriter
|
||
func getLogWriter(filePath string, period, count int) WriteSyncer {
|
||
warnIoWriter := getWriter(filePath, period, count)
|
||
return addSync(warnIoWriter)
|
||
}
|
||
|
||
// 日志文件切割
|
||
func getWriter(filename string, period, count int) io.Writer {
|
||
// 保存日志count天，每period小时分割一次日志
|
||
duration := time.Hour * time.Duration(period)
|
||
var logfile string
|
||
if period >= 24 {
|
||
logfile = filename + "-%Y%m%d"
|
||
} else {
|
||
logfile = filename + "-%Y%m%d%H"
|
||
}
|
||
hook, err := rotatelogs.New(
|
||
|
||
logfile,
|
||
rotatelogs.WithLinkName(filename),
|
||
// rotatelogs.WithMaxAge(duration),
|
||
rotatelogs.WithRotationCount(count),
|
||
rotatelogs.WithRotationTime(duration),
|
||
rotatelogs.WithLocation(time.Local),
|
||
)
|
||
|
||
//保存日志30天，每1分钟分割一次日志
|
||
/*
|
||
hook, err := rotatelogs.New(
|
||
filename+"_%Y%m%d%H%M.log",
|
||
rotatelogs.WithLinkName(filename),
|
||
rotatelogs.WithMaxAge(time.Hour*24*30),
|
||
rotatelogs.WithRotationTime(time.Minute*1),
|
||
)
|
||
*/
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return hook
|
||
}
|
||
|
||
func addSync(w io.Writer) WriteSyncer {
|
||
switch w := w.(type) {
|
||
case WriteSyncer:
|
||
return w
|
||
default:
|
||
return writerWrapper{w}
|
||
}
|
||
}
|
||
|
||
type writerWrapper struct {
|
||
io.Writer
|
||
}
|
||
|
||
func (w writerWrapper) Sync() error {
|
||
return nil
|
||
}
|