117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
var runningTest bool
|
|
|
|
type multiHandler struct {
|
|
handlers []slog.Handler
|
|
}
|
|
|
|
var runID string
|
|
|
|
func init() {
|
|
timestamp := time.Now().UnixNano()
|
|
pid := os.Getpid()
|
|
high := timestamp >> 32
|
|
low := timestamp & 0xFFFFFFFF
|
|
combined := high ^ low
|
|
combined ^= int64(pid)
|
|
combined ^= int64(timestamp >> 16)
|
|
combined ^= (combined >> 8)
|
|
combined ^= (combined << 16)
|
|
runID = fmt.Sprintf("%08X", combined&0xFFFFFFFF)
|
|
}
|
|
|
|
func (h *multiHandler) Enabled(ctx context.Context, level slog.Level) bool {
|
|
for _, handler := range h.handlers {
|
|
if handler.Enabled(ctx, level) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (h *multiHandler) Handle(ctx context.Context, r slog.Record) error {
|
|
for _, handler := range h.handlers {
|
|
if handler.Enabled(ctx, r.Level) {
|
|
if err := handler.Handle(ctx, r); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (h *multiHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
|
|
handlers := make([]slog.Handler, len(h.handlers))
|
|
for i, handler := range h.handlers {
|
|
handlers[i] = handler.WithAttrs(attrs)
|
|
}
|
|
return &multiHandler{handlers: handlers}
|
|
}
|
|
|
|
func (h *multiHandler) WithGroup(name string) slog.Handler {
|
|
handlers := make([]slog.Handler, len(h.handlers))
|
|
for i, handler := range h.handlers {
|
|
handlers[i] = handler.WithGroup(name)
|
|
}
|
|
return &multiHandler{handlers: handlers}
|
|
}
|
|
|
|
func setupSlog(logPath string) error {
|
|
handlers := []slog.Handler{}
|
|
if logPath != "" {
|
|
// Text file handler for debug logs
|
|
debugTextFile, err := os.OpenFile(logPath,
|
|
os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o640)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
debugTextHandler := slog.NewTextHandler(debugTextFile,
|
|
&slog.HandlerOptions{Level: slog.LevelDebug})
|
|
handlers = append(handlers, debugTextHandler)
|
|
// Json file handler for debug logs
|
|
debugJsonFile, err := os.OpenFile(logPath+".ndjson",
|
|
os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o640)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
debugJsonHandler := slog.NewJSONHandler(debugJsonFile,
|
|
&slog.HandlerOptions{Level: slog.LevelDebug})
|
|
handlers = append(handlers, debugJsonHandler)
|
|
}
|
|
stderrLogLevel := slog.LevelInfo
|
|
if runningTest {
|
|
stderrLogLevel = slog.LevelDebug
|
|
}
|
|
// Stderr handler for info logs and above
|
|
stderrHandler := slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
|
|
Level: stderrLogLevel,
|
|
})
|
|
handlers = append(handlers, stderrHandler)
|
|
if runningTest {
|
|
stderrJSONHandler := slog.NewJSONHandler(os.Stderr, &slog.HandlerOptions{
|
|
Level: stderrLogLevel,
|
|
})
|
|
handlers = append(handlers, stderrJSONHandler)
|
|
}
|
|
// Create a multi-handler
|
|
multiHandler := &multiHandler{handlers: handlers}
|
|
multiHandlerWithAttrs := multiHandler.WithAttrs(
|
|
[]slog.Attr{slog.String("runID", runID)})
|
|
// Set the default logger
|
|
logger := slog.New(multiHandlerWithAttrs)
|
|
slog.SetDefault(logger)
|
|
if logPath != "" {
|
|
slog.Info("debug log", "path", logPath)
|
|
}
|
|
return nil
|
|
}
|