Merge branch 'task/CJ-157' into dev/task/CJ-157

atlanhq · Sep 27, 2024 · a5782c6 · a5782c6
2 parents a1f9099 + 467a647
commit a5782c6
Showing 1 changed file with 18 additions and 7 deletions.
diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java
@@ -1669,19 +1669,30 @@ private void executePreProcessor(EntityMutationContext context) throws AtlasBase
         List<AtlasEntity> copyOfAppendRelationshipAttributes = new ArrayList<>(context.getUpdatedEntitiesForAppendRelationshipAttribute());
         for (AtlasEntity entity : copyOfAppendRelationshipAttributes) {
             entityType = context.getType(entity.getGuid());
-            preProcessors = getPreProcessor(entityType.getTypeName());
-            for(PreProcessor processor : preProcessors){
-                LOG.debug("Executing preprocessor {} for entity {}", processor.getClass().getName(), entity.getGuid());
-                processor.processAttributes(entity, context, UPDATE);
+            if( entityType.getTypeName().equals(ATLAS_DM_ENTITY_TYPE) ||
+                    entityType.getTypeName().equals(ATLAS_DM_ATTRIBUTE_TYPE) ||
+                    entityType.getTypeName().equals(ATLAS_DM_ENTITY_ASSOCIATION_TYPE) ||
+                    entity.getTypeName().equals(ATLAS_DM_ATTRIBUTE_ASSOCIATION_TYPE)
+            ){
+                preProcessors = getPreProcessor(entityType.getTypeName());
+                for (PreProcessor processor : preProcessors) {
+                    processor.processAttributes(entity, context, UPDATE);
+                }
             }
         }
 
         List<AtlasEntity> copyOfRemoveRelationshipAttributes = new ArrayList<>(context.getEntitiesUpdatedWithRemoveRelationshipAttribute());
         for (AtlasEntity entity : copyOfRemoveRelationshipAttributes) {
             entityType = context.getType(entity.getGuid());
-            preProcessors = getPreProcessor(entityType.getTypeName());
-            for (PreProcessor processor : preProcessors) {
-                processor.processAttributes(entity, context, UPDATE);
+            if( entityType.getTypeName().equals(ATLAS_DM_ENTITY_TYPE)
+                    || entityType.getTypeName().equals(ATLAS_DM_ATTRIBUTE_TYPE) ||
+                    entityType.getTypeName().equals(ATLAS_DM_ENTITY_ASSOCIATION_TYPE) ||
+                    entity.getTypeName().equals(ATLAS_DM_ATTRIBUTE_ASSOCIATION_TYPE)
+            ){
+                preProcessors = getPreProcessor(entityType.getTypeName());
+                for (PreProcessor processor : preProcessors) {
+                    processor.processAttributes(entity, context, UPDATE);
+                }
             }
         }
     }