diff --git a/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/repository/config/CassandraRepositoryConfigurationExtension.java b/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/repository/config/CassandraRepositoryConfigurationExtension.java index a714504d2..51b080d8e 100644 --- a/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/repository/config/CassandraRepositoryConfigurationExtension.java +++ b/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/repository/config/CassandraRepositoryConfigurationExtension.java @@ -1,5 +1,5 @@ /* - * Copyright 2013-2014 the original author or authors + * Copyright 2013-2016 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. @@ -19,6 +19,7 @@ import org.springframework.beans.factory.support.BeanDefinitionBuilder; import org.springframework.cassandra.config.xml.ParsingUtils; import org.springframework.core.annotation.AnnotationAttributes; import org.springframework.data.cassandra.config.DefaultBeanNames; +import org.springframework.data.cassandra.mapping.Table; import org.springframework.data.cassandra.repository.support.CassandraRepositoryFactoryBean; import org.springframework.data.repository.config.AnnotationRepositoryConfigurationSource; import org.springframework.data.repository.config.RepositoryConfigurationExtension; @@ -27,10 +28,15 @@ import org.springframework.data.repository.config.XmlRepositoryConfigurationSour import org.springframework.util.StringUtils; import org.w3c.dom.Element; +import java.lang.annotation.Annotation; +import java.util.Collection; +import java.util.Collections; + /** * {@link RepositoryConfigurationExtension} for Cassandra. * * @author Alex Shvid + * @author Mark Paluch */ public class CassandraRepositoryConfigurationExtension extends RepositoryConfigurationExtensionSupport { @@ -65,4 +71,14 @@ public class CassandraRepositoryConfigurationExtension extends RepositoryConfigu builder.addPropertyReference("cassandraTemplate", cassandraTemplateRef); } } + + /* + * (non-Javadoc) + * @see org.springframework.data.repository.config.RepositoryConfigurationExtensionSupport#getIdentifyingAnnotations() + */ + @Override + protected Collection> getIdentifyingAnnotations() { + return Collections.> singleton(Table.class); + } + }