feat(log): WithPrefixMessage func

reearth · Jan 30, 2025 · 7860571 · 7860571
1 parent 471511f
commit 7860571
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/log/context.go b/log/context.go
@@ -29,6 +29,12 @@ func UpdateContext(ctx context.Context, f func(logger *Logger) *Logger) context.
 	return AttachLoggerToContext(ctx, f(GetLoggerFromContextOrDefault(ctx)))
 }
 
+func WithPrefixMessage(ctx context.Context, prefix string) context.Context {
+	return UpdateContext(ctx, func(logger *Logger) *Logger {
+		return logger.AppendPrefixMessage(prefix)
+	})
+}
+
 func Tracefc(ctx context.Context, format string, args ...any) {
 	getLoggerFromContextOrDefault(ctx).Debugf(format, args...)
 }