#55 Fixing issue with count by hash and count by hash and range queries

Author: michaellavelle

src/main/java/org/socialsignin/spring/data/dynamodb/query/CountByHashAndRangeKeyQuery.java

@@ -24,16 +24,18 @@ public class CountByHashAndRangeKeyQuery<T> extends AbstractSingleEntityQuery<Lo
 
 	private Object hashKey;
 	private Object rangeKey;
+	private Class<T> entityClass;
 
 	public CountByHashAndRangeKeyQuery(DynamoDBOperations dynamoDBOperations, Class<T> clazz, Object hashKey, Object rangeKey) {
 		super(dynamoDBOperations, Long.class);
 		this.hashKey = hashKey;
 		this.rangeKey = rangeKey;
+		this.entityClass = clazz;
 	}
 
 	@Override
 	public Long getSingleResult() {
-		return dynamoDBOperations.load(clazz, hashKey, rangeKey) == null ? 0l : 1l;
+		return dynamoDBOperations.load(entityClass, hashKey, rangeKey) == null ? 0l : 1l;
 	}
 
 }

src/main/java/org/socialsignin/spring/data/dynamodb/query/CountByHashKeyQuery.java

@@ -23,15 +23,17 @@
 public class CountByHashKeyQuery<T> extends AbstractSingleEntityQuery<Long> implements Query<Long> {
 
 	private Object hashKey;
+	private Class<T> entityClass;
 
 	public CountByHashKeyQuery(DynamoDBOperations dynamoDBOperations, Class<T> clazz, Object hashKey) {
 		super(dynamoDBOperations, Long.class);
 		this.hashKey = hashKey;
+		this.entityClass = clazz;
 	}
 
 	@Override
 	public Long getSingleResult() {
-		return dynamoDBOperations.load(clazz, hashKey) == null ? 0l : 1l;
+		return dynamoDBOperations.load(entityClass, hashKey) == null ? 0l : 1l;
 	}
 
 }