From c9b577a37270f378d727d41ab1bdb910d5ca4092 Mon Sep 17 00:00:00 2001 From: Nikhil P Bonte Date: Fri, 21 Jun 2024 15:51:47 +0530 Subject: [PATCH] DG-1533 Fix Restore domain/product issue --- .../v2/preprocessor/datamesh/AbstractDomainPreProcessor.java | 2 ++ .../graph/v2/preprocessor/datamesh/DataProductPreProcessor.java | 1 - 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/AbstractDomainPreProcessor.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/AbstractDomainPreProcessor.java index 368cfae4fe..9f930ab831 100644 --- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/AbstractDomainPreProcessor.java +++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/AbstractDomainPreProcessor.java @@ -60,6 +60,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; @@ -78,6 +79,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/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataProductPreProcessor.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataProductPreProcessor.java index 9009f3fb76..411fd45e50 100644 --- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataProductPreProcessor.java +++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataProductPreProcessor.java @@ -5,7 +5,6 @@ import org.apache.atlas.RequestContext; 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;