Fix AerospikeRepositoryConfigurationExtension and ReactiveAerospikeRe…

…positoryConfigurationExtension for multi-storage usage

src/main/java/org/springframework/data/aerospike/repository/AerospikeRepository.java

@@ -17,7 +17,6 @@
 
 import com.aerospike.client.query.IndexType;
 import org.springframework.data.aerospike.IndexAlreadyExistsException;
-import org.springframework.data.repository.NoRepositoryBean;
 import org.springframework.data.repository.PagingAndSortingRepository;
 import org.springframework.data.repository.Repository;
 
@@ -28,7 +27,6 @@
  * @author Peter Milne
  * @author Jean Mercier
  */
-@NoRepositoryBean
 public interface AerospikeRepository<T, ID> extends PagingAndSortingRepository<T, ID> {
 
 	<T> void createIndex(Class<T> domainType, String indexName, String binName, IndexType indexType);

src/main/java/org/springframework/data/aerospike/repository/ReactiveAerospikeRepository.java

@@ -1,6 +1,5 @@
 package org.springframework.data.aerospike.repository;
 
-import org.springframework.data.repository.NoRepositoryBean;
 import org.springframework.data.repository.Repository;
 import org.springframework.data.repository.reactive.ReactiveCrudRepository;
 
@@ -9,7 +8,6 @@
  *
  * @author Igor Ermolenko
  */
-@NoRepositoryBean
 public interface ReactiveAerospikeRepository<T, ID> extends ReactiveCrudRepository<T, ID> {
 
 }

src/main/java/org/springframework/data/aerospike/repository/config/AerospikeRepositoriesRegistrar.java

@@ -27,19 +27,11 @@
  */
 public class AerospikeRepositoriesRegistrar extends RepositoryBeanDefinitionRegistrarSupport {
 
-	/*
-	 * (non-Javadoc)
-	 * @see org.springframework.data.repository.config.RepositoryBeanDefinitionRegistrarSupport#getAnnotation()
-	 */
 	@Override
 	protected Class<? extends Annotation> getAnnotation() {
 		return EnableAerospikeRepositories.class;
 	}
 
-	/*
-	 * (non-Javadoc)
-	 * @see org.springframework.data.repository.config.RepositoryBeanDefinitionRegistrarSupport#getExtension()
-	 */
 	@Override
 	protected RepositoryConfigurationExtension getExtension() {
 		return new AerospikeRepositoryConfigurationExtension();

src/main/java/org/springframework/data/aerospike/repository/config/AerospikeRepositoryConfigurationExtension.java

@@ -1,42 +1,52 @@
 package org.springframework.data.aerospike.repository.config;
 
+import org.springframework.data.aerospike.mapping.Document;
+import org.springframework.data.aerospike.repository.AerospikeRepository;
+import org.springframework.data.aerospike.repository.support.AerospikeRepositoryFactoryBean;
 import org.springframework.data.repository.config.RepositoryConfigurationExtension;
 import org.springframework.data.repository.core.RepositoryMetadata;
 
+import java.lang.annotation.Annotation;
+import java.util.Collection;
+import java.util.Collections;
+
 /**
  * {@link RepositoryConfigurationExtension} for Aerospike.
  *
  * @author Oliver Gierke
  */
 public class AerospikeRepositoryConfigurationExtension extends BaseAerospikeRepositoryConfigurationExtension {
 
-    /*
-     * (non-Javadoc)
-     * @see org.springframework.data.keyvalue.repository.config.KeyValueRepositoryConfigurationExtension#getModuleName()
-     */
     @Override
     public String getModuleName() {
         return "Aerospike";
     }
 
-    /*
-     * (non-Javadoc)
-     * @see org.springframework.data.keyvalue.repository.config.KeyValueRepositoryConfigurationExtension#getModulePrefix()
-     */
     @Override
     protected String getModulePrefix() {
         return "aerospike";
     }
 
-    /*
-     * (non-Javadoc)
-     * @see org.springframework.data.keyvalue.repository.config.KeyValueRepositoryConfigurationExtension#getDefaultKeyValueTemplateRef()
-     */
     @Override
     protected String getDefaultKeyValueTemplateRef() {
         return "aerospikeTemplate";
     }
 
+    @Override
+    public String getRepositoryFactoryBeanClassName() {
+        return AerospikeRepositoryFactoryBean.class.getName();
+    }
+
+    @Override
+    protected Collection<Class<? extends Annotation>> getIdentifyingAnnotations() {
+        return Collections.singleton(Document.class);
+    }
+
+    @Override
+    protected Collection<Class<?>> getIdentifyingTypes() {
+        return Collections.singleton(AerospikeRepository.class);
+    }
+
     @Override
     protected boolean useRepositoryConfiguration(RepositoryMetadata metadata) {
         return !metadata.isReactiveRepository();

src/main/java/org/springframework/data/aerospike/repository/config/ReactiveAerospikeRepositoriesRegistrar.java

@@ -12,19 +12,11 @@
  */
 public class ReactiveAerospikeRepositoriesRegistrar extends RepositoryBeanDefinitionRegistrarSupport {
 
-    /*
-     * (non-Javadoc)
-     * @see org.springframework.data.repository.config.RepositoryBeanDefinitionRegistrarSupport#getAnnotation()
-     */
     @Override
     protected Class<? extends Annotation> getAnnotation() {
         return EnableReactiveAerospikeRepositories.class;
     }
 
-    /*
-     * (non-Javadoc)
-     * @see org.springframework.data.repository.config.RepositoryBeanDefinitionRegistrarSupport#getExtension()
-     */
     @Override
     protected RepositoryConfigurationExtension getExtension() {
         return new ReactiveAerospikeRepositoryConfigurationExtension();

src/main/java/org/springframework/data/aerospike/repository/config/ReactiveAerospikeRepositoryConfigurationExtension.java

@@ -1,8 +1,15 @@
 package org.springframework.data.aerospike.repository.config;
 
+import org.springframework.data.aerospike.mapping.Document;
+import org.springframework.data.aerospike.repository.ReactiveAerospikeRepository;
+import org.springframework.data.aerospike.repository.support.ReactiveAerospikeRepositoryFactoryBean;
 import org.springframework.data.repository.config.RepositoryConfigurationExtension;
 import org.springframework.data.repository.core.RepositoryMetadata;
 
+import java.lang.annotation.Annotation;
+import java.util.Collection;
+import java.util.Collections;
+
 /**
  * {@link RepositoryConfigurationExtension} for reactive Aerospike.
  *
@@ -25,6 +32,21 @@ protected String getDefaultKeyValueTemplateRef() {
         return "reactiveAerospikeTemplate";
     }
 
+    @Override
+    public String getRepositoryFactoryBeanClassName() {
+        return ReactiveAerospikeRepositoryFactoryBean.class.getName();
+    }
+
+    @Override
+    protected Collection<Class<? extends Annotation>> getIdentifyingAnnotations() {
+        return Collections.singleton(Document.class);
+    }
+
+    @Override
+    protected Collection<Class<?>> getIdentifyingTypes() {
+        return Collections.singleton(ReactiveAerospikeRepository.class);
+    }
+
     @Override
     protected boolean useRepositoryConfiguration(RepositoryMetadata metadata) {
         return metadata.isReactiveRepository();