package org.apache.lucene.index; /** * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You under the Apache License, Version 2.0 * (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ import java.io.IOException; import org.apache.lucene.util.LuceneTestCase; import org.apache.lucene.analysis.MockAnalyzer; import org.apache.lucene.document.Document; import org.apache.lucene.document.Field.Index; import org.apache.lucene.document.Field.Store; import org.apache.lucene.store.Directory; public class TestParallelTermEnum extends LuceneTestCase { private IndexReader ir1; private IndexReader ir2; private Directory rd1; private Directory rd2; @Override public void setUp() throws Exception { super.setUp(); Document doc; rd1 = newDirectory(); IndexWriter iw1 = new IndexWriter(rd1, newIndexWriterConfig( TEST_VERSION_CURRENT, new MockAnalyzer(random))); doc = new Document(); doc.add(newField("field1", "the quick brown fox jumps", Store.YES, Index.ANALYZED)); doc.add(newField("field2", "the quick brown fox jumps", Store.YES, Index.ANALYZED)); doc.add(newField("field4", "", Store.NO, Index.ANALYZED)); iw1.addDocument(doc); iw1.close(); rd2 = newDirectory(); IndexWriter iw2 = new IndexWriter(rd2, newIndexWriterConfig( TEST_VERSION_CURRENT, new MockAnalyzer(random))); doc = new Document(); doc.add(newField("field0", "", Store.NO, Index.ANALYZED)); doc.add(newField("field1", "the fox jumps over the lazy dog", Store.YES, Index.ANALYZED)); doc.add(newField("field3", "the fox jumps over the lazy dog", Store.YES, Index.ANALYZED)); iw2.addDocument(doc); iw2.close(); this.ir1 = IndexReader.open(rd1, true); this.ir2 = IndexReader.open(rd2, true); } @Override public void tearDown() throws Exception { ir1.close(); ir2.close(); rd1.close(); rd2.close(); super.tearDown(); } public void test1() throws IOException { ParallelReader pr = new ParallelReader(); pr.add(ir1); pr.add(ir2); TermDocs td = pr.termDocs(); TermEnum te = pr.terms(); assertTrue(te.next()); assertEquals("field1:brown", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field1:fox", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field1:jumps", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field1:quick", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field1:the", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field2:brown", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field2:fox", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field2:jumps", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field2:quick", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field2:the", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field3:dog", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field3:fox", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field3:jumps", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field3:lazy", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field3:over", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertTrue(te.next()); assertEquals("field3:the", te.term().toString()); td.seek(te.term()); assertTrue(td.next()); assertEquals(0, td.doc()); assertFalse(td.next()); assertFalse(te.next()); } }