Redmage/log/log.go

152 lines
3.9 KiB
Go
Raw Normal View History

2024-04-06 01:22:00 +07:00
package log
import (
"context"
"fmt"
"io"
"log/slog"
"os"
"runtime"
"strings"
"time"
"github.com/go-chi/chi/v5/middleware"
"github.com/mattn/go-colorable"
"github.com/mattn/go-isatty"
"github.com/tigorlazuardi/redmage/config"
)
var handler slog.Handler = NullHandler{}
func NewHandler(cfg *config.Config) slog.Handler {
if !cfg.Bool("log.enable") {
return NullHandler{}
}
var output io.Writer
if strings.ToLower(cfg.String("log.output")) == "stdout" {
output = colorable.NewColorableStdout()
} else {
output = colorable.NewColorableStderr()
}
var lvl slog.Level
_ = lvl.UnmarshalText(cfg.Bytes("log.level"))
opts := &slog.HandlerOptions{
AddSource: cfg.Bool("log.source"),
Level: lvl,
}
format := strings.ToLower(cfg.String("log.format"))
if isatty.IsTerminal(os.Stdout.Fd()) && format == "pretty" {
return NewPrettyHandler(output, opts)
} else {
return slog.NewJSONHandler(output, opts)
}
}
type Entry struct {
ctx context.Context
handler slog.Handler
caller uintptr
time time.Time
}
// Log prepares a new entry to write logs.
func Log(ctx context.Context) *Entry {
h := FromContext(ctx)
if h == nil {
h = handler
}
return &Entry{ctx: ctx, handler: h, time: time.Now()}
}
func (entry *Entry) Caller(pc uintptr) *Entry {
entry.caller = pc
return entry
}
func (entry *Entry) Info(message string, fields ...any) {
record := slog.NewRecord(entry.time, slog.LevelInfo, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
record.AddAttrs(slog.Group("context", fields...))
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) Infof(format string, args ...any) {
message := fmt.Sprintf(format, args...)
record := slog.NewRecord(entry.time, slog.LevelInfo, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) Error(message string, fields ...any) {
record := slog.NewRecord(entry.time, slog.LevelError, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
record.AddAttrs(slog.Group("context", fields...))
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) Errorf(format string, args ...any) {
message := fmt.Sprintf(format, args...)
record := slog.NewRecord(entry.time, slog.LevelError, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) Debug(message string, fields ...any) {
record := slog.NewRecord(entry.time, slog.LevelDebug, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
record.AddAttrs(slog.Group("context", fields...))
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) Debugf(format string, args ...any) {
message := fmt.Sprintf(format, args...)
record := slog.NewRecord(entry.time, slog.LevelDebug, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) Warn(message string, fields ...any) {
record := slog.NewRecord(entry.time, slog.LevelWarn, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
record.AddAttrs(slog.Group("context", fields...))
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) Warnf(format string, args ...any) {
message := fmt.Sprintf(format, args...)
record := slog.NewRecord(entry.time, slog.LevelWarn, message, entry.getCaller())
record.AddAttrs(entry.getExtra()...)
_ = entry.handler.Handle(entry.ctx, record)
}
func (entry *Entry) getCaller() uintptr {
if entry.caller != 0 {
return entry.caller
}
return GetCaller(4)
}
func GetCaller(skip int) uintptr {
pc := make([]uintptr, 1)
n := runtime.Callers(skip, pc)
if n == 0 {
return 0
}
return pc[0]
}
func (entry *Entry) getExtra() []slog.Attr {
out := make([]slog.Attr, 0, 1)
if reqid := middleware.GetReqID(entry.ctx); reqid != "" {
out = append(out, slog.String("request.id", reqid))
}
return out
}
func SetDefault(h slog.Handler) {
handler = h
}