Added task info in audit and kafka headers.

atlanhq · Dec 11, 2024 · 9bf9002 · 9bf9002
1 parent 6e1f8e9
commit 9bf9002
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...ry/src/main/java/org/apache/atlas/repository/store/graph/v2/tasks/ClassificationTask.java b/...ry/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);