Add util.EventHookWithContext.

util/service.go

@@ -1,34 +1,39 @@
 package util
 
 import (
+	"context"
 	"log/slog"
 
 	"github.com/thejerf/suture/v4"
 )
 
-func EventHook(e suture.Event) {
-	var args []any
-	for k, v := range e.Map() {
-		args = append(args, k, v)
-	}
+var EventHook = EventHookWithContext(context.Background())
+
+func EventHookWithContext(ctx context.Context) suture.EventHook {
+	return func(e suture.Event) {
+		var args []any
+		for k, v := range e.Map() {
+			args = append(args, k, v)
+		}
 
-	switch e.Type() {
-	case suture.EventTypeBackoff:
-		slog.Info(e.String(), args...)
+		switch e.Type() {
+		case suture.EventTypeBackoff:
+			slog.InfoContext(ctx, e.String(), args...)
 
-	case suture.EventTypeResume:
-		slog.Info(e.String(), args...)
+		case suture.EventTypeResume:
+			slog.InfoContext(ctx, e.String(), args...)
 
-	case suture.EventTypeServicePanic:
-		slog.Warn(e.String(), args...)
+		case suture.EventTypeServicePanic:
+			slog.WarnContext(ctx, e.String(), args...)
 
-	case suture.EventTypeServiceTerminate:
-		slog.Warn(e.String(), args...)
+		case suture.EventTypeServiceTerminate:
+			slog.WarnContext(ctx, e.String(), args...)
 
-	case suture.EventTypeStopTimeout:
-		slog.Error(e.String(), args...)
+		case suture.EventTypeStopTimeout:
+			slog.ErrorContext(ctx, e.String(), args...)
 
-	default:
-		panic(e) // unhandled event
+		default:
+			panic(e) // unhandled event
+		}
 	}
 }