lyd2004888 发表于 2017-12-19 10:39:55

Solr(五)Solr实现简单的类似百度搜索高亮功能

package com.sgcc.ge.solr;  

  
import java.io.IOException;
  
import java.util.ArrayList;
  
import java.util.Iterator;
  
import java.util.List;
  
import java.util.Map;
  

  
import org.apache.solr.client.solrj.SolrQuery;
  
import org.apache.solr.client.solrj.SolrServerException;
  
import org.apache.solr.client.solrj.impl.HttpSolrClient;
  
import org.apache.solr.client.solrj.response.QueryResponse;
  
import org.apache.solr.common.SolrDocument;
  
import org.apache.solr.common.SolrDocumentList;
  
import org.springframework.stereotype.Service;
  

  
import com.sgcc.ge.article.bean.ArticleBean;
  

  
@Service("solrRepository")

  
public>  /**
  *
  *
  *<p>Description:Get data from solr core return Json</p>
  *
  * @author:SongJia
  *
  * @date: 2017-8-9下午7:56:04
  *
  * @param tag
  * @return
  * @throws SolrServerException
  * @throws IOException
  */
  public List<ArticleBean> getSolrDataJson(String tag) throws SolrServerException, IOException{
  HttpSolrClient client = new HttpSolrClient("http://192.168.3.208:8080/solr/Test");
  QueryResponse resp=null;
  SolrQuery query = new SolrQuery("art_title:" +"*"+tag+"*" + " or art_tag:" + "*"+tag+"*"+" or art_abstract:"+"*"+tag+"*");
  query.setHighlight(true);
  query.addHighlightField("art_title");
  query.addHighlightField("art_tag");
  query.addHighlightField("art_abstract");
  query.setHighlightSimplePre("<font color=\"red\">");
  query.setHighlightSimplePost("</font>");
  query.setHighlightFragsize(300);
  resp = client.query(query);
  SolrDocumentList solrDocumentList = resp.getResults();
  //获取高亮的字段
  Map<String, Map<String, List<String>>> highlighting = resp.getHighlighting();
  List<ArticleBean> list =new ArrayList<ArticleBean>();
  //List<Map> result = new ArrayList<Map>();
  //取出数据并封装到Map里面
  for (int i = 0; i < solrDocumentList.size(); i++) {
  SolrDocument sd = solrDocumentList.get(i);
  ArticleBean bean = new ArticleBean();
  String art_title = (String) sd.getFieldValue("art_title");

  String>  String art_tag = (String) sd.getFieldValue("art_tag");
  String art_abstract = (String) sd.getFieldValue("art_abstract");
  String art_time = (String) sd.getFieldValue("art_time");
  bean.setArtTag(art_tag);
  bean.setArtTime(art_time);
  bean.setArtTitle(art_title);
  bean.setArtAbstract(art_abstract);
  Iterator<Map.Entry<String, Map<String, List<String>>>> it = highlighting.entrySet().iterator();
  while (it.hasNext()) {
  Map.Entry<String, Map<String, List<String>>> entry = it.next();
  String key = entry.getKey();
  Map<String, List<String>> value = entry.getValue();
  if(key==id||key.equals(id)){
  for (Map.Entry<String, List<String>> inentry : value.entrySet()) {
  String keyin = inentry.getKey();
  List<String> valuein = inentry.getValue();
  if(keyin=="art_abstract"||keyin.equals("art_abstract")){
  bean.setArtAbstract(valuein.get(0));
  }else if(keyin=="art_tag"||keyin.equals("art_tag")){
  bean.setArtTag(valuein.get(0));
  }else if(keyin=="art_title"||keyin.equals("art_title")){
  bean.setArtTitle(valuein.get(0));
  }
  }
  }
  }
  list.add(bean);
  }
  return list;
  }
  
}
  
页: [1]
查看完整版本: Solr(五)Solr实现简单的类似百度搜索高亮功能