搜索学习入门--使用LuceneHighlighter高亮显示Lucene检索结果的关键词

野性酷女 2022-02-20 15:45 210阅读 0赞

转载于： [https://blog.csdn.net/qq\_18860653/article/details/78593328][https_blog.csdn.net_qq_18860653_article_details_78593328]

package top.yuyufeng.learn.lucene.demo1;
    
    import org.apache.lucene.analysis.Analyzer;
    import org.apache.lucene.analysis.standard.StandardAnalyzer;
    import org.apache.lucene.document.Document;
    import org.apache.lucene.index.DirectoryReader;
    import org.apache.lucene.index.IndexReader;
    import org.apache.lucene.index.IndexWriterConfig;
    import org.apache.lucene.queryparser.classic.ParseException;
    import org.apache.lucene.queryparser.classic.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.SimpleFSDirectory;
    import org.apache.lucene.util.Version;
    
    import java.io.File;
    import java.io.IOException;
    
    /**
     * @author yuyufeng
     * @date 2017/11/21
     */
    public class LuceneHighlighterDemo {
        public static void main(String[] args) {
            // Lucene Document的域名
            String fieldName = "blog";
            Analyzer analyzer = new StandardAnalyzer(Version.LUCENE_47);
            Directory directory = null;
            IndexReader ireader = null;
            IndexSearcher isearcher;
    
            try {
                //索引目录
                directory = new SimpleFSDirectory(new File("D://test/lucene_index"));
                // 配置IndexWriterConfig
                IndexWriterConfig iwConfig = new IndexWriterConfig(Version.LUCENE_47, analyzer);
                iwConfig.setOpenMode(IndexWriterConfig.OpenMode.CREATE_OR_APPEND);
    
                // 搜索过程**********************************
                // 实例化搜索器
                ireader = DirectoryReader.open(directory);
                isearcher = new IndexSearcher(ireader);
    
                String keyword = "达摩院";
                // 使用QueryParser查询分析器构造Query对象
                QueryParser qp = new QueryParser(Version.LUCENE_47, fieldName, analyzer);
                qp.setDefaultOperator(QueryParser.OR_OPERATOR);  // and or 跟数据库查询语法类似
                Query query = qp.parse(keyword);
                System.out.println("Query = " + query);
    
                // 搜索相似度最高的5条记录
                TopDocs topDocs = isearcher.search(query, 5);
                System.out.println("命中：" + topDocs.totalHits);
                // 遍历输出结果
                ScoreDoc[] scoreDocs = topDocs.scoreDocs;
    
    
                Highlighter highlighter = new Highlighter(new SimpleHTMLFormatter("<B>", "</B>"),   //高亮格式，用<B>标签包裹
                        new QueryScorer(query));
                Fragmenter fragmenter = new SimpleFragmenter(100);   //高亮后的段落范围在100字内
                highlighter.setTextFragmenter(fragmenter);
    
                for (int i = 0; i < topDocs.totalHits; i++) {
                    Document targetDoc = isearcher.doc(scoreDocs[i].doc);
                    System.out.println("内容：" + highlighter.getBestFragment(analyzer, fieldName, targetDoc.get("blog")));  //只对“blog”属性高亮
                }
    
            } catch (ParseException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            } catch (InvalidTokenOffsetsException e) {
                e.printStackTrace();
            } finally {
                if (ireader != null) {
                    try {
                        ireader.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
                if (directory != null) {
                    try {
                        directory.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }

[https_blog.csdn.net_qq_18860653_article_details_78593328]: https://blog.csdn.net/qq_18860653/article/details/78593328