onasty: internal/logger/logger.go (320377a)

onasty/internal/logger/logger.go (view raw)

Smirnov Oleksandr refactor: logger set default interface (#131)..., 12 months ago

package logger

import (
	"context"
	"errors"
	"log/slog"
	"os"

	"github.com/olexsmir/onasty/internal/transport/http/reqid"
)

var (
	ErrUnknownLevel  = errors.New("unknown log level")
	ErrUnknownFormat = errors.New("unknown log format")
)

// SetDefault configures and set default [slog.Logger]
func SetDefault(lvl, format string, showLine bool) error {
	loggerLevels := map[string]slog.Level{
		"info":  slog.LevelInfo,
		"debug": slog.LevelDebug,
		"error": slog.LevelError,
		"warn":  slog.LevelWarn,
	}

	logLevel, ok := loggerLevels[lvl]
	if !ok {
		return ErrUnknownLevel
	}

	handlerOptions := &slog.HandlerOptions{
		Level:     logLevel,
		AddSource: showLine,
	}

	var slogHandler slog.Handler
	switch format {
	case "json":
		slogHandler = slog.NewJSONHandler(os.Stdout, handlerOptions)
	case "text", "txt":
		slogHandler = slog.NewTextHandler(os.Stdout, handlerOptions)
	default:
		return ErrUnknownFormat
	}

	slog.SetDefault(slog.New(&CustomLogger{Handler: slogHandler}))
	return nil
}

type CustomLogger struct{ slog.Handler }

func (l *CustomLogger) Handle(ctx context.Context, r slog.Record) error {
	if requestID := reqid.GetContext(ctx); requestID != "" {
		r.AddAttrs(slog.String("request_id", requestID))
	}

	return l.Handler.Handle(ctx, r)
}

1	package logger
2
3	import (
4	"context"
5	"errors"
6	"log/slog"
7	"os"
8
9	"github.com/olexsmir/onasty/internal/transport/http/reqid"
10	)
11
12	var (
13	ErrUnknownLevel = errors.New("unknown log level")
14	ErrUnknownFormat = errors.New("unknown log format")
15	)
16
17	// SetDefault configures and set default [slog.Logger]
18	func SetDefault(lvl, format string, showLine bool) error {
19	loggerLevels := map[string]slog.Level{
20	"info": slog.LevelInfo,
21	"debug": slog.LevelDebug,
22	"error": slog.LevelError,
23	"warn": slog.LevelWarn,
24	}
25
26	logLevel, ok := loggerLevels[lvl]
27	if !ok {
28	return ErrUnknownLevel
29	}
30
31	handlerOptions := &slog.HandlerOptions{
32	Level: logLevel,
33	AddSource: showLine,
34	}
35
36	var slogHandler slog.Handler
37	switch format {
38	case "json":
39	slogHandler = slog.NewJSONHandler(os.Stdout, handlerOptions)
40	case "text", "txt":
41	slogHandler = slog.NewTextHandler(os.Stdout, handlerOptions)
42	default:
43	return ErrUnknownFormat
44	}
45
46	slog.SetDefault(slog.New(&CustomLogger{Handler: slogHandler}))
47	return nil
48	}
49
50	type CustomLogger struct{ slog.Handler }
51
52	func (l *CustomLogger) Handle(ctx context.Context, r slog.Record) error {
53	if requestID := reqid.GetContext(ctx); requestID != "" {
54	r.AddAttrs(slog.String("request_id", requestID))
55	}
56
57	return l.Handler.Handle(ctx, r)
58	}