skip deleting Titles with active stakeholders

atlanhq · Jul 22, 2024 · fea7d5d · fea7d5d
1 parent 3827bed
commit fea7d5d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/.../apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataDomainPreProcessor.java b/.../apache/atlas/repository/store/graph/v2/preprocessor/datamesh/DataDomainPreProcessor.java
@@ -460,9 +460,13 @@ public void processDelete(AtlasVertex vertex) throws AtlasBaseException {
                     AtlasVertex stakeholderTitleVertex = entityRetriever.getEntityVertex(stakeholderTitle.getGuid());
                     AtlasGraphUtilsV2.removeItemFromListPropertyValue(stakeholderTitleVertex, ATTR_DOMAIN_QUALIFIED_NAMES, vertex.getProperty(QUALIFIED_NAME, String.class));
                     List<String> domainQualifiedNames = stakeholderTitleVertex.getMultiValuedProperty(ATTR_DOMAIN_QUALIFIED_NAMES, String.class);
+
                     if (CollectionUtils.isEmpty(domainQualifiedNames)) {
-                        entityStore.deleteById(stakeholderTitle.getGuid());
-                        LOG.info("Deleted Stakeholder Title: {}", stakeholderTitle.getGuid());
+                        Iterator<AtlasVertex> stakeholders = getActiveChildrenVertices(stakeholderTitleVertex, STAKEHOLDER_TITLE_EDGE_LABEL);
+                        if (!stakeholders.hasNext()) {
+                            entityStore.deleteById(stakeholderTitle.getGuid());
+                            LOG.info("Deleted Stakeholder Title: {}", stakeholderTitle.getGuid());
+                        }
                     }
                 }
             }