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

文章目录

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

配置文件schama.xml

​​​以下查询，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;
    }
}

完