Added ctxkeys package to better handle addition context when logging and

partially integrated it with keys for user id and ip addr

1 files changed, 5 insertions, 1 deletions

diff --git a/cmd/server/main.go b/cmd/server/main.go
index 6b6d04b..58998ca 100644
--- a/cmd/server/main.go
+++ b/cmd/server/main.go
@@ -12,6 +12,7 @@ import (
 
 	"github.com/kyren223/eko/internal/server"
 	"github.com/kyren223/eko/internal/server/api"
+	"github.com/kyren223/eko/internal/server/ctxkeys"
 	"github.com/kyren223/eko/pkg/assert"
 	"gopkg.in/natefinch/lumberjack.v2"
 )
@@ -27,6 +28,8 @@ func main() {
 
 	setupLogging()
 
+	log.Println("Debugging prod", "prod", prod)
+
 	api.ConnectToDatabase()
 	assert.AddFlush(api.DB())
 	defer api.DB().Close()
@@ -68,10 +71,11 @@ func setupLogging() {
 	if prod {
 		level = slog.LevelInfo
 	}
-	handler := slog.NewJSONHandler(rotator, &slog.HandlerOptions{
+	baseHandler := slog.NewJSONHandler(rotator, &slog.HandlerOptions{
 		AddSource: true,
 		Level:     level,
 	})
+	handler := ctxkeys.WrapLogHandler(baseHandler)
 
 	logger := slog.New(handler)
 	slog.SetDefault(logger)