LUCENE-7983: IndexWriter.IndexReaderWarmer is now a functional interface instead of an abstract class with a single method.

This commit is contained in:
Dawid Weiss 2017-10-04 10:54:13 +02:00
parent b09c99f516
commit 46cd679e91
7 changed files with 64 additions and 82 deletions

View File

@ -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)

View File

@ -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 {

View File

@ -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;
/** /**

View File

@ -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:

View File

@ -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();

View File

@ -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;

View File

@ -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) {