mirror of https://github.com/apache/lucene.git
LUCENE-7983: IndexWriter.IndexReaderWarmer is now a functional interface instead of an abstract class with a single method.
This commit is contained in:
parent
b09c99f516
commit
46cd679e91
|
@ -106,6 +106,10 @@ Other
|
||||||
* LUCENE-7978: Add some more documentation about setting up build
|
* LUCENE-7978: Add some more documentation about setting up build
|
||||||
environment. (Anton R. Yuste via Uwe Schindler)
|
environment. (Anton R. Yuste via Uwe Schindler)
|
||||||
|
|
||||||
|
* LUCENE-7983: IndexWriter.IndexReaderWarmer is now a functional interface
|
||||||
|
instead of an abstract class with a single method (Dawid Weiss)
|
||||||
|
|
||||||
|
|
||||||
======================= Lucene 7.0.1 =======================
|
======================= Lucene 7.0.1 =======================
|
||||||
(No Changes)
|
(No Changes)
|
||||||
|
|
||||||
|
|
|
@ -4804,19 +4804,16 @@ public class IndexWriter implements Closeable, TwoPhaseCommit, Accountable {
|
||||||
*
|
*
|
||||||
* @lucene.experimental
|
* @lucene.experimental
|
||||||
*
|
*
|
||||||
* <p><b>NOTE</b>: warm is called before any deletes have
|
* <p><b>NOTE</b>: {@link #warm(LeafReader)} is called before any
|
||||||
* been carried over to the merged segment. */
|
* deletes have been carried over to the merged segment. */
|
||||||
public static abstract class IndexReaderWarmer {
|
@FunctionalInterface
|
||||||
|
public static interface IndexReaderWarmer {
|
||||||
/** Sole constructor. (For invocation by subclass
|
/**
|
||||||
* constructors, typically implicit.) */
|
* Invoked on the {@link LeafReader} for the newly
|
||||||
protected IndexReaderWarmer() {
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Invoked on the {@link LeafReader} for the newly
|
|
||||||
* merged segment, before that segment is made visible
|
* merged segment, before that segment is made visible
|
||||||
* to near-real-time readers. */
|
* to near-real-time readers.
|
||||||
public abstract void warm(LeafReader reader) throws IOException;
|
*/
|
||||||
|
void warm(LeafReader reader) throws IOException;
|
||||||
}
|
}
|
||||||
|
|
||||||
void tragicEvent(Throwable tragedy, String location) throws IOException {
|
void tragicEvent(Throwable tragedy, String location) throws IOException {
|
||||||
|
|
|
@ -26,7 +26,7 @@ import org.apache.lucene.util.InfoStream;
|
||||||
* A very simple merged segment warmer that just ensures
|
* A very simple merged segment warmer that just ensures
|
||||||
* data structures are initialized.
|
* data structures are initialized.
|
||||||
*/
|
*/
|
||||||
public class SimpleMergedSegmentWarmer extends IndexReaderWarmer {
|
public class SimpleMergedSegmentWarmer implements IndexReaderWarmer {
|
||||||
private final InfoStream infoStream;
|
private final InfoStream infoStream;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -129,16 +129,13 @@ public class TestDemoParallelLeafReader extends LuceneTestCase {
|
||||||
}
|
}
|
||||||
w = new IndexWriter(indexDir, iwc);
|
w = new IndexWriter(indexDir, iwc);
|
||||||
|
|
||||||
w.getConfig().setMergedSegmentWarmer(new IndexWriter.IndexReaderWarmer() {
|
w.getConfig().setMergedSegmentWarmer((reader) -> {
|
||||||
@Override
|
|
||||||
public void warm(LeafReader reader) throws IOException {
|
|
||||||
// This will build the parallel index for the merged segment before the merge becomes visible, so reopen delay is only due to
|
// This will build the parallel index for the merged segment before the merge becomes visible, so reopen delay is only due to
|
||||||
// newly flushed segments:
|
// newly flushed segments:
|
||||||
if (DEBUG) System.out.println(Thread.currentThread().getName() +": TEST: now warm " + reader);
|
if (DEBUG) System.out.println(Thread.currentThread().getName() +": TEST: now warm " + reader);
|
||||||
// TODO: it's not great that we pass false here; it means we close the reader & reopen again for NRT reader; still we did "warm" by
|
// TODO: it's not great that we pass false here; it means we close the reader & reopen again for NRT reader; still we did "warm" by
|
||||||
// building the parallel index, if necessary
|
// building the parallel index, if necessary
|
||||||
getParallelLeafReader(reader, false, getCurrentSchemaGen());
|
getParallelLeafReader(reader, false, getCurrentSchemaGen());
|
||||||
}
|
|
||||||
});
|
});
|
||||||
|
|
||||||
// start with empty commit:
|
// start with empty commit:
|
||||||
|
|
|
@ -612,24 +612,15 @@ public class TestIndexWriterReader extends LuceneTestCase {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private static class MyWarmer extends IndexWriter.IndexReaderWarmer {
|
|
||||||
int warmCount;
|
|
||||||
@Override
|
|
||||||
public void warm(LeafReader reader) throws IOException {
|
|
||||||
warmCount++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public void testMergeWarmer() throws Exception {
|
public void testMergeWarmer() throws Exception {
|
||||||
|
|
||||||
Directory dir1 = getAssertNoDeletesDirectory(newDirectory());
|
Directory dir1 = getAssertNoDeletesDirectory(newDirectory());
|
||||||
// Enroll warmer
|
// Enroll warmer
|
||||||
MyWarmer warmer = new MyWarmer();
|
AtomicInteger warmCount = new AtomicInteger();
|
||||||
IndexWriter writer = new IndexWriter(
|
IndexWriter writer = new IndexWriter(
|
||||||
dir1,
|
dir1,
|
||||||
newIndexWriterConfig(new MockAnalyzer(random()))
|
newIndexWriterConfig(new MockAnalyzer(random()))
|
||||||
.setMaxBufferedDocs(2)
|
.setMaxBufferedDocs(2)
|
||||||
.setMergedSegmentWarmer(warmer)
|
.setMergedSegmentWarmer((leafReader) -> warmCount.incrementAndGet())
|
||||||
.setMergeScheduler(new ConcurrentMergeScheduler())
|
.setMergeScheduler(new ConcurrentMergeScheduler())
|
||||||
.setMergePolicy(newLogMergePolicy())
|
.setMergePolicy(newLogMergePolicy())
|
||||||
);
|
);
|
||||||
|
@ -648,12 +639,12 @@ public class TestIndexWriterReader extends LuceneTestCase {
|
||||||
}
|
}
|
||||||
((ConcurrentMergeScheduler) writer.getConfig().getMergeScheduler()).sync();
|
((ConcurrentMergeScheduler) writer.getConfig().getMergeScheduler()).sync();
|
||||||
|
|
||||||
assertTrue(warmer.warmCount > 0);
|
assertTrue(warmCount.get() > 0);
|
||||||
final int count = warmer.warmCount;
|
final int count = warmCount.get();
|
||||||
|
|
||||||
writer.addDocument(DocHelper.createDocument(17, "test", 4));
|
writer.addDocument(DocHelper.createDocument(17, "test", 4));
|
||||||
writer.forceMerge(1);
|
writer.forceMerge(1);
|
||||||
assertTrue(warmer.warmCount > count);
|
assertTrue(warmCount.get() > count);
|
||||||
|
|
||||||
writer.close();
|
writer.close();
|
||||||
r1.close();
|
r1.close();
|
||||||
|
@ -969,19 +960,16 @@ public class TestIndexWriterReader extends LuceneTestCase {
|
||||||
final AtomicBoolean didWarm = new AtomicBoolean();
|
final AtomicBoolean didWarm = new AtomicBoolean();
|
||||||
IndexWriter w = new IndexWriter(
|
IndexWriter w = new IndexWriter(
|
||||||
dir,
|
dir,
|
||||||
newIndexWriterConfig(new MockAnalyzer(random())).
|
newIndexWriterConfig(new MockAnalyzer(random()))
|
||||||
setMaxBufferedDocs(2).
|
.setMaxBufferedDocs(2)
|
||||||
setReaderPooling(true).
|
.setReaderPooling(true)
|
||||||
setMergedSegmentWarmer(new IndexWriter.IndexReaderWarmer() {
|
.setMergedSegmentWarmer((r) -> {
|
||||||
@Override
|
|
||||||
public void warm(LeafReader r) throws IOException {
|
|
||||||
IndexSearcher s = newSearcher(r);
|
IndexSearcher s = newSearcher(r);
|
||||||
TopDocs hits = s.search(new TermQuery(new Term("foo", "bar")), 10);
|
TopDocs hits = s.search(new TermQuery(new Term("foo", "bar")), 10);
|
||||||
assertEquals(20, hits.totalHits);
|
assertEquals(20, hits.totalHits);
|
||||||
didWarm.set(true);
|
didWarm.set(true);
|
||||||
}
|
})
|
||||||
}).
|
.setMergePolicy(newLogMergePolicy(10))
|
||||||
setMergePolicy(newLogMergePolicy(10))
|
|
||||||
);
|
);
|
||||||
|
|
||||||
Document doc = new Document();
|
Document doc = new Document();
|
||||||
|
|
|
@ -34,7 +34,7 @@ import org.apache.lucene.index.SegmentReader;
|
||||||
|
|
||||||
// TODO: or ... replica node can do merging locally? tricky to keep things in sync, when one node merges more slowly than others...
|
// TODO: or ... replica node can do merging locally? tricky to keep things in sync, when one node merges more slowly than others...
|
||||||
|
|
||||||
class PreCopyMergedSegmentWarmer extends IndexReaderWarmer {
|
class PreCopyMergedSegmentWarmer implements IndexReaderWarmer {
|
||||||
|
|
||||||
private final PrimaryNode primary;
|
private final PrimaryNode primary;
|
||||||
|
|
||||||
|
|
|
@ -16,7 +16,6 @@
|
||||||
*/
|
*/
|
||||||
package org.apache.lucene.index;
|
package org.apache.lucene.index;
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.nio.file.Path;
|
import java.nio.file.Path;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
@ -468,9 +467,7 @@ public abstract class ThreadedIndexingAndSearchingTestCase extends LuceneTestCas
|
||||||
mp.setNoCFSRatio(Math.max(0.25d, mp.getNoCFSRatio()));
|
mp.setNoCFSRatio(Math.max(0.25d, mp.getNoCFSRatio()));
|
||||||
}
|
}
|
||||||
|
|
||||||
conf.setMergedSegmentWarmer(new IndexWriter.IndexReaderWarmer() {
|
conf.setMergedSegmentWarmer((reader) -> {
|
||||||
@Override
|
|
||||||
public void warm(LeafReader reader) throws IOException {
|
|
||||||
if (VERBOSE) {
|
if (VERBOSE) {
|
||||||
System.out.println("TEST: now warm merged reader=" + reader);
|
System.out.println("TEST: now warm merged reader=" + reader);
|
||||||
}
|
}
|
||||||
|
@ -492,7 +489,6 @@ public abstract class ThreadedIndexingAndSearchingTestCase extends LuceneTestCas
|
||||||
if (VERBOSE) {
|
if (VERBOSE) {
|
||||||
System.out.println("TEST: warm visited " + sum + " fields");
|
System.out.println("TEST: warm visited " + sum + " fields");
|
||||||
}
|
}
|
||||||
}
|
|
||||||
});
|
});
|
||||||
|
|
||||||
if (VERBOSE) {
|
if (VERBOSE) {
|
||||||
|
|
Loading…
Reference in New Issue