diff --git a/src/main/java/org/springframework/data/envers/repository/support/DefaultRevisionMetadata.java b/src/main/java/org/springframework/data/envers/repository/support/DefaultRevisionMetadata.java index bc9411e..03bfb07 100755 --- a/src/main/java/org/springframework/data/envers/repository/support/DefaultRevisionMetadata.java +++ b/src/main/java/org/springframework/data/envers/repository/support/DefaultRevisionMetadata.java @@ -15,8 +15,6 @@ */ package org.springframework.data.envers.repository.support; -import lombok.NonNull; - import java.time.Instant; import java.time.LocalDateTime; import java.time.ZoneOffset; @@ -25,6 +23,7 @@ import java.util.Optional; import org.hibernate.envers.DefaultRevisionEntity; import org.springframework.data.history.RevisionMetadata; +import org.springframework.util.Assert; /** * {@link RevisionMetadata} working with a {@link DefaultRevisionEntity}. The entity/delegate itself gets ignored for @@ -36,14 +35,17 @@ import org.springframework.data.history.RevisionMetadata; */ public final class DefaultRevisionMetadata implements RevisionMetadata { - private final @NonNull DefaultRevisionEntity entity; + private final DefaultRevisionEntity entity; private final RevisionType revisionType; public DefaultRevisionMetadata(DefaultRevisionEntity entity) { this(entity, RevisionType.UNKNOWN); } - public DefaultRevisionMetadata(@NonNull DefaultRevisionEntity entity, RevisionType revisionType) { + public DefaultRevisionMetadata(DefaultRevisionEntity entity, RevisionType revisionType) { + + Assert.notNull(entity, "DefaultRevisionEntity must not be null"); + this.entity = entity; this.revisionType = revisionType; } diff --git a/src/main/java/org/springframework/data/envers/repository/support/ReflectionRevisionEntityInformation.java b/src/main/java/org/springframework/data/envers/repository/support/ReflectionRevisionEntityInformation.java index 687f99e..144ab1c 100644 --- a/src/main/java/org/springframework/data/envers/repository/support/ReflectionRevisionEntityInformation.java +++ b/src/main/java/org/springframework/data/envers/repository/support/ReflectionRevisionEntityInformation.java @@ -15,9 +15,8 @@ */ package org.springframework.data.envers.repository.support; -import lombok.Getter; - import org.hibernate.envers.RevisionNumber; + import org.springframework.data.repository.history.support.RevisionEntityInformation; import org.springframework.data.util.AnnotationDetectionFieldCallback; import org.springframework.util.Assert; @@ -26,10 +25,9 @@ import org.springframework.util.ReflectionUtils; /** * {@link RevisionEntityInformation} that uses reflection to inspect a property annotated with {@link RevisionNumber} to * find out about the revision number type. - * + * * @author Oliver Gierke */ -@Getter public class ReflectionRevisionEntityInformation implements RevisionEntityInformation { private final Class revisionEntityClass; @@ -37,7 +35,7 @@ public class ReflectionRevisionEntityInformation implements RevisionEntityInform /** * Creates a new {@link ReflectionRevisionEntityInformation} inspecting the given revision entity class. - * + * * @param revisionEntityClass must not be {@literal null}. */ public ReflectionRevisionEntityInformation(Class revisionEntityClass) { @@ -59,4 +57,12 @@ public class ReflectionRevisionEntityInformation implements RevisionEntityInform public boolean isDefaultRevisionEntity() { return false; } + + public Class getRevisionEntityClass() { + return this.revisionEntityClass; + } + + public Class getRevisionNumberType() { + return this.revisionNumberType; + } }