forked from mirrors/ntfy
129 lines
2.5 KiB
Go
129 lines
2.5 KiB
Go
package log
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
// Level is a well-known log level, as defined below
|
|
type Level int
|
|
|
|
// Well known log levels
|
|
const (
|
|
TraceLevel Level = iota
|
|
DebugLevel
|
|
InfoLevel
|
|
WarnLevel
|
|
ErrorLevel
|
|
)
|
|
|
|
func (l Level) String() string {
|
|
switch l {
|
|
case TraceLevel:
|
|
return "TRACE"
|
|
case DebugLevel:
|
|
return "DEBUG"
|
|
case InfoLevel:
|
|
return "INFO"
|
|
case WarnLevel:
|
|
return "WARN"
|
|
case ErrorLevel:
|
|
return "ERROR"
|
|
}
|
|
return "unknown"
|
|
}
|
|
|
|
var (
|
|
level = InfoLevel
|
|
mu = &sync.Mutex{}
|
|
)
|
|
|
|
// Trace prints the given message, if the current log level is TRACE
|
|
func Trace(message string, v ...any) {
|
|
logIf(TraceLevel, message, v...)
|
|
}
|
|
|
|
// Debug prints the given message, if the current log level is DEBUG or lower
|
|
func Debug(message string, v ...any) {
|
|
logIf(DebugLevel, message, v...)
|
|
}
|
|
|
|
// Info prints the given message, if the current log level is INFO or lower
|
|
func Info(message string, v ...any) {
|
|
logIf(InfoLevel, message, v...)
|
|
}
|
|
|
|
// Warn prints the given message, if the current log level is WARN or lower
|
|
func Warn(message string, v ...any) {
|
|
logIf(WarnLevel, message, v...)
|
|
}
|
|
|
|
// Error prints the given message, if the current log level is ERROR or lower
|
|
func Error(message string, v ...any) {
|
|
logIf(ErrorLevel, message, v...)
|
|
}
|
|
|
|
// Fatal prints the given message, and exits the program
|
|
func Fatal(v ...any) {
|
|
log.Fatalln(v...)
|
|
}
|
|
|
|
// CurrentLevel returns the current log level
|
|
func CurrentLevel() Level {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
return level
|
|
}
|
|
|
|
// SetLevel sets a new log level
|
|
func SetLevel(newLevel Level) {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
level = newLevel
|
|
}
|
|
|
|
// DisableDates disables the date/time prefix
|
|
func DisableDates() {
|
|
log.SetFlags(0)
|
|
}
|
|
|
|
// ToLevel converts a string to a Level. It returns InfoLevel if the string
|
|
// does not match any known log levels.
|
|
func ToLevel(s string) Level {
|
|
switch strings.ToUpper(s) {
|
|
case "TRACE":
|
|
return TraceLevel
|
|
case "DEBUG":
|
|
return DebugLevel
|
|
case "INFO":
|
|
return InfoLevel
|
|
case "WARN", "WARNING":
|
|
return WarnLevel
|
|
case "ERROR":
|
|
return ErrorLevel
|
|
default:
|
|
return InfoLevel
|
|
}
|
|
}
|
|
|
|
// Loggable returns true if the given log level is lower or equal to the current log level
|
|
func Loggable(l Level) bool {
|
|
return CurrentLevel() <= l
|
|
}
|
|
|
|
// IsTrace returns true if the current log level is TraceLevel
|
|
func IsTrace() bool {
|
|
return Loggable(TraceLevel)
|
|
}
|
|
|
|
// IsDebug returns true if the current log level is DebugLevel or below
|
|
func IsDebug() bool {
|
|
return Loggable(DebugLevel)
|
|
}
|
|
|
|
func logIf(l Level, message string, v ...any) {
|
|
if CurrentLevel() <= l {
|
|
log.Printf(l.String()+" "+message, v...)
|
|
}
|
|
}
|