#41 Fixing findAll(Iterable<ID>) method, from pull request by @tpenakov and modifying with fix from @sirjasiu for overridden table names

michaellavelle · michaellavelle · commit e3fedcbe2394 · 2015-06-18T13:32:37.000+01:00
diff --git a/src/main/java/org/socialsignin/spring/data/dynamodb/repository/support/SimpleDynamoDBCrudRepository.java b/src/main/java/org/socialsignin/spring/data/dynamodb/repository/support/SimpleDynamoDBCrudRepository.java
@@ -20,6 +20,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Map.Entry;
 
 import org.socialsignin.spring.data.dynamodb.core.DynamoDBOperations;
 import org.socialsignin.spring.data.dynamodb.repository.DynamoDBCrudRepository;
@@ -91,7 +92,7 @@ public List<T> findAll(Iterable<ID> ids) {
 			}
 		}
 		keyPairsMap.put(domainType, keyPairs);
-		return (List<T>) dynamoDBOperations.batchLoad(keyPairsMap).get(domainType);
+		return (List<T>) dynamoDBOperations.batchLoad(keyPairsMap).get(dynamoDBOperations.getOverriddenTableName(entityInformation.getDynamoDBTableName()));
 	}
 
 	protected T load(ID id) {

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,7 @@`
`20`	`20`	`import java.util.HashMap;`
`21`	`21`	`import java.util.List;`
`22`	`22`	`import java.util.Map;`
	`23`	`+import java.util.Map.Entry;`
`23`	`24`
`24`	`25`	`import org.socialsignin.spring.data.dynamodb.core.DynamoDBOperations;`
`25`	`26`	`import org.socialsignin.spring.data.dynamodb.repository.DynamoDBCrudRepository;`
`@@ -91,7 +92,7 @@ public List<T> findAll(Iterable<ID> ids) {`
`91`	`92`	`}`
`92`	`93`	`}`
`93`	`94`	`keyPairsMap.put(domainType, keyPairs);`
`94`		`- return (List<T>) dynamoDBOperations.batchLoad(keyPairsMap).get(domainType);`
	`95`	`+ return (List<T>) dynamoDBOperations.batchLoad(keyPairsMap).get(dynamoDBOperations.getOverriddenTableName(entityInformation.getDynamoDBTableName()));`
`95`	`96`	`}`
`96`	`97`
`97`	`98`	`protected T load(ID id) {`