Register ManagedTypes post-processor.

See #1280

Author: mp911de

spring-data-cassandra/src/main/java/org/springframework/data/cassandra/aot/CassandraManagedTypesBeanRegistrationAotProcessor.java

@@ -0,0 +1,38 @@
+/*
+ * Copyright 2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.cassandra.aot;
+
+import org.springframework.data.aot.ManagedTypesBeanRegistrationAotProcessor;
+import org.springframework.data.cassandra.CassandraManagedTypes;
+import org.springframework.lang.Nullable;
+import org.springframework.util.ClassUtils;
+
+/**
+ * Cassandra-specific extension to {@link ManagedTypesBeanRegistrationAotProcessor}.
+ *
+ * @author Mark Paluch
+ * @since 4.0
+ */
+class CassandraManagedTypesBeanRegistrationAotProcessor extends ManagedTypesBeanRegistrationAotProcessor {
+
+	public CassandraManagedTypesBeanRegistrationAotProcessor() {
+		setModuleIdentifier("cassandra");
+	}
+
+	protected boolean matchesByType(@Nullable Class<?> beanType) {
+		return beanType != null && ClassUtils.isAssignable(CassandraManagedTypes.class, beanType);
+	}
+}

spring-data-cassandra/src/main/resources/META-INF/spring/aot.factories

@@ -2,4 +2,4 @@ org.springframework.aot.hint.RuntimeHintsRegistrar=\
 	org.springframework.data.cassandra.aot.CassandraRuntimeHintsRegistrar
 
 org.springframework.beans.factory.aot.BeanRegistrationAotProcessor=\
-	org.springframework.data.aot.ManagedTypesBeanRegistrationAotProcessor
+	org.springframework.data.cassandra.aot.CassandraManagedTypesBeanRegistrationAotProcessor