spring data solr实现关键字搜索+高亮显示+分组查询

男娘i 2023-06-13 04:56 44阅读 0赞

### 文章目录 ###

*  配置文件schama.xml
 *  普通查询(关键代码)
 *  高亮查询（关键代码）
 *  分组查询（关键代码）

# 配置文件schama.xml #

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70]  
**`以下查询，controller调用的均是service的search方法！！`**

# 普通查询(关键代码) #

import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.data.solr.core.SolrTemplate;
    import org.springframework.data.solr.core.query.Criteria;
    import org.springframework.data.solr.core.query.Query;
    import org.springframework.data.solr.core.query.SimpleQuery;
    import org.springframework.data.solr.core.query.result.ScoredPage;
    import com.alibaba.dubbo.config.annotation.Service;
    import com.pinyougou.pojo.TbItem;
    import com.pinyougou.search.service.ItemSearchService;
    @Service(timeout=5000)
    public class ItemSearchServiceImpl implements ItemSearchService { 
    
    	@Autowired
    	private SolrTemplate solrTemplate;
    	
    	@Override
    	public Map search(Map searchMap) { 
    		Map 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());
    		
    		return map;
    	}
    
    }

# 高亮查询（关键代码） #

@Service(timeout=5000)
    public class ItemSearchServiceImpl implements ItemSearchService { 
    
    	@Autowired
    	private SolrTemplate solrTemplate;
    
    
    	/** * 根据关键字搜索列表 * @param searchMap * @return */
    	private Map searchList(Map searchMap){ 
    		Map map=new HashMap();
    		HighlightQuery query=new SimpleHighlightQuery();
    		HighlightOptions highlightOptions=new HighlightOptions().addField("item_title");//设置高亮的域
    		highlightOptions.setSimplePrefix("<em style='color:red'>");//高亮前缀
    		highlightOptions.setSimplePostfix("</em>");//高亮后缀
    		query.setHighlightOptions(highlightOptions);//设置高亮选项
    		//按照关键字查询
    		Criteria criteria=new Criteria("item_keywords").is(searchMap.get("keywords"));
    		query.addCriteria(criteria);
    		HighlightPage<TbItem> page = solrTemplate.queryForHighlightPage(query, TbItem.class);
    		for(HighlightEntry<TbItem> h: page.getHighlighted()){ //循环高亮入口集合
    			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",page.getContent());
    		return map;
    	}
    
    
    	@Override
    	public Map search(Map searchMap) { 
    		Map map=new HashMap();
    		//1.查询列表
    		map.putAll(searchList(searchMap));
    		return map;
    	}
    
    }

# 分组查询（关键代码） #

import java.util.ArrayList;
    import java.util.HashMap;
    import java.util.List;
    import java.util.Map;
    
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.data.domain.Page;
    import org.springframework.data.solr.core.SolrTemplate;
    import org.springframework.data.solr.core.query.*;
    import org.springframework.data.solr.core.query.result.*;
    import com.alibaba.dubbo.config.annotation.Service;
    import com.pinyougou.pojo.TbItem;
    import com.pinyougou.search.service.ItemSearchService;
    
    @Service(timeout = 5000)
    public class ItemSearchServiceImpl implements ItemSearchService { 
    
        @Autowired
        private SolrTemplate solrTemplate;
    
    
        @Override
        public Map search(Map searchMap) { 
            Map map = new HashMap();
            //1.查询列表
            map.putAll(searchList(searchMap));
            List categoryList = searchCategoryList(searchMap);
            map.put("categoryList",categoryList);
    
            return map;
        }
    
        /** * 查询分类列表 * * @param searchMap * @return */
        private List searchCategoryList(Map searchMap) { 
            List<String> list = new ArrayList();
            Query query = new SimpleQuery();
            //按照关键字查询
            Criteria criteria = new Criteria("item_keywords").is(searchMap.get("keywords"));
            query.addCriteria(criteria);
            //设置分组选项
            GroupOptions groupOptions = new GroupOptions().addGroupByField("item_category");
            query.setGroupOptions(groupOptions);
            //得到分组页
            GroupPage<TbItem> page = solrTemplate.queryForGroupPage(query, TbItem.class);
            //根据列得到分组结果集
            GroupResult<TbItem> groupResult = page.getGroupResult("item_category");
            //得到分组结果入口页
            Page<GroupEntry<TbItem>> groupEntries = groupResult.getGroupEntries();
            //得到分组入口集合
            List<GroupEntry<TbItem>> content = groupEntries.getContent();
            for (GroupEntry<TbItem> entry : content) { 
                list.add(entry.getGroupValue());//将分组结果的名称封装到返回值中
            }
            return list;
        }
    
    
        /** * 根据关键字搜索列表 * * @param searchMap * @return */
        private Map searchList(Map searchMap) { 
            Map map = new HashMap();
            HighlightQuery query = new SimpleHighlightQuery();
            HighlightOptions highlightOptions = new HighlightOptions().addField("item_title");//设置高亮的域
            highlightOptions.setSimplePrefix("<em style='color:red'>");//高亮前缀
            highlightOptions.setSimplePostfix("</em>");//高亮后缀
            query.setHighlightOptions(highlightOptions);//设置高亮选项
            //按照关键字查询
            Criteria criteria = new Criteria("item_keywords").is(searchMap.get("keywords"));
            query.addCriteria(criteria);
            HighlightPage<TbItem> page = solrTemplate.queryForHighlightPage(query, TbItem.class);
            for (HighlightEntry<TbItem> h : page.getHighlighted()) { //循环高亮入口集合
                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", page.getContent());
            return map;
        }
    }

--------------------

完

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191114105828700.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1,size_16,color_FFFFFF,t_70