117 lines
2.5 KiB
Go
117 lines
2.5 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
)
|
|
|
|
type Logger interface {
|
|
Error(format string, args ...any)
|
|
Warning(format string, args ...any)
|
|
Notice(format string, args ...any)
|
|
Info(format string, args ...any)
|
|
Debug(format string, args ...any)
|
|
Trace(format string, args ...any)
|
|
}
|
|
|
|
// Dummy Logger does nothing.
|
|
type DummyLogger struct{}
|
|
|
|
func (DummyLogger) Error(format string, args ...any) {
|
|
}
|
|
|
|
func (DummyLogger) Warning(format string, args ...any) {
|
|
}
|
|
|
|
func (DummyLogger) Notice(format string, args ...any) {
|
|
}
|
|
|
|
func (DummyLogger) Info(format string, args ...any) {
|
|
}
|
|
|
|
func (DummyLogger) Debug(format string, args ...any) {
|
|
}
|
|
|
|
func (DummyLogger) Trace(format string, args ...any) {
|
|
}
|
|
|
|
// Simple Console Logger that the tests use.
|
|
type LogLevel int
|
|
|
|
const (
|
|
LogLevelTrace LogLevel = 5
|
|
LogLevelDebug LogLevel = 4
|
|
LogLevelInfo LogLevel = 3
|
|
LogLevelNotice LogLevel = 2
|
|
LogLevelWarning LogLevel = 1
|
|
LogLevelError LogLevel = 0
|
|
)
|
|
|
|
type ConsoleLogger struct {
|
|
LogLevel LogLevel
|
|
}
|
|
|
|
func NewConsoleLogger(logLevel LogLevel) *ConsoleLogger {
|
|
logger := ConsoleLogger{}
|
|
logger.LogLevel = logLevel
|
|
return &logger
|
|
}
|
|
|
|
func (cL ConsoleLogger) Error(format string, args ...any) {
|
|
if cL.LogLevel >= LogLevelError {
|
|
cL.output(os.Stdout, "[ERROR] ", format, args...)
|
|
}
|
|
}
|
|
|
|
func (cL ConsoleLogger) Warning(format string, args ...any) {
|
|
if cL.LogLevel >= LogLevelWarning {
|
|
cL.output(os.Stdout, "[WARNING] ", format, args...)
|
|
}
|
|
}
|
|
|
|
func (cL ConsoleLogger) Notice(format string, args ...any) {
|
|
if cL.LogLevel >= LogLevelNotice {
|
|
cL.output(os.Stdout, "[NOTICE] ", format, args...)
|
|
}
|
|
}
|
|
|
|
func (cL ConsoleLogger) Info(format string, args ...any) {
|
|
if cL.LogLevel >= LogLevelInfo {
|
|
cL.output(os.Stdout, "[INFO] ", format, args...)
|
|
}
|
|
}
|
|
|
|
func (cL ConsoleLogger) Debug(format string, args ...any) {
|
|
if cL.LogLevel >= LogLevelDebug {
|
|
cL.output(os.Stdout, "[DEBUG] ", format, args...)
|
|
}
|
|
}
|
|
|
|
func (cL ConsoleLogger) Trace(format string, args ...any) {
|
|
if cL.LogLevel >= LogLevelTrace {
|
|
cL.output(os.Stdout, "[TRACE] ", format, args...)
|
|
}
|
|
}
|
|
|
|
var Log Logger = DummyLogger{}
|
|
|
|
func SetLogger(logger Logger) {
|
|
Log = logger
|
|
}
|
|
|
|
// output writes `format`, `args` log message prefixed by the source file name, line and `prefix`
|
|
func (ConsoleLogger) output(f *os.File, prefix string, format string, args ...any) {
|
|
_, file, line, ok := runtime.Caller(3)
|
|
if !ok {
|
|
file = "???"
|
|
line = 0
|
|
} else {
|
|
file = filepath.Base(file)
|
|
}
|
|
|
|
src := fmt.Sprintf("%s %s:%d ", prefix, file, line) + format + "\n"
|
|
fmt.Fprintf(f, src, args...)
|
|
}
|