Skip to content

Commit 6b2cc59

Browse files
javannajavanna
authored
Replace some test usages of search(Query, Collector) (#113818)
The leftover usages of the deprecated search(Query, TotalHitCountCollector) have been replaced with search(Query, TotalHitCountCollectorManager)
1 parent 4ef5ea6 commit 6b2cc59

File tree

4 files changed

+42
-48
lines changed

4 files changed

+42
-48
lines changed

‎server/src/test/java/org/elasticsearch/index/engine/InternalEngineTests.java‎

Lines changed: 26 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@
5555
import org.apache.lucene.search.SortedSetSortField;
5656
import org.apache.lucene.search.TermQuery;
5757
import org.apache.lucene.search.TopDocs;
58-
import org.apache.lucene.search.TotalHitCountCollector;
58+
import org.apache.lucene.search.TotalHitCountCollectorManager;
5959
import org.apache.lucene.store.AlreadyClosedException;
6060
import org.apache.lucene.store.Directory;
6161
import org.apache.lucene.store.Lock;
@@ -640,9 +640,8 @@ public void testTranslogMultipleOperationsSameDocument() throws IOException {
640640
recoverFromTranslog(recoveringEngine, translogHandler, Long.MAX_VALUE);
641641
recoveringEngine.refresh("test");
642642
try (Engine.Searcher searcher = recoveringEngine.acquireSearcher("test")) {
643-
final TotalHitCountCollector collector = new TotalHitCountCollector();
644-
searcher.search(new MatchAllDocsQuery(), collector);
645-
assertThat(collector.getTotalHits(), equalTo(operations.get(operations.size() - 1) instanceof Engine.Delete ? 0 : 1));
643+
Integer totalHits = searcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager());
644+
assertThat(totalHits, equalTo(operations.get(operations.size() - 1) instanceof Engine.Delete ? 0 : 1));
646645
}
647646
}
648647
}
@@ -2009,16 +2008,20 @@ public void testConcurrentOutOfOrderDocsOnReplica() throws IOException, Interrup
20092008

20102009
if (lastFieldValueDoc1 != null) {
20112010
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
2012-
final TotalHitCountCollector collector = new TotalHitCountCollector();
2013-
searcher.search(new TermQuery(new Term("value", lastFieldValueDoc1)), collector);
2014-
assertThat(collector.getTotalHits(), equalTo(1));
2011+
Integer totalHits = searcher.search(
2012+
new TermQuery(new Term("value", lastFieldValueDoc1)),
2013+
new TotalHitCountCollectorManager()
2014+
);
2015+
assertThat(totalHits, equalTo(1));
20152016
}
20162017
}
20172018
if (lastFieldValueDoc2 != null) {
20182019
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
2019-
final TotalHitCountCollector collector = new TotalHitCountCollector();
2020-
searcher.search(new TermQuery(new Term("value", lastFieldValueDoc2)), collector);
2021-
assertThat(collector.getTotalHits(), equalTo(1));
2020+
Integer totalHits = searcher.search(
2021+
new TermQuery(new Term("value", lastFieldValueDoc2)),
2022+
new TotalHitCountCollectorManager()
2023+
);
2024+
assertThat(totalHits, equalTo(1));
20222025
}
20232026
}
20242027

@@ -2244,9 +2247,11 @@ private int assertOpsOnPrimary(List<Engine.Operation> ops, long currentOpVersion
22442247
// first op and it failed.
22452248
if (docDeleted == false && lastFieldValue != null) {
22462249
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
2247-
final TotalHitCountCollector collector = new TotalHitCountCollector();
2248-
searcher.search(new TermQuery(new Term("value", lastFieldValue)), collector);
2249-
assertThat(collector.getTotalHits(), equalTo(1));
2250+
Integer totalHits = searcher.search(
2251+
new TermQuery(new Term("value", lastFieldValue)),
2252+
new TotalHitCountCollectorManager()
2253+
);
2254+
assertThat(totalHits, equalTo(1));
22502255
}
22512256
}
22522257
}
@@ -2270,9 +2275,8 @@ private int assertOpsOnPrimary(List<Engine.Operation> ops, long currentOpVersion
22702275
assertVisibleCount(engine, docDeleted ? 0 : 1);
22712276
if (docDeleted == false) {
22722277
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
2273-
final TotalHitCountCollector collector = new TotalHitCountCollector();
2274-
searcher.search(new TermQuery(new Term("value", lastFieldValue)), collector);
2275-
assertThat(collector.getTotalHits(), equalTo(1));
2278+
Integer totalHits = searcher.search(new TermQuery(new Term("value", lastFieldValue)), new TotalHitCountCollectorManager());
2279+
assertThat(totalHits, equalTo(1));
22762280
}
22772281
}
22782282
return opsPerformed;
@@ -2357,9 +2361,8 @@ public void testNonInternalVersioningOnPrimary() throws IOException {
23572361
if (docDeleted == false) {
23582362
logger.info("searching for [{}]", lastFieldValue);
23592363
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
2360-
final TotalHitCountCollector collector = new TotalHitCountCollector();
2361-
searcher.search(new TermQuery(new Term("value", lastFieldValue)), collector);
2362-
assertThat(collector.getTotalHits(), equalTo(1));
2364+
Integer totalHits = searcher.search(new TermQuery(new Term("value", lastFieldValue)), new TotalHitCountCollectorManager());
2365+
assertThat(totalHits, equalTo(1));
23632366
}
23642367
}
23652368
}
@@ -2375,9 +2378,8 @@ public void testVersioningPromotedReplica() throws IOException {
23752378
final int opsOnPrimary = assertOpsOnPrimary(primaryOps, finalReplicaVersion, deletedOnReplica, replicaEngine);
23762379
final long currentSeqNo = getSequenceID(replicaEngine, new Engine.Get(false, false, Term.toString(lastReplicaOp.uid()))).v1();
23772380
try (Engine.Searcher searcher = engine.acquireSearcher("test", Engine.SearcherScope.INTERNAL)) {
2378-
final TotalHitCountCollector collector = new TotalHitCountCollector();
2379-
searcher.search(new MatchAllDocsQuery(), collector);
2380-
if (collector.getTotalHits() > 0) {
2381+
Integer totalHits = searcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager());
2382+
if (totalHits > 0) {
23812383
// last op wasn't delete
23822384
assertThat(currentSeqNo, equalTo(finalReplicaSeqNo + opsOnPrimary));
23832385
}
@@ -2400,9 +2402,8 @@ public void testConcurrentExternalVersioningOnPrimary() throws IOException, Inte
24002402
assertVisibleCount(engine, lastFieldValue == null ? 0 : 1);
24012403
if (lastFieldValue != null) {
24022404
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
2403-
final TotalHitCountCollector collector = new TotalHitCountCollector();
2404-
searcher.search(new TermQuery(new Term("value", lastFieldValue)), collector);
2405-
assertThat(collector.getTotalHits(), equalTo(1));
2405+
Integer totalHits = searcher.search(new TermQuery(new Term("value", lastFieldValue)), new TotalHitCountCollectorManager());
2406+
assertThat(totalHits, equalTo(1));
24062407
}
24072408
}
24082409
}

‎server/src/test/java/org/elasticsearch/search/SearchCancellationTests.java‎

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
import org.apache.lucene.index.TermsEnum;
2323
import org.apache.lucene.search.IndexSearcher;
2424
import org.apache.lucene.search.MatchAllDocsQuery;
25-
import org.apache.lucene.search.TotalHitCountCollector;
25+
import org.apache.lucene.search.TotalHitCountCollectorManager;
2626
import org.apache.lucene.store.Directory;
2727
import org.apache.lucene.tests.index.RandomIndexWriter;
2828
import org.apache.lucene.tests.util.TestUtil;
@@ -96,7 +96,6 @@ public void testAddingCancellationActions() throws IOException {
9696
}
9797

9898
public void testCancellableCollector() throws IOException {
99-
TotalHitCountCollector collector1 = new TotalHitCountCollector();
10099
Runnable cancellation = () -> { throw new TaskCancelledException("cancelled"); };
101100
ContextIndexSearcher searcher = new ContextIndexSearcher(
102101
reader,
@@ -106,16 +105,15 @@ public void testCancellableCollector() throws IOException {
106105
true
107106
);
108107

109-
searcher.search(new MatchAllDocsQuery(), collector1);
110-
assertThat(collector1.getTotalHits(), equalTo(reader.numDocs()));
108+
Integer totalHits = searcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager());
109+
assertThat(totalHits, equalTo(reader.numDocs()));
111110

112111
searcher.addQueryCancellation(cancellation);
113-
expectThrows(TaskCancelledException.class, () -> searcher.search(new MatchAllDocsQuery(), collector1));
112+
expectThrows(TaskCancelledException.class, () -> searcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager()));
114113

115114
searcher.removeQueryCancellation(cancellation);
116-
TotalHitCountCollector collector2 = new TotalHitCountCollector();
117-
searcher.search(new MatchAllDocsQuery(), collector2);
118-
assertThat(collector2.getTotalHits(), equalTo(reader.numDocs()));
115+
Integer totalHits2 = searcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager());
116+
assertThat(totalHits2, equalTo(reader.numDocs()));
119117
}
120118

121119
public void testExitableDirectoryReader() throws IOException {

‎test/framework/src/main/java/org/elasticsearch/index/engine/EngineTestCase.java‎

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
import org.apache.lucene.search.Scorer;
4343
import org.apache.lucene.search.Sort;
4444
import org.apache.lucene.search.TermQuery;
45-
import org.apache.lucene.search.TotalHitCountCollector;
45+
import org.apache.lucene.search.TotalHitCountCollectorManager;
4646
import org.apache.lucene.search.Weight;
4747
import org.apache.lucene.store.AlreadyClosedException;
4848
import org.apache.lucene.store.Directory;
@@ -178,9 +178,8 @@ protected static void assertVisibleCount(Engine engine, int numDocs, boolean ref
178178
engine.refresh("test");
179179
}
180180
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
181-
final TotalHitCountCollector collector = new TotalHitCountCollector();
182-
searcher.search(new MatchAllDocsQuery(), collector);
183-
assertThat(collector.getTotalHits(), equalTo(numDocs));
181+
Integer totalHits = searcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager());
182+
assertThat(totalHits, equalTo(numDocs));
184183
}
185184
}
186185

@@ -971,9 +970,8 @@ protected static void assertVisibleCount(InternalEngine engine, int numDocs, boo
971970
engine.refresh("test");
972971
}
973972
try (Engine.Searcher searcher = engine.acquireSearcher("test")) {
974-
final TotalHitCountCollector collector = new TotalHitCountCollector();
975-
searcher.search(new MatchAllDocsQuery(), collector);
976-
assertThat(collector.getTotalHits(), equalTo(numDocs));
973+
Integer totalHits = searcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager());
974+
assertThat(totalHits, equalTo(numDocs));
977975
}
978976
}
979977

@@ -1170,9 +1168,8 @@ public static void assertOpsOnReplica(
11701168
assertVisibleCount(replicaEngine, lastFieldValue == null ? 0 : 1);
11711169
if (lastFieldValue != null) {
11721170
try (Engine.Searcher searcher = replicaEngine.acquireSearcher("test")) {
1173-
final TotalHitCountCollector collector = new TotalHitCountCollector();
1174-
searcher.search(new TermQuery(new Term("value", lastFieldValue)), collector);
1175-
assertThat(collector.getTotalHits(), equalTo(1));
1171+
Integer totalHits = searcher.search(new TermQuery(new Term("value", lastFieldValue)), new TotalHitCountCollectorManager());
1172+
assertThat(totalHits, equalTo(1));
11761173
}
11771174
}
11781175
}

‎x-pack/plugin/core/src/test/java/org/elasticsearch/xpack/core/security/authz/accesscontrol/SecurityIndexReaderWrapperIntegrationTests.java‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import org.apache.lucene.search.MatchAllDocsQuery;
2121
import org.apache.lucene.search.ScoreDoc;
2222
import org.apache.lucene.search.TermQuery;
23-
import org.apache.lucene.search.TotalHitCountCollector;
23+
import org.apache.lucene.search.TotalHitCountCollectorManager;
2424
import org.apache.lucene.store.Directory;
2525
import org.elasticsearch.client.internal.Client;
2626
import org.elasticsearch.common.bytes.BytesArray;
@@ -193,10 +193,8 @@ protected IndicesAccessControl getIndicesAccessControl() {
193193
int expectedHitCount = valuesHitCount[i];
194194
logger.info("Going to verify hit count with query [{}] with expected total hits [{}]", parsedQuery.query(), expectedHitCount);
195195

196-
TotalHitCountCollector countCollector = new TotalHitCountCollector();
197-
indexSearcher.search(new MatchAllDocsQuery(), countCollector);
198-
199-
assertThat(countCollector.getTotalHits(), equalTo(expectedHitCount));
196+
Integer totalHits = indexSearcher.search(new MatchAllDocsQuery(), new TotalHitCountCollectorManager());
197+
assertThat(totalHits, equalTo(expectedHitCount));
200198
assertThat(wrappedDirectoryReader.numDocs(), equalTo(expectedHitCount));
201199
}
202200

0 commit comments

Comments
 (0)