feat: add simplejson.WithLogger() to pass a specific slog Logger

option.go

@@ -2,6 +2,7 @@ package simplejson
 
 import (
 	"github.com/clambin/go-common/httpserver/middleware"
+	"golang.org/x/exp/slog"
 )
 
 // Option specified configuration options for Server
@@ -29,3 +30,12 @@ type WithHTTPMetrics struct {
 func (o WithHTTPMetrics) apply(s *Server) {
 	s.prometheusMetrics = middleware.NewPrometheusMetrics(o.Option)
 }
+
+// WithLogger configures the router to use the provided slog Logger for logging. If no logger is provided, slog.Default() is used instead.
+type WithLogger struct {
+	Logger *slog.Logger
+}
+
+func (o WithLogger) apply(s *Server) {
+	s.logger = o.Logger
+}

server.go

@@ -16,6 +16,7 @@ type Server struct {
 	Handlers          map[string]Handler
 	prometheusMetrics *middleware.PrometheusMetrics
 	queryMetrics      *QueryMetrics
+	logger            *slog.Logger
 }
 
 var _ prometheus.Collector = &Server{}
@@ -25,14 +26,15 @@ func New(handlers map[string]Handler, options ...Option) *Server {
 	s := Server{
 		Handlers: handlers,
 		Router:   chi.NewRouter(),
+		logger:   slog.Default(),
 	}
 	for _, o := range options {
 		o.apply(&s)
 	}
 
 	s.Router.Use(middleware2.Heartbeat("/"))
 	s.Router.Group(func(r chi.Router) {
-		r.Use(middleware.Logger(slog.Default()))
+		r.Use(middleware.Logger(s.logger))
 		if s.prometheusMetrics != nil {
 			r.Use(s.prometheusMetrics.Handle)
 		}

server_test.go

@@ -11,16 +11,21 @@ import (
 	"github.com/prometheus/client_golang/prometheus/testutil"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	"golang.org/x/exp/slog"
 	"net/http"
 	"net/http/httptest"
+	"os"
 	"testing"
 	"time"
 )
 
 var update = flag.Bool("update", false, "update .golden files")
 
 func TestNewRouter(t *testing.T) {
-	r := simplejson.New(nil)
+	logger := slog.New(slog.NewJSONHandler(os.Stdout).WithAttrs([]slog.Attr{
+		slog.String("module", "simplejson"),
+	}))
+	r := simplejson.New(nil, simplejson.WithLogger{Logger: logger})
 
 	for _, path := range []string{"/"} {
 		w := httptest.NewRecorder()