SpringBoot+Lucene搜索结果高亮显示Demo

package com.lucene.test; import java.io.IOException; import java.io.StringReader; import java.nio.file.Paths; import java.util.HashMap; import java.util.Map; import javax.servlet.http.HttpServletRequest; import org.apache.lucene.analysis.Analyzer; import org.apache.lucene.analysis.TokenStream; import org.apache.lucene.analysis.cn.smart.SmartChineseAnalyzer; import org.apache.lucene.analysis.standard.StandardAnalyzer; import org.apache.lucene.analysis.tokenattributes.CharTermAttribute; import org.apache.lucene.analysis.tokenattributes.OffsetAttribute; import org.apache.lucene.document.Document; import org.apache.lucene.index.DirectoryReader; import org.apache.lucene.index.IndexReader; 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.Fragmenter; import org.apache.lucene.search.highlight.Highlighter; import org.apache.lucene.search.highlight.InvalidTokenOffsetsException; import org.apache.lucene.search.highlight.QueryScorer; import org.apache.lucene.search.highlight.SimpleHTMLFormatter; import org.apache.lucene.search.highlight.SimpleSpanFragmenter; import org.apache.lucene.store.Directory; import org.apache.lucene.store.FSDirectory; import org.springframework.boot.SpringApplication; import org.springframework.boot.autoconfigure.SpringBootApplication; import org.springframework.boot.context.web.SpringBootServletInitializer; import org.springframework.stereotype.Controller; import org.springframework.ui.Model; import org.springframework.ui.ModelMap; import org.springframework.web.bind.annotation.RequestMapping; import com.springboot.test.Application1; @SpringBootApplication @Controller public class Searcher extends SpringBootServletInitializer { @RequestMapping("/hello1") public String search(ModelMap map) throws Exception{ Map<Integer,String> m = new HashMap<Integer, String>(); //查询文件路径 String indexDir="E:\\LuceneDemo\\lucene4"; //查询词汇 String q="篮球"; Directory dir = FSDirectory.open(Paths.get(indexDir)); IndexReader reader= DirectoryReader.open(dir); IndexSearcher is=new IndexSearcher(reader); SmartChineseAnalyzer analyzer=new SmartChineseAnalyzer();//中文分词器 QueryParser parser=new QueryParser("desc", analyzer); Query query= parser.parse(q); long start=System.currentTimeMillis(); TopDocs hits=is.search(query, 10); long end=System.currentTimeMillis(); System.out.println("匹配 "+q+" ，总共花费"+(end-start)+"毫秒"+"查询到"+hits.totalHits+"个记录"); QueryScorer scorer=new QueryScorer(query); Fragmenter fragmenter=new SimpleSpanFragmenter(scorer); SimpleHTMLFormatter simpleHTMLFormatter=new SimpleHTMLFormatter("<b><font color='red'>","</font></b>"); Highlighter highlighter=new Highlighter(simpleHTMLFormatter, scorer); highlighter.setTextFragmenter(fragmenter); for(ScoreDoc scoreDoc:hits.scoreDocs){ Document doc=is.doc(scoreDoc.doc); System.out.println(doc.get("name")); System.out.println(doc.get("id")); //Integer id =doc.get("id"); String desc=doc.get("desc"); Integer id=Integer.parseInt(doc.get("id")); if(desc!=null){ TokenStream tokenStream=analyzer.tokenStream("desc", new StringReader(desc)); String content=highlighter.getBestFragment(tokenStream, desc); System.out.println(content); m.put(id, content); } } reader.close(); map.addAttribute("map", m); return "index"; } public static void main(String[] args) { SpringApplication.run(Searcher.class, args); } }