ᴍᴏᴏɴD4ʀᴋ
3 years ago
parent
47921a9eec
commit
c9feef66ab
@ -1,107 +1,119 @@ |
||||
package log |
||||
|
||||
import ( |
||||
"fmt" |
||||
"io" |
||||
"log" |
||||
"os" |
||||
) |
||||
|
||||
type Level int |
||||
|
||||
const ( |
||||
LevelDebug Level = iota |
||||
LevelWarn |
||||
LevelError |
||||
"github.com/gookit/color" |
||||
"github.com/gookit/slog" |
||||
) |
||||
|
||||
func (l Level) String() string { |
||||
switch l { |
||||
case LevelDebug: |
||||
return "debug" |
||||
case LevelError: |
||||
return "error" |
||||
var std = &slog.SugaredLogger{} |
||||
|
||||
func Init(l string) { |
||||
if l == "debug" { |
||||
std = newStdLogger(slog.DebugLevel) |
||||
} else { |
||||
std = newStdLogger(slog.ErrorLevel) |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
var ( |
||||
formatLogger *Logger |
||||
levelMap = map[string]Level{ |
||||
"debug": LevelDebug, |
||||
"error": LevelError, |
||||
} |
||||
) |
||||
const template = "[{{datetime}}] [{{level}}] [{{caller}}] {{message}} {{data}} {{extra}}\n" |
||||
|
||||
// NewStdLogger instance
|
||||
func newStdLogger(level slog.Level) *slog.SugaredLogger { |
||||
return slog.NewSugaredLogger(os.Stdout, level).Configure(func(sl *slog.SugaredLogger) { |
||||
sl.SetName("stdLogger") |
||||
sl.ReportCaller = true |
||||
sl.CallerSkip = 3 |
||||
// auto enable console color
|
||||
sl.Formatter.(*slog.TextFormatter).EnableColor = color.SupportColor() |
||||
sl.Formatter.(*slog.TextFormatter).SetTemplate(template) |
||||
}) |
||||
} |
||||
|
||||
func InitLog(l string) { |
||||
formatLogger = newLog(os.Stdout).setLevel(levelMap[l]).setFlags(log.Lshortfile) |
||||
// Trace logs a message at level Trace
|
||||
func Trace(args ...interface{}) { |
||||
std.Log(slog.TraceLevel, args...) |
||||
} |
||||
|
||||
type Logger struct { |
||||
level Level |
||||
l *log.Logger |
||||
// Tracef logs a message at level Trace
|
||||
func Tracef(format string, args ...interface{}) { |
||||
std.Logf(slog.TraceLevel, format, args...) |
||||
} |
||||
|
||||
func newLog(w io.Writer) *Logger { |
||||
return &Logger{ |
||||
l: log.New(w, "", 0), |
||||
} |
||||
// Info logs a message at level Info
|
||||
func Info(args ...interface{}) { |
||||
std.Log(slog.InfoLevel, args...) |
||||
} |
||||
|
||||
func (l *Logger) setFlags(flag int) *Logger { |
||||
l.l.SetFlags(flag) |
||||
return l |
||||
// Infof logs a message at level Info
|
||||
func Infof(format string, args ...interface{}) { |
||||
std.Logf(slog.InfoLevel, format, args...) |
||||
} |
||||
|
||||
func (l *Logger) setLevel(level Level) *Logger { |
||||
l.level = level |
||||
return l |
||||
// Notice logs a message at level Notice
|
||||
func Notice(args ...interface{}) { |
||||
std.Log(slog.NoticeLevel, args...) |
||||
} |
||||
|
||||
func (l *Logger) doLog(level Level, v ...interface{}) bool { |
||||
if level < l.level { |
||||
return false |
||||
} |
||||
l.l.Output(3, level.String()+" "+fmt.Sprintln(v...)) |
||||
return true |
||||
// Noticef logs a message at level Notice
|
||||
func Noticef(format string, args ...interface{}) { |
||||
std.Logf(slog.NoticeLevel, format, args...) |
||||
} |
||||
|
||||
func (l *Logger) doLogf(level Level, format string, v ...interface{}) bool { |
||||
if level < l.level { |
||||
return false |
||||
// Warn logs a message at level Warn
|
||||
func Warn(args ...interface{}) { |
||||
std.Log(slog.WarnLevel, args...) |
||||
} |
||||
|
||||
// Warnf logs a message at level Warn
|
||||
func Warnf(format string, args ...interface{}) { |
||||
std.Logf(slog.WarnLevel, format, args...) |
||||
} |
||||
|
||||
// Error logs a message at level Error
|
||||
func Error(args ...interface{}) { |
||||
std.Log(slog.ErrorLevel, args...) |
||||
} |
||||
|
||||
// ErrorT logs a error type at level Error
|
||||
func ErrorT(err error) { |
||||
if err != nil { |
||||
std.Log(slog.ErrorLevel, err) |
||||
} |
||||
l.l.Output(3, level.String()+" "+fmt.Sprintln(fmt.Sprintf(format, v...))) |
||||
return true |
||||
} |
||||
|
||||
func Debug(v ...interface{}) { |
||||
formatLogger.doLog(LevelDebug, v...) |
||||
// Errorf logs a message at level Error
|
||||
func Errorf(format string, args ...interface{}) { |
||||
std.Logf(slog.ErrorLevel, format, args...) |
||||
} |
||||
|
||||
func Warn(v ...interface{}) { |
||||
formatLogger.doLog(LevelWarn, v...) |
||||
// Debug logs a message at level Debug
|
||||
func Debug(args ...interface{}) { |
||||
std.Log(slog.DebugLevel, args...) |
||||
} |
||||
|
||||
func Error(v ...interface{}) { |
||||
formatLogger.doLog(LevelError, v...) |
||||
// Debugf logs a message at level Debug
|
||||
func Debugf(format string, args ...interface{}) { |
||||
std.Logf(slog.DebugLevel, format, args...) |
||||
} |
||||
|
||||
func Errorf(format string, v ...interface{}) { |
||||
formatLogger.doLogf(LevelError, format, v...) |
||||
// Fatal logs a message at level Fatal
|
||||
func Fatal(args ...interface{}) { |
||||
std.Log(slog.FatalLevel, args...) |
||||
} |
||||
|
||||
func Warnf(format string, v ...interface{}) { |
||||
formatLogger.doLogf(LevelWarn, format, v...) |
||||
// Fatalf logs a message at level Fatal
|
||||
func Fatalf(format string, args ...interface{}) { |
||||
std.Logf(slog.FatalLevel, format, args...) |
||||
} |
||||
|
||||
func Debugf(format string, v ...interface{}) { |
||||
formatLogger.doLogf(LevelDebug, format, v...) |
||||
// Panic logs a message at level Panic
|
||||
func Panic(args ...interface{}) { |
||||
std.Log(slog.PanicLevel, args...) |
||||
} |
||||
|
||||
// NewSugaredLogger(os.Stdout, DebugLevel).Configure(func(sl *SugaredLogger) {
|
||||
// sl.SetName("stdLogger")
|
||||
// sl.ReportCaller = true
|
||||
// // auto enable console color
|
||||
// sl.Formatter.(*TextFormatter).EnableColor = color.SupportColor()
|
||||
// sl.Formatter.SetCallerSkip(1)
|
||||
// })
|
||||
// Panicf logs a message at level Panic
|
||||
func Panicf(format string, args ...interface{}) { |
||||
std.Logf(slog.PanicLevel, format, args...) |
||||
} |
||||
|
||||
Loading…
Reference in new issue