Spotless

mayya-sharipova · Oct 2, 2024 · 426db4d · 426db4d
1 parent 6e6abae
commit 426db4d
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 15 deletions.
diff --git a/server/src/main/java/org/elasticsearch/search/vectors/KnnScoreDocQueryBuilder.java b/server/src/main/java/org/elasticsearch/search/vectors/KnnScoreDocQueryBuilder.java
@@ -50,8 +50,13 @@ public class KnnScoreDocQueryBuilder extends AbstractQueryBuilder<KnnScoreDocQue
  * @param scoreDocs the docs and scores this query should match. The array must be
  * sorted in order of ascending doc IDs.
  */
- public KnnScoreDocQueryBuilder(ScoreDoc[] scoreDocs, String fieldName,
- VectorData queryVector, Float vectorSimilarity, List<QueryBuilder> filterQueries) {
+ public KnnScoreDocQueryBuilder(
+ ScoreDoc[] scoreDocs,
+ String fieldName,
+ VectorData queryVector,
+ Float vectorSimilarity,
+ List<QueryBuilder> filterQueries
+ ) {
  this.scoreDocs = scoreDocs;
  this.fieldName = fieldName;
  this.queryVector = queryVector;
@@ -83,7 +88,7 @@ public KnnScoreDocQueryBuilder(StreamInput in) throws IOException {
  } else {
  this.vectorSimilarity = null;
  }
- if (in.getTransportVersion().onOrAfter(TransportVersions.TO_CHILD_BLOCK_JOIN_QUERY)){
+ if (in.getTransportVersion().onOrAfter(TransportVersions.TO_CHILD_BLOCK_JOIN_QUERY)) {
  this.filterQueries = readQueries(in);
  } else {
  this.filterQueries = List.of();
@@ -111,8 +116,6 @@ Float vectorSimilarity() {
  return vectorSimilarity;
  }
 
-
-
  @Override
  protected void doWriteTo(StreamOutput out) throws IOException {
  out.writeArray(Lucene::writeScoreDoc, scoreDocs);
@@ -133,7 +136,7 @@ protected void doWriteTo(StreamOutput out) throws IOException {
  || out.getTransportVersion().isPatchFrom(TransportVersions.FIX_VECTOR_SIMILARITY_INNER_HITS_BACKPORT_8_15)) {
  out.writeOptionalFloat(vectorSimilarity);
  }
- if (out.getTransportVersion().onOrAfter(TransportVersions.TO_CHILD_BLOCK_JOIN_QUERY)){
+ if (out.getTransportVersion().onOrAfter(TransportVersions.TO_CHILD_BLOCK_JOIN_QUERY)) {
  writeQueries(out, filterQueries);
  }
  }
@@ -199,9 +202,9 @@ protected QueryBuilder doRewrite(QueryRewriteContext queryRewriteContext) throws
  // so add them as should clauses to a filter
  BoolQueryBuilder boolQuery = new BoolQueryBuilder();
  boolQuery.must(exactKnnQuery);
- boolQuery.filter(new BoolQueryBuilder()
- .should(filterQueryChildren)
-  .should(new ToChildBlockJoinQueryBuilder(filterQueryChildren)));
+ boolQuery.filter(
+ new BoolQueryBuilder().should(filterQueryChildren).should(new ToChildBlockJoinQueryBuilder(filterQueryChildren))
+ );
  return boolQuery;
  }
  }

diff --git a/server/src/main/java/org/elasticsearch/search/vectors/KnnVectorQueryBuilder.java b/server/src/main/java/org/elasticsearch/search/vectors/KnnVectorQueryBuilder.java
@@ -434,9 +434,9 @@ protected QueryBuilder doRewrite(QueryRewriteContext ctx) throws IOException {
  // so add them as should clauses to a filter
  BoolQueryBuilder boolQuery = new BoolQueryBuilder();
  boolQuery.must(exactKnnQuery);
- boolQuery.filter(new BoolQueryBuilder()
- .should(filterQueryChildren)
-  .should(new ToChildBlockJoinQueryBuilder(filterQueryChildren)));
+ boolQuery.filter(
+ new BoolQueryBuilder().should(filterQueryChildren).should(new ToChildBlockJoinQueryBuilder(filterQueryChildren))
+ );
  return boolQuery;
  }
  }

diff --git a/server/src/test/java/org/elasticsearch/index/query/ToChildBlockJoinQueryBuilderTests.java b/server/src/test/java/org/elasticsearch/index/query/ToChildBlockJoinQueryBuilderTests.java
@@ -17,16 +17,16 @@
 
 import static org.hamcrest.CoreMatchers.instanceOf;
 
-public class ToChildBlockJoinQueryBuilderTests extends AbstractQueryTestCase< ToChildBlockJoinQueryBuilder> {
+public class ToChildBlockJoinQueryBuilderTests extends AbstractQueryTestCase<ToChildBlockJoinQueryBuilder> {
  @Override
  protected ToChildBlockJoinQueryBuilder doCreateTestQueryBuilder() {
  String filterFieldName = randomBoolean() ? KEYWORD_FIELD_NAME : TEXT_FIELD_NAME;
  return new ToChildBlockJoinQueryBuilder(QueryBuilders.termQuery(filterFieldName, randomAlphaOfLength(10)));
  }
 
  @Override
- protected void doAssertLuceneQuery(ToChildBlockJoinQueryBuilder queryBuilder, Query query,
-  SearchExecutionContext context) throws IOException {
+ protected void doAssertLuceneQuery(ToChildBlockJoinQueryBuilder queryBuilder, Query query, SearchExecutionContext context)
+ throws IOException {
  assertThat(query, instanceOf(ToChildBlockJoinQuery.class));
  }