Skip to content

Commit 9725ee2

Browse files
committed
Add workload ID to APM pipeline trace (#9842)
Improve observability of APM traces in workload launcher
1 parent 65ee1bd commit 9725ee2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

airbyte-workload-launcher/src/main/kotlin/pipeline/LaunchPipeline.kt

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class LaunchPipeline(
4242
// scheduler (see the fromExecutorService method on Schedulers).
4343
@Trace(operationName = LAUNCH_PIPELINE_OPERATION_NAME)
4444
fun accept(msg: LauncherInput) {
45-
addTagsToTrace()
45+
addTagsToTrace(msg)
4646
setLoggingScopeForWorkload(msg).use {
4747
metricPublisher.count(WorkloadLauncherMetricMetadata.WORKLOAD_RECEIVED, MetricAttribute(WORKLOAD_ID_TAG, msg.workloadId))
4848
LaunchStageIO(msg)
@@ -68,9 +68,10 @@ class LaunchPipeline(
6868
}
6969
}
7070

71-
private fun addTagsToTrace() {
71+
private fun addTagsToTrace(msg: LauncherInput) {
7272
val commonTags = hashMapOf<String, Any>()
73-
commonTags.put(DATA_PLANE_ID_TAG, dataplaneId)
73+
commonTags[DATA_PLANE_ID_TAG] = dataplaneId
74+
commonTags[WORKLOAD_ID_TAG] = msg.workloadId
7475
ApmTraceUtils.addTagsToTrace(commonTags)
7576
}
7677

0 commit comments

Comments
 (0)