Safe3-openresty-manager/pkg/util/log.go
UUSEC Technology 12467d8fd1 open source
2025-04-30 13:05:17 +08:00

49 lines
1.1 KiB
Go

package util
import (
"io"
"path"
"runtime"
"strconv"
"time"
log "github.com/sirupsen/logrus"
"gopkg.in/natefinch/lumberjack.v2"
)
// InitLog parses and sets log-level input
func InitLog(logLevel, logPath string) error {
level, err := log.ParseLevel(logLevel)
if err != nil {
log.Errorf("Failed parsing log-level %s: %s", logLevel, err)
return err
}
lumberjackLogger := &lumberjack.Logger{
// Log file absolute path, os agnostic
Filename: logPath,
MaxSize: 100, // MB
MaxBackups: 10,
MaxAge: 30, // days
Compress: true,
}
log.SetOutput(io.Writer(lumberjackLogger))
logFormatter := new(log.TextFormatter)
logFormatter.TimestampFormat = time.RFC3339 // or RFC3339
logFormatter.FullTimestamp = true
logFormatter.CallerPrettyfier = func(frame *runtime.Frame) (function string, file string) {
fileName := path.Base(frame.File) + ":" + strconv.Itoa(frame.Line)
//return frame.Function, fileName
return "", fileName
}
if level == log.DebugLevel {
log.SetReportCaller(true)
}
log.SetFormatter(logFormatter)
log.SetLevel(level)
return nil
}