onasty: internal/logger/logger.go (51d3b53b0c8469cfd0ca12bb2a505d7fe77d228e)

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

Smirnov Oleksandr feat: add password support to notes (#41)..., 1 year ago

package logger

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

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

type CustomLogger struct{ slog.Handler }

//nolint:err113
func NewCustomLogger(lvl, format string, showLine bool) (*slog.Logger, 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 nil, errors.New("unknown log level")
	}

	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 nil, errors.New("unknown log format")
	}

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

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	type CustomLogger struct{ slog.Handler }
13
14	//nolint:err113
15	func NewCustomLogger(lvl, format string, showLine bool) (*slog.Logger, error) {
16	loggerLevels := map[string]slog.Level{
17	"info": slog.LevelInfo,
18	"debug": slog.LevelDebug,
19	"error": slog.LevelError,
20	"warn": slog.LevelWarn,
21	}
22
23	logLevel, ok := loggerLevels[lvl]
24	if !ok {
25	return nil, errors.New("unknown log level")
26	}
27
28	handlerOptions := &slog.HandlerOptions{
29	Level: logLevel,
30	AddSource: showLine,
31	}
32
33	var slogHandler slog.Handler
34	switch format {
35	case "json":
36	slogHandler = slog.NewJSONHandler(os.Stdout, handlerOptions)
37	case "text", "txt":
38	slogHandler = slog.NewTextHandler(os.Stdout, handlerOptions)
39	default:
40	return nil, errors.New("unknown log format")
41	}
42
43	return slog.New(&CustomLogger{Handler: slogHandler}), nil
44	}
45
46	func (l *CustomLogger) Handle(ctx context.Context, r slog.Record) error {
47	if requestID := reqid.GetContext(ctx); requestID != "" {
48	r.AddAttrs(slog.String("request_id", requestID))
49	}
50
51	return l.Handler.Handle(ctx, r)
52	}