Merge pull request #3270 from atlanhq/DG-1533

DG-1533 Fix Restore domain/product issue
atlanhq · Jun 21, 2024 · 910893d · 910893d
2 parents 640ea65 + c9b577a
commit 910893d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/...che/atlas/repository/store/graph/v2/preprocessor/datamesh/AbstractDomainPreProcessor.java b/...che/atlas/repository/store/graph/v2/preprocessor/datamesh/AbstractDomainPreProcessor.java
@@ -61,6 +61,7 @@ public abstract class AbstractDomainPreProcessor implements PreProcessor {
     private static final Logger LOG = LoggerFactory.getLogger(AbstractDomainPreProcessor.class);
 
 
+    protected final AtlasGraph graph;
     protected final AtlasTypeRegistry typeRegistry;
     protected final EntityGraphRetriever entityRetriever;
     protected EntityGraphRetriever entityRetrieverNoRelations;
@@ -79,6 +80,7 @@ public abstract class AbstractDomainPreProcessor implements PreProcessor {
     }
 
     AbstractDomainPreProcessor(AtlasTypeRegistry typeRegistry, EntityGraphRetriever entityRetriever, AtlasGraph graph) {
+        this.graph = graph;
         this.entityRetriever = entityRetriever;
         this.typeRegistry = typeRegistry;
         this.preProcessor = new AuthPolicyPreProcessor(graph, typeRegistry, entityRetriever);

diff --git a/...apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataProductPreProcessor.java b/...apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataProductPreProcessor.java
@@ -8,7 +8,6 @@
 import org.apache.atlas.authorize.AtlasPrivilege;
 import org.apache.atlas.exception.AtlasBaseException;
 import org.apache.atlas.model.instance.*;
-import org.apache.atlas.repository.graph.GraphHelper;
 import org.apache.atlas.repository.graphdb.AtlasEdge;
 import org.apache.atlas.repository.graphdb.AtlasEdgeDirection;
 import org.apache.atlas.repository.graphdb.AtlasGraph;