117 lines
2.1 KiB
Go
117 lines
2.1 KiB
Go
package debug
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log/slog"
|
|
"os"
|
|
)
|
|
|
|
const (
|
|
DEBUG_CRITICAL = 1
|
|
DEBUG_ERROR = 2
|
|
DEBUG_INFO = 3
|
|
DEBUG_VERBOSE = 4
|
|
DEBUG_TRACE = 5
|
|
DEBUG_PACKETS = 6
|
|
DEBUG_ALL = 7
|
|
)
|
|
|
|
var (
|
|
debugLevel = flag.Int("debug", 3, "debug level (1-7)")
|
|
logger *slog.Logger
|
|
initialized bool
|
|
)
|
|
|
|
func Init() {
|
|
if initialized {
|
|
return
|
|
}
|
|
initialized = true
|
|
|
|
var level slog.Level
|
|
switch {
|
|
case *debugLevel >= DEBUG_ALL:
|
|
level = slog.LevelDebug
|
|
case *debugLevel >= DEBUG_PACKETS:
|
|
level = slog.LevelDebug
|
|
case *debugLevel >= DEBUG_TRACE:
|
|
level = slog.LevelDebug
|
|
case *debugLevel >= DEBUG_VERBOSE:
|
|
level = slog.LevelDebug
|
|
case *debugLevel >= DEBUG_INFO:
|
|
level = slog.LevelInfo
|
|
case *debugLevel >= DEBUG_ERROR:
|
|
level = slog.LevelWarn
|
|
case *debugLevel >= DEBUG_CRITICAL:
|
|
level = slog.LevelError
|
|
default:
|
|
level = slog.LevelError
|
|
}
|
|
|
|
opts := &slog.HandlerOptions{
|
|
Level: level,
|
|
}
|
|
logger = slog.New(slog.NewTextHandler(os.Stderr, opts))
|
|
slog.SetDefault(logger)
|
|
}
|
|
|
|
func GetLogger() *slog.Logger {
|
|
if !initialized {
|
|
Init()
|
|
}
|
|
return logger
|
|
}
|
|
|
|
func Log(level int, msg string, args ...interface{}) {
|
|
if !initialized {
|
|
Init()
|
|
}
|
|
|
|
if *debugLevel < level {
|
|
return
|
|
}
|
|
|
|
var slogLevel slog.Level
|
|
switch {
|
|
case level >= DEBUG_ALL:
|
|
slogLevel = slog.LevelDebug
|
|
case level >= DEBUG_PACKETS:
|
|
slogLevel = slog.LevelDebug
|
|
case level >= DEBUG_TRACE:
|
|
slogLevel = slog.LevelDebug
|
|
case level >= DEBUG_VERBOSE:
|
|
slogLevel = slog.LevelDebug
|
|
case level >= DEBUG_INFO:
|
|
slogLevel = slog.LevelInfo
|
|
case level >= DEBUG_ERROR:
|
|
slogLevel = slog.LevelWarn
|
|
case level >= DEBUG_CRITICAL:
|
|
slogLevel = slog.LevelError
|
|
default:
|
|
slogLevel = slog.LevelError
|
|
}
|
|
|
|
if !logger.Enabled(context.TODO(), slogLevel) {
|
|
return
|
|
}
|
|
|
|
allArgs := make([]interface{}, len(args)+2)
|
|
copy(allArgs, args)
|
|
allArgs[len(args)] = "debug_level"
|
|
allArgs[len(args)+1] = level
|
|
logger.Log(context.TODO(), slogLevel, msg, allArgs...)
|
|
}
|
|
|
|
func SetDebugLevel(level int) {
|
|
*debugLevel = level
|
|
if initialized {
|
|
Init()
|
|
}
|
|
}
|
|
|
|
func GetDebugLevel() int {
|
|
return *debugLevel
|
|
}
|
|
|