diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java index a6aa1695f37..765aa7ddc75 100644 --- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java +++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java @@ -18,7 +18,9 @@ package org.apache.atlas.repository.store.graph.v2; import com.fasterxml.jackson.core.type.TypeReference; +import org.apache.atlas.AtlasConfiguration; import org.apache.atlas.AtlasErrorCode; +import org.apache.atlas.RequestContext; import org.apache.atlas.exception.AtlasBaseException; import org.apache.atlas.model.TimeBoundary; import org.apache.atlas.model.glossary.AtlasGlossaryCategory; @@ -162,6 +164,7 @@ public class EntityGraphRetriever { private static final String GLOSSARY_CATEGORY_HIERARCHY_EDGE_LABEL = "r:AtlasGlossaryCategoryHierarchyLink"; private static final String GLOSSARY_CATEGORY_TYPE_NAME = AtlasGlossaryCategory.class.getSimpleName(); private static final String PARENT_GLOSSARY_CATEGORY_GUID = "parentCategoryGuid"; + private boolean deferredActionEnabled = AtlasConfiguration.TASKS_USE_ENABLED.getBoolean(); private static final TypeReference> TIME_BOUNDARIES_LIST_TYPE = new TypeReference>() {}; @@ -698,7 +701,7 @@ private void traverseImpactedVertices(final AtlasVertex entityVertexStart, final Iterable propagationEdges = entityVertex.getEdges(AtlasEdgeDirection.BOTH, tagPropagationEdges); for (AtlasEdge propagationEdge : propagationEdges) { - if (getEdgeStatus(propagationEdge) != ACTIVE) { + if (getEdgeStatus(propagationEdge) != ACTIVE && !deferredActionEnabled && !RequestContext.get().isImportInProgress()) { continue; }