|
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;
}
}
|
|
|