Lucene3.1 メモその2 – ドキュメントを追加する

基本その２，ドキュメントの追加

以下のことをやります
・メモリ上(RAMDirectory)にインデックスを格納
・英語ドキュメントを1つ追加
・検索すると１件ヒットすることを確認
・もう1つドキュメントを追加
・検索すると１件しかヒットしないことを確認
・IndexReaderをリオープンし、再度検索し、ドキュメントが2件ヒットすることを確認

つまり、IndexWriter.addDocument()/commit()してもすぐには検索可能になりません。IndexReaderを作り直すか、リオープンしてIndexSearcherを作り直すことで検索可能になります。

    @Test
    public void index() throws Exception {
        RAMDirectory directory = new RAMDirectory();
        Analyzer analyzer = new StandardAnalyzer(Version.LUCENE_31);
        IndexWriterConfig iwc = new IndexWriterConfig(Version.LUCENE_31, analyzer);
        IndexWriter writer = new IndexWriter(directory, iwc);

        Document doc = new Document();
        doc.add(new Field("str_field", "quick brown fox jumped over the lazy dog.",
                Field.Store.YES, Field.Index.ANALYZED));
        writer.addDocument(doc);
        writer.commit();
        IndexReader reader = IndexReader.open(writer, true);
        IndexSearcher searcher = new IndexSearcher(reader);
        QueryParser parser = new QueryParser(Version.LUCENE_31, "str_field", analyzer);
        TopDocs td = searcher.search(parser.parse("fox"), 1000);
        assertThat(td.totalHits, is(1));

        Document doc2 = new Document();
        doc2.add(new Field("str_field", "quick brown dog jumped over the lazy fox.",
                Field.Store.YES, Field.Index.ANALYZED));
        writer.addDocument(doc2);
        writer.commit();

        td = searcher.search(parser.parse("fox"), 1000);
        assertThat(td.totalHits, is(1));

        searcher.close();
        reader = reader.reopen();
        searcher = new IndexSearcher(reader);

        td = searcher.search(parser.parse("fox"), 1000);
        assertThat(td.totalHits, is(2));

        writer.close();
        searcher.close();
        directory.close();
    }

すぐ試せるmavenベースのソースコードはここに置いてあります。
Lucene3.1、JUnit4.8を使っています。

yusuke.blog

プログラムと、ゲームと、諸々と

Lucene3.1 メモその2 – ドキュメントを追加する