diff --git a/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/aot/CassandraManagedTypesBeanRegistrationAotProcessor.java b/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/aot/CassandraManagedTypesBeanRegistrationAotProcessor.java new file mode 100644 index 000000000..631f533bb --- /dev/null +++ b/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); + } +} diff --git a/spring-data-cassandra/src/main/resources/META-INF/spring/aot.factories b/spring-data-cassandra/src/main/resources/META-INF/spring/aot.factories index 894bcbafa..c30649f39 100644 --- a/spring-data-cassandra/src/main/resources/META-INF/spring/aot.factories +++ b/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