SolrTest solr测试工程

package com.solrcloud; import com.solrcloud.bean.QuestionBean; import com.solrcloud.utils.IDUtils; import org.apache.http.client.utils.DateUtils; import org.apache.solr.client.solrj.SolrQuery; import org.apache.solr.client.solrj.SolrServerException; import org.apache.solr.client.solrj.beans.DocumentObjectBinder; import org.apache.solr.client.solrj.impl.CloudSolrClient; import org.apache.solr.client.solrj.impl.HttpSolrClient; import org.apache.solr.client.solrj.response.*; import org.apache.solr.common.SolrDocument; import org.apache.solr.common.SolrDocumentList; import org.apache.solr.common.SolrInputDocument; import org.apache.solr.common.params.GroupParams; import org.apache.solr.common.params.ModifiableSolrParams; import org.apache.solr.common.params.SolrParams; import org.springframework.util.StringUtils; import java.io.IOException; import java.text.SimpleDateFormat; import java.util.*; /** * CloudSolrClient */ public class SolrCloudServer { private static SolrCloudServer solrCloudServer = null; private static CloudSolrClient cloudSolrClient = null; private static List<String> zkHosts = new ArrayList<String>();//"127.0.0.1:2181;127.0.0.1:2182;127.0.0.1:2183"; // private static String zkHost = "127.0.0.1:2181/solr"; public static synchronized SolrCloudServer getInstance() { if (solrCloudServer == null) { solrCloudServer = new SolrCloudServer(); getServer(); } return solrCloudServer; } private static CloudSolrClient getServer() { try { if (cloudSolrClient == null) { zkHosts.add("127.0.0.1:2181"); zkHosts.add("127.0.0.1:2182"); zkHosts.add("127.0.0.1:2183"); cloudSolrClient = new CloudSolrClient(zkHosts,"/"); cloudSolrClient.setDefaultCollection("quectionCollection2"); // cloudSolrClient = new CloudSolrClient(zkHost); cloudSolrClient.setZkClientTimeout(1000); cloudSolrClient.setZkConnectTimeout(1000); } } catch (Exception e) { System.out.println("SolrCloud连接失败，请检查ZK是否正常启动或地址端口是否配置正确！"); e.printStackTrace(); } return cloudSolrClient; } // 简单的查询，取出二十个 public List<QuestionBean> queryTop20(String keywork,int maxResult) { System.out.println("简单查询取出前二十个"); List<QuestionBean> beanList=null; long startTime=System.currentTimeMillis(); try { ModifiableSolrParams params = new ModifiableSolrParams(); // 查询关键词 if(StringUtils.isEmpty(keywork)){ params.set("q", "*:*"); }else{ params.set("q", "question:"+keywork); } // 返回信息 params.set("fl", "id,title,url,view,answer"); // 排序 //params.set("sort", "id asc"); // 分页,start=0就是从0开始,rows=5当前返回5条记录,第二页就是变化start这个值为5就可以了 params.set("start", 0); params.set("rows", maxResult); // 返回格式 params.set("wt", "javabin"); SolrDocumentList docs = cloudSolrClient.query(params).getResults(); // System.out.println("查询到的总记录数："+docs.getNumFound()); // System.out.println("耗时：" + (System.currentTimeMillis()-startTime)+" 毫秒"); DocumentObjectBinder binder = new DocumentObjectBinder(); beanList=binder.getBeans(QuestionBean.class, docs); for (QuestionBean s : beanList) { System.out.println(s); } //cloudSolrClient.shutdown(); // String dtEnd = new SimpleDateFormat("yyyyMMddHHmmssSSS").format(new Date()); System.out.println("耗时：" + (System.currentTimeMillis()-startTime)+" 毫秒"); } catch (SolrServerException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } return beanList; } // 简单的查询，取出二十个,通过JavaBean的方式返回 public void queryBeanTop20() { System.out.println("简单查询取出前二十个"); String dtStart = new SimpleDateFormat("yyyyMMddHHmmssSSS") .format(new Date()); System.out.println("开始时间：" + dtStart); try { ModifiableSolrParams params = new ModifiableSolrParams(); // 查询关键词 params.set("q", "计算机"); // 返回信息 params.set("fl", "id,name,price,score"); // 排序 params.set("sort", "price asc"); // 分页,start=0就是从0开始,rows=5当前返回5条记录,第二页就是变化start这个值为5就可以了 params.set("start", 0); params.set("rows", 20); // 返回格式 params.set("wt", "javabin"); SolrDocumentList list = cloudSolrClient.query(params).getResults(); DocumentObjectBinder binder = new DocumentObjectBinder(); List<QuestionBean> beanList=binder.getBeans(QuestionBean.class, list); for (QuestionBean b : beanList) { System.out.println("url="+b.getUrl()+";title="+b.getTitle()); } cloudSolrClient.shutdown(); String dtEnd = new SimpleDateFormat("yyyyMMddHHmmssSSS").format(new Date()); System.out.println("结束时间：" + dtEnd); } catch (SolrServerException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } /** * 添加索引 * @param questionBean */ public void addBeanToIndex(QuestionBean questionBean) { try { //获取连接服务 SolrInputDocument doc1 = new SolrInputDocument(); doc1.setField("id",questionBean.getId()); doc1.setField("url",questionBean.getUrl()); doc1.setField("title",questionBean.getTitle() ); doc1.setField("content", questionBean.getContent()); doc1.setField("answer", questionBean.getAnswer());//回答数 doc1.setField("focus", questionBean.getFocus());//关注 doc1.setField("sideBookmarked", questionBean.getSideBookmarked()); //收藏 doc1.setField("view", questionBean.getView());//浏览 String createTime= DateUtils.formatDate(questionBean.getCreateDate(), "yyyyMMddHHmmss"); doc1.setField("createTime",createTime); System.out.println("indexId="+questionBean.getId()+";title="+questionBean.getTitle()); cloudSolrClient.add(doc1); cloudSolrClient.commit(); } catch (SolrServerException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } /** * 添加索引 批量 * @param questionBeanList */ public void addBeanToIndex(List<QuestionBean> questionBeanList) { try { //获取连接服务 SolrInputDocument doc1 ; if(questionBeanList!=null&&questionBeanList.size()>0){ for (QuestionBean q:questionBeanList) { doc1 = new SolrInputDocument(); doc1.setField("id",q.getId());