PR Fix

atlanhq · Jan 22, 2025 · ec6dab5 · ec6dab5
1 parent 9bf9002
commit ec6dab5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
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,6 @@ 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;
@@ -98,7 +97,6 @@ public AtlasTask.Status perform() throws AtlasBaseException {
         }
 
         RequestContext.get().setUser(userName, null);
-        RequestContext.get().addRequestContextHeader(X_ATLAN_TASK_GUID, getTaskGuid());
 
         try {
             setStatus(IN_PROGRESS);

diff --git a/repository/src/main/java/org/apache/atlas/tasks/AbstractTask.java b/repository/src/main/java/org/apache/atlas/tasks/AbstractTask.java
@@ -16,6 +16,7 @@
  * limitations under the License.
  */
 package org.apache.atlas.tasks;
+import org.apache.atlas.RequestContext;
 
 import org.apache.atlas.model.tasks.AtlasTask;
 import org.slf4j.Logger;
@@ -27,12 +28,15 @@ public abstract class AbstractTask {
     private static final Logger LOG = LoggerFactory.getLogger(AbstractTask.class);
     private final AtlasTask task;
 
+    public static final String X_ATLAN_TASK_GUID = "x-atlan-task-guid";
+
     public AbstractTask(AtlasTask task) {
         this.task = task;
     }
 
     public void run() throws Exception {
         try {
+            RequestContext.get().addRequestContextHeader(X_ATLAN_TASK_GUID, getTaskGuid());
             perform();
         } catch (Exception exception) {
             task.setStatus(Status.FAILED);