Skip to content

Commit 5ec0e36

Browse files
tarzanekvladak
authored andcommitted
fix context tests and highlighter
1 parent faba72a commit 5ec0e36

File tree

5 files changed

+28
-15
lines changed

5 files changed

+28
-15
lines changed

opengrok-indexer/src/main/java/org/opengrok/indexer/search/context/Context.java

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
import org.apache.lucene.index.IndexableField;
3838
import org.apache.lucene.search.IndexSearcher;
3939
import org.apache.lucene.search.Query;
40+
import org.apache.lucene.search.uhighlight.UnifiedHighlighter;
4041
import org.opengrok.indexer.analysis.AbstractAnalyzer;
4142
import org.opengrok.indexer.analysis.Definitions;
4243
import org.opengrok.indexer.analysis.Scopes;
@@ -196,9 +197,13 @@ public boolean getContext2(RuntimeEnvironment env, IndexSearcher searcher,
196197
formatter.setMoreUrl(moreURL);
197198
formatter.setMoreLimit(linelimit);
198199

199-
OGKUnifiedHighlighter uhi = new OGKUnifiedHighlighter(env, searcher, anz);
200-
uhi.setBreakIterator(StrictLineBreakIterator::new);
201-
uhi.setFormatter(formatter);
200+
UnifiedHighlighter.Builder uhBuilder = new UnifiedHighlighter.Builder(searcher, anz)
201+
// .withMaxLength(maxDocCharsToAnalyze)
202+
// .withHighlightPhrasesStrictly(true)
203+
// .withHandleMultiTermQuery(true)
204+
.withBreakIterator(StrictLineBreakIterator::new)
205+
.withFormatter(formatter);
206+
OGKUnifiedHighlighter uhi = new OGKUnifiedHighlighter(env, uhBuilder);
202207
uhi.setTabSize(tabSize);
203208

204209
try {

opengrok-indexer/src/main/java/org/opengrok/indexer/search/context/OGKUnifiedHighlighter.java

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@
3535
import org.apache.lucene.document.DateTools;
3636
import org.apache.lucene.document.Document;
3737
import org.apache.lucene.search.DocIdSetIterator;
38-
import org.apache.lucene.search.IndexSearcher;
3938
import org.apache.lucene.search.Query;
4039
import org.apache.lucene.search.uhighlight.UHComponents;
4140
import org.apache.lucene.search.uhighlight.UnifiedHighlighter;
@@ -68,12 +67,11 @@ public class OGKUnifiedHighlighter extends UnifiedHighlighter {
6867
* for the specified {@code indexSearcher} and {@code indexAnalyzer}, and
6968
* stores the {@code env} for later use.
7069
* @param env a required instance
71-
* @param indexSearcher a required instance
72-
* @param indexAnalyzer a required instance
7370
* @throws IllegalArgumentException if any argument is null
7471
*/
75-
public OGKUnifiedHighlighter(RuntimeEnvironment env, IndexSearcher indexSearcher, Analyzer indexAnalyzer) {
76-
super(indexSearcher, indexAnalyzer);
72+
public OGKUnifiedHighlighter(RuntimeEnvironment env,
73+
UnifiedHighlighter.Builder uhBuilder) {
74+
super(uhBuilder);
7775

7876
if (env == null) {
7977
throw new IllegalArgumentException("env is null");

opengrok-indexer/src/test/java/org/opengrok/indexer/search/context/SearchAndContextFormatterTest.java

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
import org.apache.lucene.document.Document;
3333
import org.apache.lucene.search.ScoreDoc;
3434

35+
import org.apache.lucene.search.uhighlight.UnifiedHighlighter;
3536
import org.junit.jupiter.api.AfterAll;
3637
import org.junit.jupiter.api.BeforeAll;
3738
import org.junit.jupiter.api.Test;
@@ -124,9 +125,13 @@ private String[] getFirstFragments(SearchEngine instance) throws IOException {
124125
AbstractAnalyzer anz = fac.getAnalyzer();
125126

126127
ContextFormatter formatter = new ContextFormatter(args);
127-
OGKUnifiedHighlighter uhi = new OGKUnifiedHighlighter(env, instance.getSearcher(), anz);
128-
uhi.setBreakIterator(StrictLineBreakIterator::new);
129-
uhi.setFormatter(formatter);
128+
UnifiedHighlighter.Builder uhBuilder = new UnifiedHighlighter.Builder(instance.getSearcher(), anz)
129+
// .withMaxLength(maxDocCharsToAnalyze)
130+
// .withHighlightPhrasesStrictly(true)
131+
// .withHandleMultiTermQuery(true)
132+
.withBreakIterator(StrictLineBreakIterator::new)
133+
.withFormatter(formatter);
134+
OGKUnifiedHighlighter uhi = new OGKUnifiedHighlighter(env, uhBuilder);
130135

131136
ScoreDoc[] docs = instance.scoreDocs();
132137
for (ScoreDoc scoreDoc : docs) {

opengrok-indexer/src/test/java/org/opengrok/indexer/search/context/SearchAndContextFormatterTest2.java

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
import org.apache.lucene.search.ScoreDoc;
4040
import org.apache.lucene.search.highlight.InvalidTokenOffsetsException;
4141

42+
import org.apache.lucene.search.uhighlight.UnifiedHighlighter;
4243
import org.junit.jupiter.api.AfterAll;
4344
import org.junit.jupiter.api.BeforeAll;
4445
import org.junit.jupiter.api.Test;
@@ -197,10 +198,14 @@ private String[] getFirstFragments(SearchEngine instance)
197198
AbstractAnalyzer anz = fac.getAnalyzer();
198199

199200
ContextFormatter formatter = new ContextFormatter(args);
201+
UnifiedHighlighter.Builder uhBuilder = new UnifiedHighlighter.Builder(instance.getSearcher(), anz)
202+
// .withMaxLength(maxDocCharsToAnalyze)
203+
// .withHighlightPhrasesStrictly(true)
204+
// .withHandleMultiTermQuery(true)
205+
.withBreakIterator(StrictLineBreakIterator::new)
206+
.withFormatter(formatter);
200207
OGKUnifiedHighlighter uhi = new OGKUnifiedHighlighter(env,
201-
instance.getSearcher(), anz);
202-
uhi.setBreakIterator(StrictLineBreakIterator::new);
203-
uhi.setFormatter(formatter);
208+
uhBuilder);
204209
uhi.setTabSize(TABSIZE);
205210

206211
ScoreDoc[] docs = instance.scoreDocs();

pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ Portions Copyright (c) 2018, 2020, Chris Fraire <[email protected]>.
6060
</scm>
6161

6262
<properties>
63-
<lucene.version>9.0.0</lucene.version>
63+
<lucene.version>9.1.0</lucene.version>
6464
<mavenjavadocplugin.version>3.2.0</mavenjavadocplugin.version>
6565
<!-- The following changed syntax from e.g. 1.8 to 11. -->
6666
<compileSource>11</compileSource>

0 commit comments

Comments
 (0)