solr搜索引擎的使用(3)--java客户端调用
package cn.emag.solr.client;import java.lang.reflect.Field;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.log4j.Logger;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrQuery.SortClause;
import org.apache.solr.client.solrj.impl.HttpSolrServer;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocumentList;
import cn.emag.cached.utils.ConfigCache;
import cn.emag.utils.CheckNull;
import cn.emag.utils.LogName;
/**
* @Title: solr客户端工具类
* @Description:
* @Copyright: Copyright (c) Aug 27, 2013
* @Company: ****
* @Author: shen
* @Version: 1.0
* @history:
*/
public class SolrClient
{
private static SolrClientclient= null;
private static HttpSolrServersolrServer= null;
/**
* log4j对象
*/
private static Loggerlogger= Logger.getLogger(LogName.SYS_LOG);
public static synchronized SolrClient getInstance()
{
if (client == null)
{
client = new SolrClient();
}
return client;
}
public SolrClient()
{
// 从缓存中读取solr服务器地址
String url = ConfigCache.getSolrServerUrl();
solrServer = new HttpSolrServer(url);
solrServer.setSoTimeout(3000); // socket read timeout
solrServer.setConnectionTimeout(1000);
solrServer.setDefaultMaxConnectionsPerHost(1000);
solrServer.setMaxTotalConnections(10);
solrServer.setFollowRedirects(false); // defaults to false
solrServer.setAllowCompression(true);
solrServer.setMaxRetries(1);
}
/**
* 添加索引数据
*
* @param bean
* 实体bean（需要注解）
*/
public void saveOrUpdate(Object bean)
{
try
{
solrServer.addBean(bean);
solrServer.commit();
}
catch (Exception e)
{
logger.error("添加更新索引数据，出错：", e);
}
}
/**
* 添加索引数据
*
* @param list
* 实体bean集合（需要注解）
*/
public void saveOrUpdate(List<?> list)
{
try
{
for (Object o : list)
{
solrServer.addBean(o);
}
solrServer.commit();
}
catch (Exception e)
{
logger.error("添加更新索引数据，出错：", e);
}
}
public void deleteByQuery(String queryStr)
{
try
{
solrServer.deleteByQuery(queryStr);
solrServer.commit();
}
catch (Exception e)
{
logger.error("根据查询条件，删除索引数据，出错:", e);
}
}
/**
* 根据查询条件返回结果集合
*
* @param queryStr
* @param fieldList
* @param descList
* @param bean
* @return
*/
public List<?> queryList(String queryStr, List<String> fieldList, List<String> descList, Class<?> bean)
{
List<?> list = null;
try
{
SolrQuery query = new SolrQuery();
// 查询的条件
query.setQuery(queryStr);
// 返回的属性（列）,不设置则返回所有
if (fieldList != null && !fieldList.isEmpty())
{
for (String filed : fieldList)
{
query.add(filed);
}
}
// 设置排序项，升序还是降序
if (descList != null && !descList.isEmpty())
{
for (String desc : descList)
{
query.addSort(new SortClause(desc, SolrQuery.ORDER.desc));
}
}
// start row 相当于mysql中的limit
query.setStart(0);
query.setRows(10);
QueryResponse qrsp = solrServer.query(query);
list = qrsp.getBeans(bean);
}
catch (Exception e)
{
logger.error("根据条件查询索引数据，出错：", e);
}
return list;
}
/**
* @param queryStr
* solr查询语句
* @param fieldList
* 展示的字段集合
* @param descList
* 倒序的字段集合
* @param ascList
* 正序的字段集合
* @param bean
* 实体
* @param hlStr
* 高亮的字段
* @param idStr
* 实体的id
* @param hlField
* 高亮的结果存放的字段
* @param pageNum
* 页码
*
* @param pageSize
* 每页显示的记录数
* @return
*/
public Map<String, Object> queryListHightLight(String queryStr, List<String> fieldList, List<String> descList, List<String> ascList, Class<?> bean, String hlStr, String idStr, String hlField, int pageNum, int pageSize)
{
// 返回查询结果map
Map<String, Object> resultMap = new HashMap<String, Object>();
List<?> list = null;
try
{
SolrQuery query = new SolrQuery();
// 设置高亮
query.setHighlight(true);
query.setHighlightSimplePre("<span class='highlight'>");// 设置开头
query.setHighlightSimplePost("</span>"); // 设置结尾
// 设置高亮的哪些区域
query.setParam("hl.fl", hlStr);
// 查询的条件
query.setQuery(queryStr);
// 返回的属性（列）,不设置则返回所有
if (!CheckNull.isNull(fieldList))
{
for (String filed : fieldList)
{
query.add(filed);
}
}
// 设置排序项，降序
if (!CheckNull.isNull(descList))
{
for (String desc : descList)
{
query.addSort(new SortClause(desc, SolrQuery.ORDER.desc));
}
}
// 设置排序项，升序
if (!CheckNull.isNull(ascList))
{
for (String asc : ascList)
{
query.addSort(new SortClause(asc, SolrQuery.ORDER.asc));
}
}
// start row 相当于mysql中的limit
query.setStart((pageNum - 1) * pageSize);
query.setRows(pageSize);
QueryResponse qrsp = solrServer.query(query);
SolrDocumentList documentList = qrsp.getResults();
// 符合条件的记录数
int count = (int) documentList.getNumFound();
// 页数
int pageCount = count / pageSize + (count % pageSize > 0 ? 1 : 0);
resultMap.put("count", count);
resultMap.put("pageCount", pageCount);
list = qrsp.getBeans(bean);
// 处理高亮字段以及对应接收的属性值
String[] hlStrs = hlStr.split(",");
String[] hlFields = hlField.split(",");
Object temp = null;
String[] setMethod = new String;
for (int i = 0; i < hlStrs.length; i++)
{
setMethod = "set" + hlFields.toString().substring(0, 1).toUpperCase() + hlFields.toString().substring(1);
}
for (int i = 0; i < list.size(); i++)
{
temp = list.get(i);
// 通过反射获取id的值
Field fildId = temp.getClass().getDeclaredField(idStr);
fildId.setAccessible(true);
Object id = (Object) fildId.get(temp);
// 将存在高亮值的字段的值设置到对应的属性值中
if (null != qrsp.getHighlighting().get(id))
{
for (int j = 0; j < hlStrs.length; j++)
{
Object glConent = qrsp.getHighlighting().get(id).get(hlStrs);
if (null != glConent)
{
bean.getMethod(setMethod, String.class).invoke(temp, glConent.toString().substring(1, glConent.toString().length() - 1));
}
}
}
}
}
catch (Exception e)
{
logger.error("根据条件查询索引数据(高亮)，出错：", e);
}
resultMap.put("list", list);
return resultMap;
}
}
页:
[1]