72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
|
package logger
|
||
|
|
||
|
import (
|
||
|
"io"
|
||
|
"log/slog"
|
||
|
"os"
|
||
|
)
|
||
|
|
||
|
// Builder object
|
||
|
type LoggerBuilder struct {
|
||
|
lvl slog.Level
|
||
|
writers []io.Writer
|
||
|
}
|
||
|
|
||
|
// NewBuilder return a new logger builder object
|
||
|
func NewBuilder() *LoggerBuilder {
|
||
|
return new(LoggerBuilder)
|
||
|
}
|
||
|
|
||
|
// WithWriter sets a specific writer
|
||
|
func (b *LoggerBuilder) WithWriter(w io.Writer) *LoggerBuilder {
|
||
|
b.writers = append(b.writers, w)
|
||
|
|
||
|
return b
|
||
|
}
|
||
|
|
||
|
// WithLevel sets log level
|
||
|
func (b *LoggerBuilder) WithLevel(l slog.Level) *LoggerBuilder {
|
||
|
b.lvl = l
|
||
|
|
||
|
return b
|
||
|
}
|
||
|
|
||
|
// Build returns the logger
|
||
|
func (b *LoggerBuilder) Build() *slog.Logger {
|
||
|
if len(b.writers) == 0 {
|
||
|
b.writers = append(b.writers, os.Stdout)
|
||
|
}
|
||
|
|
||
|
w := io.MultiWriter(b.writers...)
|
||
|
|
||
|
return newLogger(b.lvl, w)
|
||
|
}
|
||
|
|
||
|
func newLogger(lvl slog.Level, w io.Writer) *slog.Logger {
|
||
|
return slog.New(
|
||
|
slog.NewJSONHandler(w, &slog.HandlerOptions{Level: lvl}),
|
||
|
)
|
||
|
}
|
||
|
|
||
|
// Error logging attribute
|
||
|
func Err(err error) slog.Attr {
|
||
|
return slog.Attr{
|
||
|
Key: "error",
|
||
|
Value: slog.StringValue(err.Error()),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Maps level from a string. By default returns slog.LevelInfo
|
||
|
func MapLevel(lvl string) slog.Level {
|
||
|
switch lvl {
|
||
|
case "dev", "local", "debug":
|
||
|
return slog.LevelDebug
|
||
|
case "warn":
|
||
|
return slog.LevelWarn
|
||
|
case "error":
|
||
|
return slog.LevelError
|
||
|
default:
|
||
|
return slog.LevelInfo
|
||
|
}
|
||
|
}
|