版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_36059561/article/details/83960895
一、对索引文件添加高亮
现在我们有一些索引文件,怎么对这些索引文件添加高亮呢?首先需要根据搜索域和值找到这篇文档,然后通过文档获取域值,再将域值交给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("必须");
}
}