Add missing properties to BaseQueryBuilder.

Closes spring-projects#2250

Author: sothawo

src/main/java/org/springframework/data/elasticsearch/client/elc/NativeQuery.java

@@ -46,7 +46,6 @@ public class NativeQuery extends BaseQuery {
 	@Nullable private Suggester suggester;
 	@Nullable private FieldCollapse fieldCollapse;
 	private List<ScriptedField> scriptedFields = Collections.emptyList();
-	private List<RescorerQuery> rescorerQueries = Collections.emptyList();
 
 	public NativeQuery(NativeQueryBuilder builder) {
 		super(builder);
@@ -56,7 +55,6 @@ public NativeQuery(NativeQueryBuilder builder) {
 		this.suggester = builder.getSuggester();
 		this.fieldCollapse = builder.getFieldCollapse();
 		this.scriptedFields = builder.getScriptedFields();
-		this.rescorerQueries = builder.getRescorerQueries();
 	}
 
 	public NativeQuery(@Nullable Query query) {
@@ -94,9 +92,4 @@ public FieldCollapse getFieldCollapse() {
 	public List<ScriptedField> getScriptedFields() {
 		return scriptedFields;
 	}
-
-	@Override
-	public List<RescorerQuery> getRescorerQueries() {
-		return rescorerQueries;
-	}
 }

src/main/java/org/springframework/data/elasticsearch/client/elc/NativeQueryBuilder.java

@@ -45,8 +45,6 @@ public class NativeQueryBuilder extends BaseQueryBuilder<NativeQuery, NativeQuer
 	@Nullable private Suggester suggester;
 	@Nullable private FieldCollapse fieldCollapse;
 	private final List<ScriptedField> scriptedFields = new ArrayList<>();
-	private List<RescorerQuery> rescorerQueries = new ArrayList<>();
-
 	public NativeQueryBuilder() {}
 
 	@Nullable
@@ -77,10 +75,6 @@ public List<ScriptedField> getScriptedFields() {
 		return scriptedFields;
 	}
 
-	public List<RescorerQuery> getRescorerQueries() {
-		return rescorerQueries;
-	}
-
 	public NativeQueryBuilder withQuery(Query query) {
 
 		Assert.notNull(query, "query must not be null");
@@ -135,14 +129,6 @@ public NativeQueryBuilder withScriptedField(ScriptedField scriptedField) {
 		return this;
 	}
 
-	public NativeQueryBuilder withResorerQuery(RescorerQuery resorerQuery) {
-
-		Assert.notNull(resorerQuery, "resorerQuery must not be null");
-
-		this.rescorerQueries.add(resorerQuery);
-		return this;
-	}
-
 	public NativeQuery build() {
 		return new NativeQuery(this);
 	}

src/main/java/org/springframework/data/elasticsearch/core/query/BaseQuery.java

@@ -47,8 +47,8 @@
  */
 public class BaseQuery implements Query {
 
-	protected Pageable pageable = DEFAULT_PAGE;
 	@Nullable protected Sort sort;
+	protected Pageable pageable = DEFAULT_PAGE;
 	protected List<String> fields = new ArrayList<>();
 	@Nullable protected List<String> storedFields;
 	@Nullable protected SourceFilter sourceFilter;
@@ -67,29 +67,39 @@ public class BaseQuery implements Query {
 	@Nullable protected Duration timeout;
 	private boolean explain = false;
 	@Nullable protected List<Object> searchAfter;
-	protected List<RescorerQuery> rescorerQueries = new ArrayList<>();
+	@Nullable protected List<IndexBoost> indicesBoost;
+	protected final List<RescorerQuery> rescorerQueries = new ArrayList<>();
 	@Nullable protected Boolean requestCache;
 	protected List<IdWithRouting> idsWithRouting = Collections.emptyList();
 	protected final List<RuntimeField> runtimeFields = new ArrayList<>();
-	@Nullable protected List<IndexBoost> indicesBoost;
 
 	public BaseQuery() {}
 
 	public <Q extends BaseQuery, B extends BaseQueryBuilder<Q, B>> BaseQuery(BaseQueryBuilder<Q, B> builder) {
 		this.sort = builder.getSort();
 		// do a setPageable after setting the sort, because the pageable may contain an additional sort
 		this.setPageable(builder.getPageable() != null ? builder.getPageable() : DEFAULT_PAGE);
-		this.ids = builder.getIds();
-		this.trackScores = builder.getTrackScores();
-		this.maxResults = builder.getMaxResults();
-		this.indicesOptions = builder.getIndicesOptions();
-		this.minScore = builder.getMinScore();
-		this.preference = builder.getPreference();
-		this.sourceFilter = builder.getSourceFilter();
 		this.fields = builder.getFields();
-		this.highlightQuery = builder.highlightQuery;
+		this.storedFields = builder.getStoredFields();
+		this.sourceFilter = builder.getSourceFilter();
+		this.minScore = builder.getMinScore();
+		this.ids = builder.getIds().isEmpty() ? null : builder.getIds();
 		this.route = builder.getRoute();
+		this.searchType = builder.getSearchType();
+		this.indicesOptions = builder.getIndicesOptions();
+		this.trackScores = builder.getTrackScores();
+		this.preference = builder.getPreference();
+		this.maxResults = builder.getMaxResults();
+		this.highlightQuery = builder.getHighlightQuery();
+		this.trackTotalHits = builder.getTrackTotalHits();
+		this.trackTotalHitsUpTo = builder.getTrackTotalHitsUpTo();
+		this.scrollTime = builder.getScrollTime();
+		this.timeout = builder.getTimeout();
+		this.explain = builder.getExplain();
+		this.searchAfter = builder.getSearchAfter();
 		this.indicesBoost = builder.getIndicesBoost();
+		this.requestCache = builder.getRequestCache();
+		this.idsWithRouting = builder.getIdsWithRouting();
 	}
 
 	@Override