springDataSolr搜索关键词高亮的实现

电玩女神 2022-05-17 06:36 211阅读 0赞

@Service(timeout=5000)
    public class ItemSearchServiceImpl implements ItemSearchService{
        @Autowired
        private SolrTemplate solrTemplate;
        @Override
        public Map<String, Object> search(Map searchMap) {
            Map<String,Object> map=new HashMap();
            //putAll可以合并两个MAP，只不过如果有相同的key那么用后面的覆盖前面的
            map.putAll(searchList(searchMap));
            return map;
        }
        
        private Map searchList(Map searchMap) {
            Map<String,Object> map=new HashMap();
            /*Query query=new SimpleQuery();
            //添加查询条件
            Criteria criteria = new Criteria("item_keywords").is(searchMap.get("keywords"));
            query.addCriteria(criteria);
            ScoredPage<TbItem> page=solrTemplate.queryForPage(query,TbItem.class);
            map.put("rows", page.getContent());*/
            //高亮查询
            HighlightQuery query=new SimpleHighlightQuery();
            
            HighlightOptions highlightOptions = new HighlightOptions().addField("item_title");//添加高亮域
            
            highlightOptions.setSimplePrefix("<em style='color:red'>");//高亮前缀
            highlightOptions.setSimplePostfix("</em>");//高亮后缀
            query.setHighlightOptions(highlightOptions);//将高亮域相关查询条件添加到query中
            //按照关键字查询
            Criteria criteria = new Criteria("item_keywords").is(searchMap.get("keywords"));
            //添加查询条件
            query.addCriteria(criteria);
            //获得高亮页对象,包含所有的高亮记录，一条记录为一个TbItem
            HighlightPage<TbItem> queryForHighlightPage = solrTemplate.queryForHighlightPage(query, TbItem.class);
            //获取所有的高亮记录
            List<HighlightEntry<TbItem>> highlighted = queryForHighlightPage.getHighlighted();
            //循环高亮入口，获取每一条记录
            for(HighlightEntry<TbItem> h:highlighted) {
                TbItem item=h.getEntity();//获取记录中的实体类
                if(h.getHighlights().size()>0&&h.getHighlights().get(0).getSnipplets().size()>0) {
                    item.setTitle(h.getHighlights().get(0).getSnipplets().get(0));//设置高亮域的结果
                }
            }
            /**
             * 
             * 以上操作是对原生内容的修改封装
             */
            //不进行如上操作，获取的是原生的对象，即是没有经过高亮处理的对象
            map.put("rows",queryForHighlightPage.getContent());
            return map;
        }

/\*\*  
\* h.getHighlights() 获取每条高亮记录的所有高亮域  
\* 在new HighlightOptions().addField("item\_title")是添加的高亮域  
\* h.getHighlights().get(0).getSnipplets()获取第一个高亮域的内容  
\* h.getHighlights().get(0).getSnipplets().get(0) 一个高亮域中可能存在多值  
\* 取决于solr中的配置域的是否配置了multiValued是否为true  
\*/