Solr客户端自定义开发

lichengcom2009

　　
@Service
　　
@Qualifier("searchService")

　　
public>　　
@Value("${solr.url}")
　　
private String searchurl;
　　
@Value("${solr.q.fields}")
　　
private String qfields;
　　
@Value("${solr.s.fields}")
　　
private String sfields;
　　
@Value("${solr.h.fields}")
　　
private String hfields;
　　
/**
　　
* 分页查询Position
　　
* @param key 关键字
　　
* @param page 分页
　　
* @param pagesize 分页大小
　　
* @return 分页信息
　　
*/
　　
public PageInfo query(String key, int page, int pagesize) {
　　
PageInfo ps = new PageInfo(page, pagesize);
　　
SolrServer server = new HttpSolrServer(searchurl);
　　
SolrQuery params = new SolrQuery();
　　
params.setQuery(setQueryFields(key, qfields));
　　
params.setStart(ps.getStart());
　　
params.setRows(ps.getPagesize());
　　
boolean ish = setHighlight(params,hfields);
　　
setSortFields(params, sfields);
　　
SolrDocumentList list = null;
　　
//第一个Map的键是文档的ID，第二个Map的键是高亮显示的字段名
　　
Map high= null;
　　
try {
　　
QueryResponse response = server.query(params);
　　
list = response.getResults();
　　
if(ish){
　　
high = response.getHighlighting();
　　
}
　　
} catch (Exception e) {
　　
e.printStackTrace();
　　
}
　　
if(ish){
　　

　　
}
　　
ps.setItems(mappingVo(list,high,ish));
　　
ps.setTotal((int) list.getNumFound());
　　
return ps;
　　
}
　　
/**
　　
* 设置高亮字段
　　
* @param params SolrQuery对象
　　
* @param hfields2 高亮字段，以逗号分隔
　　
* @return 是否需要高亮
　　
*/
　　
private boolean setHighlight(SolrQuery params, String hfields2) {
　　
if(StringUtils.hasText(hfields2)){
　　
String[] arr = hfields2.split(",");
　　
for (String s : arr) {
　　
params.addHighlightField(s);
　　
}
　　
params.setHighlight(true);
　　
params.setHighlightSimplePre("");
　　
params.setHighlightSimplePost("");
　　
return true;
　　
}
　　
return false;
　　
}
　　
/**
　　
* 将查询结果返回映射为Position对象列表
　　
* @param list 返回的document列表
　　
* @param high 高亮信息
　　
* @param ish 是否需要高亮
　　
* @return Position对象列表
　　
*/
　　
private List mappingVo(SolrDocumentList list,Map high,boolean ish) {
　　
List poses = new ArrayList();
　　
for (SolrDocument doc : list) {
　　
Position p = new Position();
　　
p.setPid(doc.getFieldValue("CP_ID").toString());
　　
if(ish && high.get(p.getPid()).containsKey("COMPANY_NAME")){
　　
p.setCompany(high.get(p.getPid()).get(("COMPANY_NAME")).toString());
　　
}else{
　　
p.setCompany(doc.getFieldValue("COMPANY_NAME").toString());
　　
}
　　
p.setStamp((Date)doc.getFieldValue("ADD_AT"));
　　
poses.add(p);
　　
}
　　
return poses;
　　
}
　　
/**
　　
* 设置排序字段
　　
* @param params SolrQuery对象
　　
* @param sfields2 排序字段，都是按降序排
　　
*/
　　
private void setSortFields(SolrQuery params, String sfields2) {
　　
String[] arr = sfields2.split(",");
　　
for (String s : arr) {
　　
params.addSortField(s, ORDER.desc);
　　
}
　　

　　
}
　　
/**
　　
* 定义查询字段
　　
* @param key 关键字
　　
* @param qfields2 以，分隔的字段列表，以或条件
　　
* @return
　　
*/
　　
private String setQueryFields(String key, String qfields2) {
　　
StringBuilder sb = new StringBuilder();
　　
String[] arr = qfields2.split(",");
　　
for (String s : arr) {
　　
sb.append(s).append(":").append(key).append(" OR ");
　　
}
　　
String q = sb.toString();
　　
return q.substring(0, q.length() - 4);
　　
}
　　
}