make the log levels customizable

samber · Oct 25, 2023 · f2fde65 · f2fde65
1 parent 44caf28
commit f2fde65
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -73,6 +73,12 @@ Attributes will be injected in log payload.
 
 Attributes added to records are not accepted.
 
+Other global parameters:
+
+```go
+slogloki.LogLevels = map[slog.Level]promtail.LogLevel{...}
+```
+
 ### Example
 
 ```go

diff --git a/handler.go b/handler.go
@@ -34,7 +34,7 @@ func (o Option) NewLokiHandler() slog.Handler {
 	}
 
 	clients := map[slog.Level]promtail.Client{}
-	for k, v := range logLevelConverter {
+	for k, v := range LogLevels {
 		conf := promtail.ClientConfig{
 			PushURL:            o.Endpoint,
 			BatchWait:          o.BatchWait,

diff --git a/loki.go b/loki.go
@@ -6,7 +6,7 @@ import (
 	"github.com/afiskon/promtail-client/promtail"
 )
 
-var logLevelConverter = map[slog.Level]promtail.LogLevel{
+var LogLevels = map[slog.Level]promtail.LogLevel{
 	slog.LevelDebug: promtail.DEBUG,
 	slog.LevelInfo:  promtail.INFO,
 	slog.LevelWarn:  promtail.WARN,