178 lines
4.0 KiB
Go
178 lines
4.0 KiB
Go
package log
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// Defaults for package level variables
|
|
var (
|
|
DefaultLevel = InfoLevel
|
|
DefaultFormat = TextFormat
|
|
DefaultOutput = os.Stderr
|
|
)
|
|
|
|
var (
|
|
level = DefaultLevel
|
|
format = DefaultFormat
|
|
overrides = make(map[string]*levelOverride)
|
|
output io.Writer = DefaultOutput
|
|
mu = &sync.Mutex{}
|
|
)
|
|
|
|
// Fatal prints the given message, and exits the program
|
|
func Fatal(message string, v ...any) {
|
|
newEvent().Fatal(message, v...)
|
|
}
|
|
|
|
// Error prints the given message, if the current log level is ERROR or lower
|
|
func Error(message string, v ...any) {
|
|
newEvent().Error(message, v...)
|
|
}
|
|
|
|
// Warn prints the given message, if the current log level is WARN or lower
|
|
func Warn(message string, v ...any) {
|
|
newEvent().Warn(message, v...)
|
|
}
|
|
|
|
// Info prints the given message, if the current log level is INFO or lower
|
|
func Info(message string, v ...any) {
|
|
newEvent().Info(message, v...)
|
|
}
|
|
|
|
// Debug prints the given message, if the current log level is DEBUG or lower
|
|
func Debug(message string, v ...any) {
|
|
newEvent().Debug(message, v...)
|
|
}
|
|
|
|
// Trace prints the given message, if the current log level is TRACE
|
|
func Trace(message string, v ...any) {
|
|
newEvent().Trace(message, v...)
|
|
}
|
|
|
|
// With creates a new log event and adds the fields of the given Contexter structs
|
|
func With(contexts ...Contexter) *Event {
|
|
return newEvent().With(contexts...)
|
|
}
|
|
|
|
// Field creates a new log event and adds a custom field and value to it
|
|
func Field(key string, value any) *Event {
|
|
return newEvent().Field(key, value)
|
|
}
|
|
|
|
// Fields creates a new log event and adds a map of fields to it
|
|
func Fields(fields Context) *Event {
|
|
return newEvent().Fields(fields)
|
|
}
|
|
|
|
// Tag creates a new log event and adds a "tag" field to it
|
|
func Tag(tag string) *Event {
|
|
return newEvent().Tag(tag)
|
|
}
|
|
|
|
// Time creates a new log event and sets the time field
|
|
func Time(time time.Time) *Event {
|
|
return newEvent().Time(time)
|
|
}
|
|
|
|
// Timing runs f and records the time if took to execute it in "time_taken_ms"
|
|
func Timing(f func()) *Event {
|
|
return newEvent().Timing(f)
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
// SetLevelOverride adds a log override for the given field
|
|
func SetLevelOverride(field string, value string, level Level) {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
overrides[field] = &levelOverride{value: value, level: level}
|
|
}
|
|
|
|
// ResetLevelOverrides removes all log level overrides
|
|
func ResetLevelOverrides() {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
overrides = make(map[string]*levelOverride)
|
|
}
|
|
|
|
// CurrentFormat returns the current log formt
|
|
func CurrentFormat() Format {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
return format
|
|
}
|
|
|
|
// SetFormat sets a new log format
|
|
func SetFormat(newFormat Format) {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
format = newFormat
|
|
if newFormat == JSONFormat {
|
|
DisableDates()
|
|
}
|
|
}
|
|
|
|
// SetOutput sets the log output writer
|
|
func SetOutput(w io.Writer) {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
log.SetOutput(w)
|
|
output = w
|
|
}
|
|
|
|
// File returns the log file, if any, or an empty string otherwise
|
|
func File() string {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
if f, ok := output.(*os.File); ok {
|
|
return f.Name()
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// IsFile returns true if the output is a non-default file
|
|
func IsFile() bool {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
if _, ok := output.(*os.File); ok && output != DefaultOutput {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
// DisableDates disables the date/time prefix
|
|
func DisableDates() {
|
|
log.SetFlags(0)
|
|
}
|
|
|
|
// 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)
|
|
}
|