Add @Contract and @CheckReturnValue annotations.

Closes: #3195
Original pull request: #3210

Author: mp911de

src/main/java/org/springframework/data/annotation/AccessType.java

@@ -34,8 +34,6 @@
 
 	/**
 	 * The access type to be used.
-	 *
-	 * @return
 	 */
 	Type value();
 

src/main/java/org/springframework/data/annotation/PersistenceCreator.java

@@ -29,4 +29,5 @@
  */
 @Retention(RetentionPolicy.RUNTIME)
 @Target({ ElementType.CONSTRUCTOR, ElementType.METHOD, ElementType.ANNOTATION_TYPE })
-public @interface PersistenceCreator {}
+public @interface PersistenceCreator {
+}

src/main/java/org/springframework/data/annotation/Reference.java

@@ -36,17 +36,13 @@
 	/**
 	 * Explicitly define the target type of the reference. Used in case the annotated property is not the target type but
 	 * rather an identifier and/or if that identifier type is not uniquely identifying the target entity.
-	 *
-	 * @return
 	 */
 	@AliasFor(attribute = "to")
 	Class<?> value() default Class.class;
 
 	/**
 	 * Explicitly define the target type of the reference. Used in case the annotated property is not the target type but
 	 * rather an identifier and/or if that identifier type is not uniquely identifying the target entity.
-	 *
-	 * @return
 	 */
 	@AliasFor(attribute = "value")
 	Class<?> to() default Class.class;

src/main/java/org/springframework/data/annotation/TypeAlias.java

@@ -29,6 +29,7 @@
  * {@link PersistentEntity}s.
  *
  * @author Oliver Gierke
+ * @see org.springframework.data.mapping.Alias
  */
 @Documented
 @Inherited
@@ -38,9 +39,7 @@
 public @interface TypeAlias {
 
 	/**
-	 * The type alias to be used when persisting
-	 *
-	 * @return
+	 * The type alias to be used when persisting.
 	 */
 	String value();
 }

src/main/java/org/springframework/data/annotation/Version.java

@@ -33,5 +33,4 @@
 @Retention(RUNTIME)
 @Target(value = { FIELD, METHOD, ANNOTATION_TYPE })
 public @interface Version {
-
 }

src/main/java/org/springframework/data/aot/AotContext.java

@@ -140,7 +140,7 @@ default Set<Class<?>> scanPackageForTypes(Collection<Class<? extends Annotation>
 	}
 
 	/**
-	 * Returns a {@link IntrospectedBeanDefinition} to obtain further detail about the underlying bean definition. A
+	 * Returns a {@link IntrospectedBeanDefinition} to obtain further detail about the underlying bean definition. An
 	 * introspected bean definition can also point to an absent bean definition.
 	 *
 	 * @param reference {@link BeanReference} to the managed bean.
@@ -151,7 +151,7 @@ default IntrospectedBeanDefinition introspectBeanDefinition(BeanReference refere
 	}
 
 	/**
-	 * Returns a {@link IntrospectedBeanDefinition} to obtain further detail about the underlying bean definition. A
+	 * Returns a {@link IntrospectedBeanDefinition} to obtain further detail about the underlying bean definition. An
 	 * introspected bean definition can also point to an absent bean definition.
 	 *
 	 * @param beanName {@link String} containing the {@literal name} of the bean to evaluate; must not be {@literal null}.
@@ -251,7 +251,6 @@ interface IntrospectedBeanDefinition {
 		 *           bean}.
 		 * @see BeanDefinition
 		 */
-
 		BeanDefinition getBeanDefinition() throws NoSuchBeanDefinitionException;
 
 		/**

src/main/java/org/springframework/data/aot/AuditingBeanRegistrationAotProcessor.java

@@ -39,7 +39,6 @@
  */
 class AuditingBeanRegistrationAotProcessor implements BeanRegistrationAotProcessor {
 
-
 	@Nullable
 	@Override
 	public BeanRegistrationAotContribution processAheadOfTime(RegisteredBean registeredBean) {

src/main/java/org/springframework/data/aot/ManagedTypesBeanFactoryInitializationAotProcessor.java

@@ -29,6 +29,7 @@
 import org.springframework.beans.factory.support.BeanDefinitionBuilder;
 import org.springframework.beans.factory.support.BeanDefinitionRegistry;
 import org.springframework.data.domain.ManagedTypes;
+import org.springframework.lang.Contract;
 import org.springframework.lang.Nullable;
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.ObjectUtils;
@@ -44,10 +45,11 @@
  */
 public class ManagedTypesBeanFactoryInitializationAotProcessor implements BeanFactoryInitializationAotProcessor {
 
-	private static final Log logger = LogFactory.getLog(BeanFactoryInitializationAotProcessor.class);
+	private static final Log logger = LogFactory.getLog(ManagedTypesBeanFactoryInitializationAotProcessor.class);
 
-	@Nullable
 	@Override
+	@Nullable
+	@Contract("_ -> null")
 	public BeanFactoryInitializationAotContribution processAheadOfTime(ConfigurableListableBeanFactory beanFactory) {
 
 		processManagedTypes(beanFactory);
@@ -71,9 +73,9 @@ private void postProcessManagedTypes(ConfigurableListableBeanFactory beanFactory
 
 		if (hasConstructorArguments(beanDefinition)) {
 
-			ValueHolder argumentValue = beanDefinition.getConstructorArgumentValues().getArgumentValue(0, null, null, null);
+			ValueHolder holder = beanDefinition.getConstructorArgumentValues().getArgumentValue(0, null, null, null);
 
-			if (argumentValue.getValue()instanceof Supplier supplier) {
+			if (holder != null && holder.getValue() instanceof Supplier<?> supplier) {
 
 				if (logger.isDebugEnabled()) {
 					logger.info(String.format("Replacing ManagedType bean definition %s.", beanName));

src/main/java/org/springframework/data/aot/ManagedTypesBeanRegistrationAotProcessor.java

@@ -78,9 +78,9 @@ private ManagedTypes resolveManagedTypes(RegisteredBean registeredBean) {
 		if (beanDefinition.hasConstructorArgumentValues()) {
 
 			ValueHolder indexedArgumentValue = beanDefinition.getConstructorArgumentValues().getIndexedArgumentValue(0, null);
-			Object value = indexedArgumentValue.getValue();
 
-			if (value instanceof Collection<?> values && values.stream().allMatch(it -> it instanceof Class)) {
+			if (indexedArgumentValue != null && indexedArgumentValue.getValue() instanceof Collection<?> values
+					&& values.stream().allMatch(it -> it instanceof Class)) {
 				return ManagedTypes.fromIterable((Collection<Class<?>>) values);
 			}
 		}
@@ -115,7 +115,6 @@ private ManagedTypes resolveManagedTypes(RegisteredBean registeredBean) {
 	 * @param managedTypes never {@literal null}.
 	 * @return new instance of {@link BeanRegistrationAotContribution} or {@literal null} if nothing to do.
 	 */
-	@Nullable
 	protected BeanRegistrationAotContribution contribute(AotContext aotContext, ManagedTypes managedTypes,
 			RegisteredBean registeredBean) {
 		return new ManagedTypesRegistrationAotContribution(managedTypes, registeredBean, this::contributeType);

src/main/java/org/springframework/data/aot/ManagedTypesRegistrationAotContribution.java

@@ -68,8 +68,8 @@
  * @author John Blum
  * @author Christoph Strobl
  * @author Mark Paluch
- * @see org.springframework.beans.factory.aot.BeanRegistrationAotContribution
  * @since 3.0
+ * @see org.springframework.beans.factory.aot.BeanRegistrationAotContribution
  */
 class ManagedTypesRegistrationAotContribution implements RegisteredBeanAotContribution {
 
@@ -125,7 +125,7 @@ static class ManagedTypesInstanceCodeFragment extends BeanRegistrationCodeFragme
 		public static final ResolvableType MANAGED_TYPES_TYPE = ResolvableType.forType(ManagedTypes.class);
 		private final List<Class<?>> sourceTypes;
 		private final RegisteredBean source;
-		private final Lazy<Method> instanceMethod = Lazy.of(this::findInstanceFactory);
+		private final Lazy<Method> instanceMethod;
 
 		private static final TypeName WILDCARD = WildcardTypeName.subtypeOf(Object.class);
 		private static final TypeName CLASS_OF_ANY = ParameterizedTypeName.get(ClassName.get(Class.class), WILDCARD);
@@ -138,6 +138,7 @@ protected ManagedTypesInstanceCodeFragment(List<Class<?>> sourceTypes, Registere
 
 			this.sourceTypes = sourceTypes;
 			this.source = source;
+			this.instanceMethod = Lazy.of(() -> findInstanceFactory(source.getBeanClass()));
 		}
 
 		@Override
@@ -229,15 +230,15 @@ private CodeBlock toCodeBlock(List<Class<?>> values, boolean allPublic) {
 		}
 
 		@Nullable
-		private Method findInstanceFactory() {
+		private static Method findInstanceFactory(Class<?> beanClass) {
 
-			for (Method beanMethod : ReflectionUtils.getDeclaredMethods(source.getBeanClass())) {
+			for (Method beanMethod : ReflectionUtils.getDeclaredMethods(beanClass)) {
 
 				if (!isInstanceFactory(beanMethod)) {
 					continue;
 				}
 
-				ResolvableType parameterType = ResolvableType.forMethodParameter(beanMethod, 0, source.getBeanClass());
+				ResolvableType parameterType = ResolvableType.forMethodParameter(beanMethod, 0, beanClass);
 
 				if (parameterType.isAssignableFrom(LIST_TYPE) || parameterType.isAssignableFrom(MANAGED_TYPES_TYPE)) {
 					return beanMethod;