Skip to content

Commit

Permalink
action review feedback: use doc fetcher more
Browse files Browse the repository at this point in the history
  • Loading branch information
cpoerschke committed Feb 3, 2025
1 parent aa5ac10 commit 88f5823
Show file tree
Hide file tree
Showing 4 changed files with 21 additions and 22 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ protected TestTag[] pullTagsFromResponse(SolrQueryRequest req, SolrQueryResponse
DocIterator iter = docList.iterator();
while (iter.hasNext()) {
int docId = iter.next();
Document doc = searcher.storedFields().document(docId);
Document doc = searcher.getDocFetcher().doc(docId);
String id = doc.getField("id").stringValue();
String name = lookupByName(doc.get("name"));
assertEquals("looking for " + name, NAMES.indexOf(name) + "", id);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -207,8 +207,8 @@ public void testOverwrite() throws IOException {
final TopDocs docs = searcher.search(join(one("cd")), 10);
assertEquals(2, docs.totalHits.value);
final String pAct =
searcher.storedFields().document(docs.scoreDocs[0].doc).get(parent)
+ searcher.storedFields().document(docs.scoreDocs[1].doc).get(parent);
searcher.getDocFetcher().doc(docs.scoreDocs[0].doc).get(parent)
+ searcher.getDocFetcher().doc(docs.scoreDocs[1].doc).get(parent);
assertTrue(pAct.contains(dubbed) && pAct.contains(overwritten) && pAct.length() == 2);

assertQ(req("id:66", "//*[@numFound='6']"));
Expand Down Expand Up @@ -1028,7 +1028,7 @@ protected void assertSingleParentOf(
throws IOException {
final TopDocs docs = searcher.search(join(childTerm), 10);
assertEquals(1, docs.totalHits.value);
final String pAct = searcher.storedFields().document(docs.scoreDocs[0].doc).get(parent);
final String pAct = searcher.getDocFetcher().doc(docs.scoreDocs[0].doc).get(parent);
assertEquals(parentExp, pAct);
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -283,7 +283,7 @@ private Document getDoc(String id) throws IOException {
TermQuery query = new TermQuery(new Term(ID, id));
TopDocs doc1 = searcher.search(query, 1);
ScoreDoc scoreDoc = doc1.scoreDocs[0];
return searcher.storedFields().document(scoreDoc.doc);
return searcher.getDocFetcher().doc(scoreDoc.doc);
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -121,8 +121,8 @@ public void testRescorer() throws Exception {
// first run the standard query
TopDocs hits = searcher.search(bqBuilder.build(), 10);
assertEquals(2, hits.totalHits.value);
assertEquals("0", searcher.storedFields().document(hits.scoreDocs[0].doc).get("id"));
assertEquals("1", searcher.storedFields().document(hits.scoreDocs[1].doc).get("id"));
assertEquals("0", searcher.getDocFetcher().doc(hits.scoreDocs[0].doc).get("id"));
assertEquals("1", searcher.getDocFetcher().doc(hits.scoreDocs[1].doc).get("id"));

final List<Feature> features = makeFieldValueFeatures(new int[] {0, 1, 2}, "finalScore");
final List<Normalizer> norms =
Expand All @@ -145,8 +145,8 @@ public void testRescorer() throws Exception {
hits = rescorer.rescore(searcher, hits, 2);

// rerank using the field finalScore
assertEquals("1", searcher.storedFields().document(hits.scoreDocs[0].doc).get("id"));
assertEquals("0", searcher.storedFields().document(hits.scoreDocs[1].doc).get("id"));
assertEquals("1", searcher.getDocFetcher().doc(hits.scoreDocs[0].doc).get("id"));
assertEquals("0", searcher.getDocFetcher().doc(hits.scoreDocs[1].doc).get("id"));
}
}

Expand All @@ -173,11 +173,11 @@ public void testDifferentTopN() throws IOException {
TopDocs hits = searcher.search(bqBuilder.build(), 10);
assertEquals(5, hits.totalHits.value);

assertEquals("0", searcher.storedFields().document(hits.scoreDocs[0].doc).get("id"));
assertEquals("1", searcher.storedFields().document(hits.scoreDocs[1].doc).get("id"));
assertEquals("2", searcher.storedFields().document(hits.scoreDocs[2].doc).get("id"));
assertEquals("3", searcher.storedFields().document(hits.scoreDocs[3].doc).get("id"));
assertEquals("4", searcher.storedFields().document(hits.scoreDocs[4].doc).get("id"));
assertEquals("0", searcher.getDocFetcher().doc(hits.scoreDocs[0].doc).get("id"));
assertEquals("1", searcher.getDocFetcher().doc(hits.scoreDocs[1].doc).get("id"));
assertEquals("2", searcher.getDocFetcher().doc(hits.scoreDocs[2].doc).get("id"));
assertEquals("3", searcher.getDocFetcher().doc(hits.scoreDocs[3].doc).get("id"));
assertEquals("4", searcher.getDocFetcher().doc(hits.scoreDocs[4].doc).get("id"));

final List<Feature> features = makeFieldValueFeatures(new int[] {0, 1, 2}, "finalScoreFloat");
final List<Normalizer> norms =
Expand All @@ -201,11 +201,11 @@ public void testDifferentTopN() throws IOException {

// rerank @ 0 should not change the order
hits = rescorer.rescore(searcher, hits, 0);
assertEquals("0", searcher.storedFields().document(hits.scoreDocs[0].doc).get("id"));
assertEquals("1", searcher.storedFields().document(hits.scoreDocs[1].doc).get("id"));
assertEquals("2", searcher.storedFields().document(hits.scoreDocs[2].doc).get("id"));
assertEquals("3", searcher.storedFields().document(hits.scoreDocs[3].doc).get("id"));
assertEquals("4", searcher.storedFields().document(hits.scoreDocs[4].doc).get("id"));
assertEquals("0", searcher.getDocFetcher().doc(hits.scoreDocs[0].doc).get("id"));
assertEquals("1", searcher.getDocFetcher().doc(hits.scoreDocs[1].doc).get("id"));
assertEquals("2", searcher.getDocFetcher().doc(hits.scoreDocs[2].doc).get("id"));
assertEquals("3", searcher.getDocFetcher().doc(hits.scoreDocs[3].doc).get("id"));
assertEquals("4", searcher.getDocFetcher().doc(hits.scoreDocs[4].doc).get("id"));

// test rerank with different topN cuts

Expand All @@ -221,12 +221,11 @@ public void testDifferentTopN() throws IOException {
if (log.isInfoEnabled()) {
log.info(
"doc {} in pos {}",
searcher.storedFields().document(hits.scoreDocs[j].doc).get("id"),
searcher.getDocFetcher().doc(hits.scoreDocs[j].doc).get("id"),
j);
}
assertEquals(
i,
Integer.parseInt(searcher.storedFields().document(hits.scoreDocs[j].doc).get("id")));
i, Integer.parseInt(searcher.getDocFetcher().doc(hits.scoreDocs[j].doc).get("id")));
assertEquals((i + 1) * features.size() * featureWeight, hits.scoreDocs[j].score, 0.00001);
}
}
Expand Down

0 comments on commit 88f5823

Please sign in to comment.