Cwikius-Spring/spring-data-elasticsearch
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-data-elasticsearch.git synced 2025-06-23 04:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

DATAES-52 - Support Get & Multi Get

Browse Source
This commit is contained in:
Mohsin Husen 2014-02-23 15:25:30 +00:00
parent 38bbb0fd8e
commit b1edd72205
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/main/java/org/springframework/data/elasticsearch/core/ElasticsearchTemplate.java
View File

@ -254,15 +254,20 @@ public class ElasticsearchTemplate implements ElasticsearchOperations {
return resultsMapper.mapResults(getMultiResponse(searchQuery, clazz), clazz); return resultsMapper.mapResults(getMultiResponse(searchQuery, clazz), clazz);
} }
private <T> MultiGetResponse getMultiResponse(SearchQuery searchQuery, Class<T> clazz) { private <T> MultiGetResponse getMultiResponse(Query searchQuery, Class<T> clazz) {
ElasticsearchPersistentEntity<T> persistentEntity = getPersistentEntityFor(clazz); String indexName = isNotEmpty(searchQuery.getIndices()) ? searchQuery.getIndices().get(0) : getPersistentEntityFor(clazz).getIndexName();
String type = isNotEmpty(searchQuery.getTypes()) ? searchQuery.getTypes().get(0) : getPersistentEntityFor(clazz).getIndexType();
Assert.notNull(indexName, "No index defined for Query");
Assert.notNull(type, "No type define for Query");
Assert.notEmpty(searchQuery.getIds(), "No Id define for Query");
MultiGetRequestBuilder builder = client.prepareMultiGet(); MultiGetRequestBuilder builder = client.prepareMultiGet();
for (String id : searchQuery.getIds()) { for (String id : searchQuery.getIds()) {
MultiGetRequest.Item item = new MultiGetRequest.Item(persistentEntity.getIndexName(), persistentEntity.getIndexType(), id); MultiGetRequest.Item item = new MultiGetRequest.Item(indexName, type, id);
if (searchQuery.getRoute() != null) { if (searchQuery.getRoute() != null) {
item = item.routing(searchQuery.getRoute()); item = item.routing(searchQuery.getRoute());