ATLAS-5056: Entity audit should not be created for instances of internal entity-types

Author: aditya-gupta36

repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityChangeNotifier.java

@@ -69,8 +69,6 @@
 public class AtlasEntityChangeNotifier implements IAtlasEntityChangeNotifier {
     private static final Logger LOG = LoggerFactory.getLogger(AtlasEntityChangeNotifier.class);
 
-    private static final Predicate<AtlasEntityHeader> PRED_IS_NOT_TYPE_AUDIT_ENTITY = obj -> !obj.getTypeName().equals(AtlasAuditService.ENTITY_TYPE_AUDIT_ENTRY);
-
     private final Set<EntityChangeListener>   entityChangeListeners;
     private final Set<EntityChangeListenerV2> entityChangeListenersV2;
     private final AtlasInstanceConverter      instanceConverter;
@@ -401,7 +399,26 @@ private String getListenerName(EntityChangeListener listener) {
     }
 
     private boolean skipAuditEntries(List<AtlasEntityHeader> entityHeaders) {
-        return CollectionUtils.isEmpty(entityHeaders) || entityHeaders.stream().noneMatch(PRED_IS_NOT_TYPE_AUDIT_ENTITY);
+        if (CollectionUtils.isEmpty(entityHeaders)) {
+            return true;
+        }
+
+        // Skip audit if ALL entities are internal types (__AtlasAuditEntry, __AtlasMetricsStat, etc.)
+        // Return true (skip) only if ALL entities are internal types
+        // Return false (don't skip) if at least one entity is NOT an internal type
+        boolean allInternal = entityHeaders.stream().allMatch(entityHeader -> {
+            String typeName = entityHeader.getTypeName();
+            AtlasEntityType entityType = atlasTypeRegistry.getEntityTypeByName(typeName);
+
+            if (entityType != null) {
+                return entityType.isInternalType();  //return true if its -> internal type   else return false
+            }
+
+            // Fallback: if type not found in registry, check if it starts with "__" (like GraphHelper.isInternalType)
+            return GraphHelper.isInternalType(typeName);
+        });
+
+        return allInternal;
     }
 
     private void notifyListeners(List<AtlasEntityHeader> entityHeaders, EntityOperation operation, boolean isImport) throws AtlasBaseException {