Skip to content

Commit 9d0a784

Browse files
authored
try and add temporal metadata to slog.Logger (#3385)
1 parent 4cb6e7c commit 9d0a784

File tree

1 file changed

+37
-0
lines changed

1 file changed

+37
-0
lines changed

flow/internal/context.go

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,5 +130,42 @@ func SlogLoggerFromCtx(ctx context.Context) *slog.Logger {
130130
}
131131
logger = logger.With(slog.Any(string(AdditionalMetadataKey), GetAdditionalMetadata(ctx)))
132132

133+
if activity.IsActivity(ctx) {
134+
activityInfo := activity.GetInfo(ctx)
135+
var workflowTypeName string
136+
if activityInfo.WorkflowType != nil {
137+
workflowTypeName = activityInfo.WorkflowType.Name
138+
}
139+
logger = logger.With(
140+
slog.String("ActivityID", activityInfo.ActivityID),
141+
slog.String("ActivityType", activityInfo.ActivityType.Name),
142+
slog.Int64("Attempt", int64(activityInfo.Attempt)),
143+
slog.String("RunID", activityInfo.WorkflowExecution.RunID),
144+
slog.String("WorkflowID", activityInfo.WorkflowExecution.ID),
145+
slog.String("Namespace", activityInfo.WorkflowNamespace),
146+
slog.String("TaskQueue", activityInfo.TaskQueue),
147+
slog.String("WorkflowType", workflowTypeName),
148+
)
149+
}
150+
151+
if activity.IsActivity(ctx) {
152+
activityInfo := activity.GetInfo(ctx)
153+
154+
workflowTypeName := ""
155+
if activityInfo.WorkflowType != nil {
156+
workflowTypeName = activityInfo.WorkflowType.Name
157+
}
158+
logger = logger.With(
159+
slog.String("ActivityID", activityInfo.ActivityID),
160+
slog.String("ActivityType", activityInfo.ActivityType.Name),
161+
slog.Int64("Attempt", int64(activityInfo.Attempt)),
162+
slog.String("RunID", activityInfo.WorkflowExecution.RunID),
163+
slog.String("WorkflowID", activityInfo.WorkflowExecution.ID),
164+
slog.String("Namespace", activityInfo.WorkflowNamespace),
165+
slog.String("TaskQueue", activityInfo.TaskQueue),
166+
slog.String("WorkflowType", workflowTypeName),
167+
)
168+
}
169+
133170
return logger
134171
}

0 commit comments

Comments
 (0)