import org.apache.lucene.analysis.standard.StandardAnalyzer;

import org.apache.lucene.document.*;

import org.apache.lucene.index.*;

import org.apache.lucene.search.IndexSearcher;

import org.apache.lucene.search.ScoreDoc;

import org.apache.lucene.search.TermQuery;

import org.apache.lucene.search.TopDocs;

import org.apache.lucene.store.FSDirectory;

import org.junit.Test;

import org.omg.CORBA.PUBLIC_MEMBER;

import java.io.File;

import java.io.IOException;

import java.nio.file.Path;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

    /**

     * 修改

     *

     * @throws IOException

     */

    @Test

    public void updateDoc() throws IOException {

        //        1 创建索引库对象，指定索引库的位置

        //1.1 创建索引库位置

        Path path = new File("D:\\lucene").toPath();

        //1.2 创建索引库对象，关联索引库位置

        FSDirectory directory = FSDirectory.open(path);

//        2 创建IndexWriterConfig对象并指定分词器对象

        //2.1 创建分词器对象用于指定分词规则

        StandardAnalyzer standardAnalyzer = new StandardAnalyzer();//标准分词器，分词规则：单字分词

        //2.2 创建写出器配置对象，关联分词器对象

        IndexWriterConfig indexWriterConfig = new IndexWriterConfig(standardAnalyzer);

//        3 创建一个IndexWriter对象 &指定索引库的位置&指定一个IndexWriterConfig对象。

        IndexWriter indexWriter = new IndexWriter(directory, indexWriterConfig);

        // 4 构建修改的doc文档数据.

        Document document = new Document();

//        5 创建field对象，将field添加到document对象中。

        document.add(new StringField("docId", "docIdOne", Field.Store.YES));

        document.add(new TextField("title", "修改的是文档2", Field.Store.YES));

        document.add(new TextField("content", "我的祖国是一个伟大的国家", Field.Store.YES));

        document.add(new StringField("score", "100", Field.Store.YES));

        document.add(new StoredField("name", "张三"));

        //6 执行修改(删除一些或者一个，再新增一个)，按照docId分词查询文档，查到对应文档就修改，查不到就新增

        indexWriter.updateDocument(new Term("docId", "docIdTwo"), document);

        //按照title分词查询文档，查到对应文档就修改，查不到就新增  第二个参数是List<Document>，批量修改

//        List<Document> documents = Arrays.asList(document);

//        indexWriter.updateDocuments(new Term("title", "智"), Arrays.asList(document));

        //修改完提交

        indexWriter.commit();

        //7 关闭indexWriter

        indexWriter.close();

    }

    /**

     * 批量新增

     *

     * @throws IOException

     */

    @Test

    public void batchAddDoc() throws IOException {

        //1 创建索引库对象&指定索引库的位置

        FSDirectory directory = FSDirectory.open(new File("D:\\lucene").toPath());

        //2 创建IndexWriterConfig对象并指定分词器对象

        IndexWriterConfig indexWriterConfig = new IndexWriterConfig(new StandardAnalyzer());

        //3 创建一个IndexWriter对象 &指定索引库的位置&指定一个IndexWriterConfig对象。

        IndexWriter indexWriter = new IndexWriter(directory, indexWriterConfig);

        //4 批量创建document对象。

        //4.1 创建list集合用来存储Document对象

        List<Document> documents = new ArrayList<>();

        //4.2 创建10个Document对象

        for (int i = 0; i < 10; i++) {

            //4.3 创建document对象。

            Document document = new Document();

            //4.4 将field添加到document对象中。

            document.add(new StringField("docId", "docId" + i, Field.Store.YES));

            document.add(new TextField("title", "我的祖国" + i, Field.Store.YES));

            document.add(new TextField("content", "我的祖国是一个伟大的国家" + i, Field.Store.YES));

            document.add(new StringField("score", "100" + i, Field.Store.YES));

            document.add(new StoredField("name", "张三" + i));

            //4.4 添加到集合中

            documents.add(document);

        }

        //5 使用indexwriter对象将documents对象批量写入索引库中。

        indexWriter.addDocuments(documents);

        //6 关闭indexwriter对象。

        indexWriter.close();

    }

    /**

     * 删除所有

     *

     * @throws IOException

     */

    @Test

    public void deleteAllDoc() throws IOException {

        //1 创建索引库对象，指定索引库的位置

        FSDirectory directory = FSDirectory.open(new File("D:\\lucene").toPath());

//        2 创建IndexWriterConfig对象并指定分词器对象

        IndexWriterConfig indexWriterConfig = new IndexWriterConfig(new StandardAnalyzer());

//        3 创建一个IndexWriter对象 &指定索引库的位置&指定一个IndexWriterConfig对象。

        IndexWriter indexWriter = new IndexWriter(directory, indexWriterConfig);

        //4 执行方法删除所有

        indexWriter.deleteAll();

        //5 关闭资源

        indexWriter.close();

    }

    /**

     * 根据查询条件删除

     *

     * @throws IOException

     */

    @Test

    public void deleteDocByQuery() throws IOException {

        //1 创建索引库对象&指定索引库的位置

        FSDirectory directory = FSDirectory.open(new File("D:\\lucene").toPath());

        //2 创建IndexWriterConfig对象并指定分词器对象

        IndexWriterConfig indexWriterConfig = new IndexWriterConfig(new StandardAnalyzer());

        //3 创建一个IndexWriter对象 &指定索引库的位置&指定一个IndexWriterConfig对象。

        IndexWriter indexWriter = new IndexWriter(directory, indexWriterConfig);

        //4 执行删除 先查询然后把满足查询条件的数据进行查询  好比在mysql 先select然后delete

        //4.1 指定查询条件 & 关联分词对象

        TermQuery query = new TermQuery(new Term("title", "9"));

        //4.2 执行删除

        indexWriter.deleteDocuments(query);

        //5 关闭资源

        indexWriter.close();

    }

    /**

     * 根据分词删除

     *

     * @throws IOException

     */

    @Test

    public void deleteDocByTerm() throws IOException {

        //1 创建索引库对象&指定索引库的位置

        FSDirectory directory = FSDirectory.open(new File("D:\\lucene").toPath());

        //2 创建IndexWriterConfig对象并指定分词器对象

        IndexWriterConfig indexWriterConfig = new IndexWriterConfig(new StandardAnalyzer());

        //3 创建一个IndexWriter对象 &指定索引库的位置&指定一个IndexWriterConfig对象。

        IndexWriter indexWriter = new IndexWriter(directory, indexWriterConfig);

        //4 执行删除 直接删除 好比在mysql 直接delete

        //4.1 指定分词

        Term term = new Term("title", "8");

        //4.2 执行删除

        indexWriter.deleteDocuments(term);

        //5 关闭资源

        indexWriter.close();

    }