DATAES-200 Set persistent entity index and type in query when no indices/types are set.

Author: kevinleturc

src/main/java/org/springframework/data/elasticsearch/core/ElasticsearchTemplate.java

@@ -335,6 +335,7 @@ public <T> FacetedPage<T> queryForPage(StringQuery query, Class<T> clazz, Search
 	@Override
 	public <T> CloseableIterator<T> stream(CriteriaQuery query, Class<T> clazz) {
 		final long scrollTimeInMillis = TimeValue.timeValueMinutes(1).millis();
+		setPersistentEntityIndexAndType(query, clazz);
 		final String initScrollId = scan(query, scrollTimeInMillis, false);
 		return doStream(initScrollId, scrollTimeInMillis, clazz, resultsMapper);
 	}
@@ -347,6 +348,7 @@ public <T> CloseableIterator<T> stream(SearchQuery query, Class<T> clazz) {
 	@Override
 	public <T> CloseableIterator<T> stream(SearchQuery query, final Class<T> clazz, final SearchResultMapper mapper) {
 		final long scrollTimeInMillis = TimeValue.timeValueMinutes(1).millis();
+		setPersistentEntityIndexAndType(query, clazz);
 		final String initScrollId = scan(query, scrollTimeInMillis, false);
 		return doStream(initScrollId, scrollTimeInMillis, clazz, mapper);
 	}
@@ -896,12 +898,7 @@ public Map getSetting(String indexName) {
 	}
 
 	private <T> SearchRequestBuilder prepareSearch(Query query, Class<T> clazz) {
-		if (query.getIndices().isEmpty()) {
-			query.addIndices(retrieveIndexNameFromPersistentEntity(clazz));
-		}
-		if (query.getTypes().isEmpty()) {
-			query.addTypes(retrieveTypeFromPersistentEntity(clazz));
-		}
+		setPersistentEntityIndexAndType(query, clazz);
 		return prepareSearch(query);
 	}
 
@@ -1063,6 +1060,15 @@ private void setPersistentEntityId(Object entity, String id) {
 		}
 	}
 
+	private void setPersistentEntityIndexAndType(Query query, Class clazz) {
+		if (query.getIndices().isEmpty()) {
+			query.addIndices(retrieveIndexNameFromPersistentEntity(clazz));
+		}
+		if (query.getTypes().isEmpty()) {
+			query.addTypes(retrieveTypeFromPersistentEntity(clazz));
+		}
+	}
+
 	private String[] retrieveIndexNameFromPersistentEntity(Class clazz) {
 		if (clazz != null) {
 			return new String[]{getPersistentEntityFor(clazz).getIndexName()};