DATAES-449 - Pass route parameter to created search request.

Original PR: spring-projects#383

Author: sothawo

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

@@ -560,6 +560,10 @@ private SearchRequest prepareSearchRequest(Query query, @Nullable Class<?> clazz
 			sourceBuilder.trackTotalHits(query.getTrackTotalHits());
 		}
 
+		if (StringUtils.hasLength(query.getRoute())) {
+			request.routing(query.getRoute());
+		}
+
 		request.source(sourceBuilder);
 		return request;
 	}
@@ -726,6 +730,10 @@ private SearchRequestBuilder prepareSearchRequestBuilder(Query query, Client cli
 			searchRequestBuilder.setTrackTotalHits(query.getTrackTotalHits());
 		}
 
+		if (StringUtils.hasLength(query.getRoute())) {
+			searchRequestBuilder.setRouting(query.getRoute());
+		}
+
 		return searchRequestBuilder;
 	}
 

src/test/java/org/springframework/data/elasticsearch/core/RequestFactoryTest.java

@@ -15,10 +15,12 @@
  */
 package org.springframework.data.elasticsearch.core;
 
+import static org.assertj.core.api.Assertions.*;
 import static org.skyscreamer.jsonassert.JSONAssert.*;
 
 import java.util.Collections;
 
+import org.elasticsearch.action.search.SearchRequest;
 import org.json.JSONException;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
@@ -103,6 +105,18 @@ void shouldBuildSearchWithGeoSortSort() throws JSONException {
 		assertEquals(expected, searchRequest, false);
 	}
 
+	@Test // DATAES-449
+	void shouldAddRouting() throws JSONException {
+		String route = "route66";
+		CriteriaQuery query = new CriteriaQuery(new Criteria("lastName").is("Smith"));
+		query.setRoute(route);
+		converter.updateQuery(query, Person.class);
+
+		SearchRequest searchRequest = requestFactory.searchRequest(query, Person.class, IndexCoordinates.of("persons"));
+
+		assertThat(searchRequest.routing()).isEqualTo(route);
+	}
+
 	static class Person {
 		@Id String id;
 		@Field(name = "last-name") String lastName;