Release: 5.2.2

pom.xml

@@ -20,7 +20,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>io.github.boostchicken</groupId>
     <artifactId>spring-data-dynamodb</artifactId>
-    <version>5.2.2-SNAPSHOT</version>
+    <version>5.2.3-SNAPSHOT</version>
     <name>Spring Data DynamoDB</name>
     <inceptionYear>2018</inceptionYear>
 

src/main/java/org/socialsignin/spring/data/dynamodb/repository/Query.java

@@ -20,7 +20,7 @@
 import java.lang.annotation.Retention;
 import java.lang.annotation.RetentionPolicy;
 import java.lang.annotation.Target;
-
+import org.socialsignin.spring.data.dynamodb.repository.QueryConstants.ConsistentReadMode;
 import static org.socialsignin.spring.data.dynamodb.repository.QueryConstants.QUERY_LIMIT_UNLIMITED;
 
 @Retention(RetentionPolicy.RUNTIME)
@@ -47,4 +47,7 @@
 	 *      Expressions</a>
 	 */
 	int limit() default QUERY_LIMIT_UNLIMITED;
+
+
+	ConsistentReadMode consistentReads() default ConsistentReadMode.DEFAULT;
 }

src/main/java/org/socialsignin/spring/data/dynamodb/repository/QueryConstants.java

@@ -15,11 +15,16 @@
  */
 package org.socialsignin.spring.data.dynamodb.repository;
 
-public final class QueryConstants {
+public class QueryConstants {
 
 	private QueryConstants() {
 	}
 
 	public static final int QUERY_LIMIT_UNLIMITED = Integer.MIN_VALUE;
 
+	public enum ConsistentReadMode {
+		DEFAULT,
+		CONSISTENT,
+		EVENTUAL
+	}
 }

src/main/java/org/socialsignin/spring/data/dynamodb/repository/cdi/DynamoDBRepositoryBean.java

@@ -46,6 +46,8 @@ class DynamoDBRepositoryBean<T> extends CdiRepositoryBean<T> {
 
 	private final Bean<DynamoDBOperations> dynamoDBOperationsBean;
 
+	private final Bean<DynamoDBMapper> dynamoDBMapperBean;
+
 	/**
 	 * Constructs a {@link DynamoDBRepositoryBean}.
 	 * 
@@ -62,7 +64,7 @@ class DynamoDBRepositoryBean<T> extends CdiRepositoryBean<T> {
 	 */
 	DynamoDBRepositoryBean(BeanManager beanManager, Bean<AmazonDynamoDB> amazonDynamoDBBean,
 			Bean<DynamoDBMapperConfig> dynamoDBMapperConfigBean, Bean<DynamoDBOperations> dynamoDBOperationsBean,
-			Set<Annotation> qualifiers, Class<T> repositoryType) {
+						   Bean<DynamoDBMapper> dynamoDBMapperBean, Set<Annotation> qualifiers, Class<T> repositoryType) {
 
 		super(qualifiers, repositoryType, beanManager);
 		if (dynamoDBOperationsBean == null) {
@@ -77,6 +79,7 @@ class DynamoDBRepositoryBean<T> extends CdiRepositoryBean<T> {
 		this.amazonDynamoDBBean = amazonDynamoDBBean;
 		this.dynamoDBMapperConfigBean = dynamoDBMapperConfigBean;
 		this.dynamoDBOperationsBean = dynamoDBOperationsBean;
+		this.dynamoDBMapperBean = dynamoDBMapperBean;
 	}
 
 	/*
@@ -97,15 +100,20 @@ protected T create(CreationalContext<T> creationalContext, Class<T> repositoryTy
 				? null
 				: getDependencyInstance(dynamoDBMapperConfigBean, DynamoDBMapperConfig.class);
 
+		DynamoDBMapper dynamoDBMapper = dynamoDBMapperBean == null
+				? null
+				: getDependencyInstance(dynamoDBMapperBean, DynamoDBMapper.class);
+
 		DynamoDBOperations dynamoDBOperations = dynamoDBOperationsBean == null
 				? null
 				: getDependencyInstance(dynamoDBOperationsBean, DynamoDBOperations.class);
 
 		if (dynamoDBMapperConfig == null) {
 			dynamoDBMapperConfig = DynamoDBMapperConfig.DEFAULT;
 		}
-		DynamoDBMapper dynamoDBMapper = new DynamoDBMapper(amazonDynamoDB, dynamoDBMapperConfig);
-
+		if(dynamoDBMapper == null) {
+			dynamoDBMapper = new DynamoDBMapper(amazonDynamoDB, dynamoDBMapperConfig);
+		}
 		if (dynamoDBOperations == null) {
 			dynamoDBOperations = new DynamoDBTemplate(amazonDynamoDB, dynamoDBMapper, dynamoDBMapperConfig);
 		}

src/main/java/org/socialsignin/spring/data/dynamodb/repository/cdi/DynamoDBRepositoryExtension.java

@@ -17,6 +17,7 @@
 
 import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
 import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClient;
+import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper;
 import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapperConfig;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -54,6 +55,7 @@ public class DynamoDBRepositoryExtension extends CdiRepositoryExtensionSupport {
 
 	private final Map<Set<Annotation>, Bean<DynamoDBMapperConfig>> dbMapperConfigs = new HashMap<Set<Annotation>, Bean<DynamoDBMapperConfig>>();
 
+	private final Map<Set<Annotation>, Bean<DynamoDBMapper>> dbMapper = new HashMap<>();
 	public DynamoDBRepositoryExtension() {
 		LOGGER.info("Activating CDI extension for Spring Data DynamoDB repositories.");
 	}
@@ -89,6 +91,14 @@ <X> void processBean(@Observes ProcessBean<X> processBean) {
 					dbMapperConfigs.put(qualifiers, (Bean<DynamoDBMapperConfig>) bean);
 				}
 			}
+			if (type instanceof Class<?> && DynamoDBMapper.class.isAssignableFrom((Class<?>) type)) {
+				Set<Annotation> qualifiers = new HashSet<Annotation>(bean.getQualifiers());
+				if (bean.isAlternative() || !dbMapper.containsKey(qualifiers)) {
+					LOGGER.debug("Discovered '{}' with qualifiers {}.", DynamoDBMapper.class.getName(),
+							qualifiers);
+					dbMapper.put(qualifiers, (Bean<DynamoDBMapper>) bean);
+				}
+			}
 		}
 	}
 
@@ -138,16 +148,17 @@ private <T> Bean<T> createRepositoryBean(Class<T> repositoryType, Set<Annotation
 		// qualifiers of the repository.
 		Bean<DynamoDBMapperConfig> dynamoDBMapperConfigBean = dbMapperConfigs.get(qualifiers);
 
+		Bean<DynamoDBOperations> dynamoDBOperationsBean = dynamoDBOperationss.get(qualifiers);
 		if (amazonDynamoDBBean == null) {
 			throw new UnsatisfiedResolutionException(
 					String.format("Unable to resolve a bean for '%s' with qualifiers %s.",
 							AmazonDynamoDBClient.class.getName(), qualifiers));
 		}
 
-		Bean<DynamoDBOperations> dynamoDBOperationsBean = dynamoDBOperationss.get(qualifiers);
+		Bean<DynamoDBMapper> dynamoDBMapperBean = dbMapper.get(qualifiers);
 
 		// Construct and return the repository bean.
 		return new DynamoDBRepositoryBean<T>(beanManager, amazonDynamoDBBean, dynamoDBMapperConfigBean,
-				dynamoDBOperationsBean, qualifiers, repositoryType);
+				dynamoDBOperationsBean, dynamoDBMapperBean, qualifiers, repositoryType);
 	}
 }

src/main/java/org/socialsignin/spring/data/dynamodb/repository/config/DynamoDBMapperFactory.java

@@ -15,26 +15,23 @@
  */
 package org.socialsignin.spring.data.dynamodb.repository.config;
 
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.BeanFactory;
+import org.springframework.beans.factory.BeanFactoryAware;
 import org.springframework.beans.factory.FactoryBean;
-import org.springframework.beans.factory.annotation.Autowired;
 
 import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
 import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper;
 import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapperConfig;
 
-public class DynamoDBMapperFactory implements FactoryBean<DynamoDBMapper> {
-
-	private final AmazonDynamoDB amazonDynamoDB;
-	private final DynamoDBMapperConfig dynamoDBMapperConfig;
-
-	@Autowired
-	public DynamoDBMapperFactory(AmazonDynamoDB amazonDynamoDB, DynamoDBMapperConfig dynamoDBMapperConfig) {
-		this.amazonDynamoDB = amazonDynamoDB;
-		this.dynamoDBMapperConfig = dynamoDBMapperConfig;
-	}
+public class DynamoDBMapperFactory implements FactoryBean<DynamoDBMapper>, BeanFactoryAware {
 
+	private BeanFactory beanFactory;
+
 	@Override
 	public DynamoDBMapper getObject() throws Exception {
+		AmazonDynamoDB amazonDynamoDB = beanFactory.getBean(AmazonDynamoDB.class);
+		DynamoDBMapperConfig dynamoDBMapperConfig = beanFactory.getBean(DynamoDBMapperConfig.class);
 		return new DynamoDBMapper(amazonDynamoDB, dynamoDBMapperConfig);
 	}
 
@@ -43,4 +40,9 @@ public Class<?> getObjectType() {
 		return DynamoDBMapper.class;
 	}
 
+	@Override
+	public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
+		this.beanFactory = beanFactory;
+	}
+
 }

src/main/java/org/socialsignin/spring/data/dynamodb/repository/config/DynamoDBRepositoryConfigExtension.java

@@ -260,7 +260,7 @@ public void registerBeansForRoot(BeanDefinitionRegistry registry,
 		this.registry = registry;
 
 		this.dynamoDBMapperConfigName = getBeanNameWithModulePrefix("DynamoDBMapperConfig");
-		Optional dynamoDBMapperConfigRef = configurationSource.getAttribute("dynamoDBMapperConfigRef");
+		Optional<String> dynamoDBMapperConfigRef = configurationSource.getAttribute("dynamoDBMapperConfigRef");
 
 		if (!dynamoDBMapperConfigRef.isPresent()) {
 			BeanDefinitionBuilder dynamoDBMapperConfigBuiilder = BeanDefinitionBuilder
@@ -269,10 +269,13 @@ public void registerBeansForRoot(BeanDefinitionRegistry registry,
 					dynamoDBMapperConfigBuiilder.getBeanDefinition());
 		}
 
-		this.dynamoDBMapperName = getBeanNameWithModulePrefix("DynamoDBMapper");
-		BeanDefinitionBuilder dynamoDBMapperBuilder = BeanDefinitionBuilder
-				.genericBeanDefinition(DynamoDBMapperFactory.class);
-		registry.registerBeanDefinition(this.dynamoDBMapperName, dynamoDBMapperBuilder.getBeanDefinition());
+		Optional<String> dynamoDBMapperRef = configurationSource.getAttribute("dynamoDBMapperRef");
+		if(!dynamoDBMapperRef.isPresent()) {
+			this.dynamoDBMapperName = getBeanNameWithModulePrefix("DynamoDBMapper");
+			BeanDefinitionBuilder dynamoDBMapperBuilder = BeanDefinitionBuilder
+					.genericBeanDefinition(DynamoDBMapperFactory.class);
+			registry.registerBeanDefinition(this.dynamoDBMapperName, dynamoDBMapperBuilder.getBeanDefinition());
+		}
 	}
 
 	protected String getBeanNameWithModulePrefix(String baseBeanName) {

src/main/java/org/socialsignin/spring/data/dynamodb/repository/config/EnableDynamoDBRepositories.java

@@ -147,6 +147,17 @@
 	 */
 	String dynamoDBMapperConfigRef() default "";
 
+	/**
+	 * Returns the
+	 * {@link com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper }
+	 * reference to be used
+	 *
+	 * @return The
+	 *         {@link com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper}
+	 *         bean name
+	 */
+	String dynamoDBMapperRef() default "";
+
 	/**
 	 * Returns the {@link javax.validation.Validator } reference to be used for to
 	 * validate DynamoDB entities

src/main/java/org/socialsignin/spring/data/dynamodb/repository/query/AbstractDynamoDBQueryCreator.java

@@ -19,6 +19,7 @@
 import com.amazonaws.services.dynamodbv2.model.ComparisonOperator;
 import org.socialsignin.spring.data.dynamodb.core.DynamoDBOperations;
 import org.socialsignin.spring.data.dynamodb.query.Query;
+import org.socialsignin.spring.data.dynamodb.repository.QueryConstants;
 import org.socialsignin.spring.data.dynamodb.repository.support.DynamoDBEntityInformation;
 import org.socialsignin.spring.data.dynamodb.repository.support.DynamoDBIdIsHashAndRangeKeyEntityInformation;
 import org.springframework.data.mapping.PropertyPath;
@@ -47,23 +48,26 @@ public abstract class AbstractDynamoDBQueryCreator<T, ID, R>
 	protected final DynamoDBOperations dynamoDBOperations;
 	protected final Optional<String> projection;
 	protected final Optional<Integer> limit;
+	protected final QueryConstants.ConsistentReadMode consistentReads;
 
 	public AbstractDynamoDBQueryCreator(PartTree tree, DynamoDBEntityInformation<T, ID> entityMetadata,
-			Optional<String> projection, Optional<Integer> limitResults, DynamoDBOperations dynamoDBOperations) {
+			Optional<String> projection, Optional<Integer> limitResults, QueryConstants.ConsistentReadMode consistentReads, DynamoDBOperations dynamoDBOperations) {
 		super(tree);
 		this.entityMetadata = entityMetadata;
 		this.projection = projection;
 		this.limit = limitResults;
+		this.consistentReads = consistentReads;
 		this.dynamoDBOperations = dynamoDBOperations;
 	}
 
 	public AbstractDynamoDBQueryCreator(PartTree tree, ParameterAccessor parameterAccessor,
-			DynamoDBEntityInformation<T, ID> entityMetadata, Optional<String> projection,
-			Optional<Integer> limitResults, DynamoDBOperations dynamoDBOperations) {
+										DynamoDBEntityInformation<T, ID> entityMetadata, Optional<String> projection,
+										Optional<Integer> limitResults, QueryConstants.ConsistentReadMode consistentReads, DynamoDBOperations dynamoDBOperations) {
 		super(tree, parameterAccessor);
 		this.entityMetadata = entityMetadata;
 		this.projection = projection;
 		this.limit = limitResults;
+		this.consistentReads = consistentReads;
 		this.dynamoDBOperations = dynamoDBOperations;
 	}
 

src/main/java/org/socialsignin/spring/data/dynamodb/repository/query/AbstractDynamoDBQueryCriteria.java

@@ -29,6 +29,7 @@
 import org.socialsignin.spring.data.dynamodb.marshaller.Date2IsoDynamoDBMarshaller;
 import org.socialsignin.spring.data.dynamodb.marshaller.Instant2IsoDynamoDBMarshaller;
 import org.socialsignin.spring.data.dynamodb.query.Query;
+import org.socialsignin.spring.data.dynamodb.repository.QueryConstants;
 import org.socialsignin.spring.data.dynamodb.repository.support.DynamoDBEntityInformation;
 import org.socialsignin.spring.data.dynamodb.utils.SortHandler;
 import org.springframework.data.domain.Sort;
@@ -73,6 +74,7 @@ public abstract class AbstractDynamoDBQueryCriteria<T, ID> implements DynamoDBQu
 	protected Sort sort = Sort.unsorted();
 	protected Optional<String> projection = Optional.empty();
 	protected Optional<Integer> limit = Optional.empty();
+	protected QueryConstants.ConsistentReadMode consistentReads = QueryConstants.ConsistentReadMode.DEFAULT;
 
 	public abstract boolean isApplicableForLoad();
 
@@ -134,6 +136,17 @@ protected QueryRequest buildQueryRequest(String tableName, String theIndexName,
 			}
 
 			limit.ifPresent(queryRequest::setLimit);
+
+			switch (consistentReads) {
+				case CONSISTENT:
+					queryRequest.setConsistentRead(true);
+					break;
+				case EVENTUAL:
+					queryRequest.setConsistentRead(false);
+					break;
+				default:
+					break;
+			}
 			applySortIfSpecified(queryRequest, new ArrayList<>(new HashSet<>(allowedSortProperties)));
 		}
 		return queryRequest;
@@ -703,4 +716,10 @@ public DynamoDBQueryCriteria<T, ID> withLimit(Optional<Integer> limit) {
 		this.limit = limit;
 		return this;
 	}
+
+	@Override
+	public DynamoDBQueryCriteria<T, ID> withConsistentReads(QueryConstants.ConsistentReadMode consistentReads) {
+		this.consistentReads = consistentReads;
+		return this;
+	}
 }

src/main/java/org/socialsignin/spring/data/dynamodb/repository/query/DynamoDBCountQueryCreator.java

@@ -18,6 +18,7 @@
 import org.socialsignin.spring.data.dynamodb.core.DynamoDBOperations;
 import org.socialsignin.spring.data.dynamodb.query.Query;
 import org.socialsignin.spring.data.dynamodb.query.StaticQuery;
+import org.socialsignin.spring.data.dynamodb.repository.QueryConstants;
 import org.socialsignin.spring.data.dynamodb.repository.support.DynamoDBEntityInformation;
 import org.springframework.data.domain.Sort;
 import org.springframework.data.repository.query.ParameterAccessor;
@@ -32,13 +33,13 @@ public class DynamoDBCountQueryCreator<T, ID> extends AbstractDynamoDBQueryCreat
 
 	public DynamoDBCountQueryCreator(PartTree tree, DynamoDBEntityInformation<T, ID> entityMetadata,
 			DynamoDBOperations dynamoDBOperations, boolean pageQuery) {
-		super(tree, entityMetadata, Optional.empty(), Optional.empty(), dynamoDBOperations);
+		super(tree, entityMetadata, Optional.empty(), Optional.empty(), QueryConstants.ConsistentReadMode.DEFAULT, dynamoDBOperations);
 		this.pageQuery = pageQuery;
 	}
 
 	public DynamoDBCountQueryCreator(PartTree tree, ParameterAccessor parameterAccessor,
 			DynamoDBEntityInformation<T, ID> entityMetadata, DynamoDBOperations dynamoDBOperations, boolean pageQuery) {
-		super(tree, parameterAccessor, entityMetadata, Optional.empty(), Optional.empty(), dynamoDBOperations);
+		super(tree, parameterAccessor, entityMetadata, Optional.empty(), Optional.empty(), QueryConstants.ConsistentReadMode.DEFAULT, dynamoDBOperations);
 		this.pageQuery = pageQuery;
 
 	}

src/main/java/org/socialsignin/spring/data/dynamodb/repository/query/DynamoDBQueryCreator.java

@@ -18,6 +18,7 @@
 import org.socialsignin.spring.data.dynamodb.core.DynamoDBOperations;
 import org.socialsignin.spring.data.dynamodb.query.Query;
 import org.socialsignin.spring.data.dynamodb.query.StaticQuery;
+import org.socialsignin.spring.data.dynamodb.repository.QueryConstants;
 import org.socialsignin.spring.data.dynamodb.repository.support.DynamoDBEntityInformation;
 import org.springframework.data.domain.Sort;
 import org.springframework.data.repository.query.ParameterAccessor;
@@ -29,9 +30,9 @@
 public class DynamoDBQueryCreator<T, ID> extends AbstractDynamoDBQueryCreator<T, ID, T> {
 
 	public DynamoDBQueryCreator(PartTree tree, ParameterAccessor parameterAccessor,
-			DynamoDBEntityInformation<T, ID> entityMetadata, Optional<String> projection, Optional<Integer> limit,
+			DynamoDBEntityInformation<T, ID> entityMetadata, Optional<String> projection, Optional<Integer> limit, QueryConstants.ConsistentReadMode consistentReads,
 			DynamoDBOperations dynamoDBOperations) {
-		super(tree, parameterAccessor, entityMetadata, projection, limit, dynamoDBOperations);
+		super(tree, parameterAccessor, entityMetadata, projection, limit, consistentReads, dynamoDBOperations);
 	}
 
 	@Override
@@ -42,6 +43,7 @@ protected Query<T> complete(@Nullable DynamoDBQueryCriteria<T, ID> criteria, Sor
 			criteria.withSort(sort);
 			criteria.withProjection(projection);
 			criteria.withLimit(limit);
+			criteria.withConsistentReads(consistentReads);
 			return criteria.buildQuery(dynamoDBOperations);
 		}
 	}

src/main/java/org/socialsignin/spring/data/dynamodb/repository/query/DynamoDBQueryCriteria.java

@@ -18,6 +18,7 @@
 import com.amazonaws.services.dynamodbv2.model.ComparisonOperator;
 import org.socialsignin.spring.data.dynamodb.core.DynamoDBOperations;
 import org.socialsignin.spring.data.dynamodb.query.Query;
+import org.socialsignin.spring.data.dynamodb.repository.QueryConstants;
 import org.springframework.data.domain.Sort;
 
 import java.util.Optional;
@@ -45,6 +46,8 @@ DynamoDBQueryCriteria<T, ID> withSingleValueCriteria(String propertyName, Compar
 
 	DynamoDBQueryCriteria<T, ID> withLimit(Optional<Integer> limit);
 
+	DynamoDBQueryCriteria<T, ID> withConsistentReads(QueryConstants.ConsistentReadMode reads);
+
 	Query<T> buildQuery(DynamoDBOperations dynamoDBOperations);
 
 	Query<Long> buildCountQuery(DynamoDBOperations dynamoDBOperations, boolean pageQuery);