Lucene笔记35-Lucene的扩展-高亮索引文件

一、对索引文件添加高亮

现在我们有一些索引文件，怎么对这些索引文件添加高亮呢？首先需要根据搜索域和值找到这篇文档，然后通过文档获取域值，再将域值交给highlighter进行加工并返回，这个时候，返回的值就是添加过高亮的了。

二、代码展示

package com.wsy;

import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.document.Document;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.queryParser.MultiFieldQueryParser;
import org.apache.lucene.queryParser.ParseException;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.ScoreDoc;
import org.apache.lucene.search.TopDocs;
import org.apache.lucene.search.highlight.*;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.FSDirectory;
import org.apache.lucene.util.Version;
import org.apache.tika.Tika;
import org.apache.tika.exception.TikaException;
import org.wltea.analyzer.lucene.IKAnalyzer;

import java.io.File;
import java.io.IOException;

public class HighLightTest {
    public void search(String name) {
        Analyzer analyzer = new IKAnalyzer();
        try {
            Directory directory = FSDirectory.open(new File("E:\\Lucene\\IndexLibrary"));
            IndexSearcher indexSearcher = new IndexSearcher(IndexReader.open(directory));
            // 这里只能接受fileName这一个域，所以下面highLight2()之后，content是null
            // QueryParser queryParser = new QueryParser(Version.LUCENE_35, "fileName", analyzer);
            // 使用MultiFieldQueryParser可以接受多个域，表示既查询fileName，又查询content，这时候，经过highLight2()之后，content就有值了
            MultiFieldQueryParser queryParser = new MultiFieldQueryParser(Version.LUCENE_35, new String[]{"fileName", "content"}, analyzer);
            Query query = queryParser.parse(name);
            TopDocs topDocs = indexSearcher.search(query, 10);
            for (ScoreDoc scoreDoc : topDocs.scoreDocs) {
                Document document = indexSearcher.doc(scoreDoc.doc);
                String fileName = document.get("fileName");
                fileName = highLight2(analyzer, query, fileName, "fileName");
                System.out.println(fileName);
                String content = new Tika().parseToString(new File(document.get("path")));
                content = highLight2(analyzer, query, content, "content");
                System.out.println(content);
            }
            indexSearcher.close();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ParseException e) {
            e.printStackTrace();
        } catch (InvalidTokenOffsetsException e) {
            e.printStackTrace();
        } catch (TikaException e) {
            e.printStackTrace();
        }
    }

    public String highLight2(Analyzer analyzer, Query query, String text, String field) throws IOException, InvalidTokenOffsetsException {
        QueryScorer queryScorer = new QueryScorer(query);
        Fragmenter fragmenter = new SimpleSpanFragmenter(queryScorer);
        Formatter formatter = new SimpleHTMLFormatter("<span>", "</span>");
        Highlighter highlighter = new Highlighter(formatter, queryScorer);
        highlighter.setTextFragmenter(fragmenter);
        String highlightString = highlighter.getBestFragment(analyzer, field, text);
        // 如果查询的highlightString为null，表示在field中没有查到，就返回原来的内容
        return highlightString == null ? text : highlightString;
    }

    public static void main(String[] args) {
        HighLightTest highLightTest = new HighLightTest();
        highLightTest.search("必须");
    }
}