diff --git a/hibernate-envers/src/main/java/org/hibernate/envers/boot/internal/EnversMetadataBuildingContextImpl.java b/hibernate-envers/src/main/java/org/hibernate/envers/boot/internal/EnversMetadataBuildingContextImpl.java index ac25e10e24ed..5f85da14c12f 100644 --- a/hibernate-envers/src/main/java/org/hibernate/envers/boot/internal/EnversMetadataBuildingContextImpl.java +++ b/hibernate-envers/src/main/java/org/hibernate/envers/boot/internal/EnversMetadataBuildingContextImpl.java @@ -12,6 +12,7 @@ import org.hibernate.boot.spi.MetadataBuildingOptions; import org.hibernate.envers.boot.spi.EnversMetadataBuildingContext; import org.hibernate.envers.configuration.Configuration; +import org.hibernate.envers.configuration.internal.ClassesAuditingData; import org.hibernate.envers.configuration.internal.MappingCollector; import org.hibernate.envers.configuration.internal.metadata.AuditEntityConfigurationRegistry; import org.hibernate.envers.configuration.internal.metadata.AuditEntityNameRegister; @@ -33,6 +34,7 @@ public class EnversMetadataBuildingContextImpl implements EnversMetadataBuilding private final ObjectNameNormalizer objectNameNormalizer; private final AuditEntityNameRegister auditEntityNameRegistry; private final AuditEntityConfigurationRegistry auditEntityConfigurationRegistry; + private final ClassesAuditingData classesAuditingData; public EnversMetadataBuildingContextImpl( Configuration configuration, @@ -45,6 +47,7 @@ public EnversMetadataBuildingContextImpl( this.mappingCollector = mappingCollector; this.auditEntityNameRegistry = new AuditEntityNameRegister(); this.auditEntityConfigurationRegistry = new AuditEntityConfigurationRegistry(); + this.classesAuditingData = new ClassesAuditingData(); this.objectNameNormalizer = new ObjectNameNormalizer(this); } @@ -118,4 +121,9 @@ public AuditEntityNameRegister getAuditEntityNameRegistry() { public AuditEntityConfigurationRegistry getAuditEntityConfigurationRegistry() { return auditEntityConfigurationRegistry; } + + @Override + public ClassesAuditingData getClassesAuditingData() { + return classesAuditingData; + } } diff --git a/hibernate-envers/src/main/java/org/hibernate/envers/boot/spi/EnversMetadataBuildingContext.java b/hibernate-envers/src/main/java/org/hibernate/envers/boot/spi/EnversMetadataBuildingContext.java index 77f2b776860a..201f5619785d 100644 --- a/hibernate-envers/src/main/java/org/hibernate/envers/boot/spi/EnversMetadataBuildingContext.java +++ b/hibernate-envers/src/main/java/org/hibernate/envers/boot/spi/EnversMetadataBuildingContext.java @@ -6,6 +6,7 @@ import org.hibernate.boot.spi.MetadataBuildingContext; import org.hibernate.envers.configuration.Configuration; +import org.hibernate.envers.configuration.internal.ClassesAuditingData; import org.hibernate.envers.configuration.internal.MappingCollector; import org.hibernate.envers.configuration.internal.metadata.AuditEntityConfigurationRegistry; import org.hibernate.envers.configuration.internal.metadata.AuditEntityNameRegister; @@ -32,4 +33,6 @@ public interface EnversMetadataBuildingContext extends MetadataBuildingContext { AuditEntityNameRegister getAuditEntityNameRegistry(); AuditEntityConfigurationRegistry getAuditEntityConfigurationRegistry(); + + ClassesAuditingData getClassesAuditingData(); } diff --git a/hibernate-envers/src/main/java/org/hibernate/envers/configuration/internal/ClassesAuditingData.java b/hibernate-envers/src/main/java/org/hibernate/envers/configuration/internal/ClassesAuditingData.java index 0d8a057b2d27..796ea84f9d66 100644 --- a/hibernate-envers/src/main/java/org/hibernate/envers/configuration/internal/ClassesAuditingData.java +++ b/hibernate-envers/src/main/java/org/hibernate/envers/configuration/internal/ClassesAuditingData.java @@ -61,6 +61,10 @@ public Collection getAllClassAuditedData() { return persistentClassToAuditingData.values(); } + public ClassAuditingData getClassAuditingData(String entityName) { + return entityNameToAuditingData.get( entityName ); + } + /** * After all meta-data is read, updates calculated fields. This includes: *