From 9bf9002b9e541c8a8c7be33ce6a853718fe2a77e Mon Sep 17 00:00:00 2001 From: hr2904 Date: Wed, 11 Dec 2024 13:50:05 +0530 Subject: [PATCH] Added task info in audit and kafka headers. --- .../repository/store/graph/v2/tasks/ClassificationTask.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/tasks/ClassificationTask.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/tasks/ClassificationTask.java index deac12e565..eba842d32e 100644 --- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/tasks/ClassificationTask.java +++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/tasks/ClassificationTask.java @@ -58,7 +58,8 @@ public abstract class ClassificationTask extends AbstractTask { public static final String PARAM_PREVIOUS_CLASSIFICATION_RESTRICT_PROPAGATE_THROUGH_LINEAGE = "previousRestrictPropagationThroughLineage"; public static final String PARAM_PREVIOUS_CLASSIFICATION_RESTRICT_PROPAGATE_THROUGH_HIERARCHY = "previousRestrictPropagationThroughHierarchy"; - + public static final String X_ATLAN_TASK_GUID = "x-atlan-task-guid"; + protected final AtlasGraph graph; protected final EntityGraphMapper entityGraphMapper; protected final DeleteHandlerDelegate deleteDelegate; @@ -97,6 +98,7 @@ public AtlasTask.Status perform() throws AtlasBaseException { } RequestContext.get().setUser(userName, null); + RequestContext.get().addRequestContextHeader(X_ATLAN_TASK_GUID, getTaskGuid()); try { setStatus(IN_PROGRESS);