little bit refactoring

spring-projects · Oct 12, 2024 · 2893de4 · 2893de4
1 parent ddc9914
commit 2893de4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...-jpa/src/main/java/org/springframework/data/jpa/repository/support/DefaultJpaContext.java b/...-jpa/src/main/java/org/springframework/data/jpa/repository/support/DefaultJpaContext.java
@@ -21,6 +21,7 @@
 import jakarta.persistence.EntityManager;
 import jakarta.persistence.metamodel.ManagedType;
 
+import jakarta.validation.constraints.NotNull;
 import org.springframework.data.jpa.repository.JpaContext;
 import org.springframework.util.Assert;
 import org.springframework.util.LinkedMultiValueMap;
@@ -42,7 +43,7 @@ public class DefaultJpaContext implements JpaContext {
 	 *
 	 * @param entityManagers must not be {@literal null}.
 	 */
-	public DefaultJpaContext(Set<EntityManager> entityManagers) {
+	public DefaultJpaContext(@NotNull Set<EntityManager> entityManagers) {
 
 		Assert.notNull(entityManagers, "EntityManagers must not be null");
 		Assert.notEmpty(entityManagers, "EntityManagers must not be empty");
@@ -57,7 +58,7 @@ public DefaultJpaContext(Set<EntityManager> entityManagers) {
 	}
 
 	@Override
-	public EntityManager getEntityManagerByManagedType(Class<?> type) {
+	public EntityManager getEntityManagerByManagedType(@NotNull Class<?> type) {
 
 		Assert.notNull(type, "Type must not be null");