ddsdjn 发表于 2017-12-20 11:17:07

13.4SolrCloud集群使用手册之CRUD

package cn.ljh.ssm.test;  

  

import java.io.IOException;  

import java.util.ArrayList;  

import java.util.List;  

  

import org.apache.solr.client.solrj.SolrQuery;  

import org.apache.solr.client.solrj.SolrServerException;  

import org.apache.solr.client.solrj.impl.CloudSolrServer;  

import org.apache.solr.client.solrj.response.QueryResponse;  

import org.apache.solr.common.SolrDocument;  

import org.apache.solr.common.SolrDocumentList;  

import org.apache.solr.common.SolrInputDocument;  

import org.junit.After;  

import org.junit.Before;  

import org.junit.Test;  

  

public>
private CloudSolrClient client;private CloudSolrServer cloudSolrServer;  

public synchronized void open(final String zkHost, final StringdefaultCollection,intzkClientTimeout, final int zkConnectTimeout) {if (cloudSolrServer == null) {try {  cloudSolrServer
= new CloudSolrServer(zkHost);  cloudSolrServer.setDefaultCollection(defaultCollection);
  cloudSolrServer.setZkClientTimeout(zkClientTimeout);
  cloudSolrServer.setZkConnectTimeout(zkConnectTimeout);
  }
catch (Exception e) {  e.printStackTrace();
  }
  }
  }
  @Before
public void cloudSolrServerBuild(){final String zkHost = "192.168.137.171:2181,192.168.137.172:2181,192.168.137.173:2181";final StringdefaultCollection = "myc";final intzkClientTimeout = 20000;final int zkConnectTimeout = 2000;  

  client
= new CloudSolrClient();  client.open(zkHost, defaultCollection, zkClientTimeout, zkConnectTimeout);
  }
  @After
public void clean(){  client
= null;  }
  @Test
public void testAdd(){try {//先删除所有数据  client.cloudSolrServer.deleteByQuery("*:*");
  SolrInputDocument doc = new SolrInputDocument();
  doc.addField("id","200");
  doc.addField("stu_name","张三");
  doc.addField("stu_sex",0);
  doc.addField("stu_address","nanchang");
  client.cloudSolrServer.add(doc);
  SolrInputDocument doc2 = new SolrInputDocument();
  doc2.addField("id","201");
  doc2.addField("stu_name","李四");
  doc2.addField("stu_sex",1);
  doc2.addField("stu_address","changsha");
  client.cloudSolrServer.add(doc2);
  client.cloudSolrServer.commit();//提交,将所有更新提交到索引中
  } catch (SolrServerException e) {
  e.printStackTrace();
  } catch (IOException e) {
  e.printStackTrace();
  }
  }
  /**
  * 使用POJO添加document
  */
  @Test
  public void testAddStudent(){
  try {
  List<Student> studentList = new ArrayList<Student>();
  Student stu1 = new Student();
  stu1.setId("103");
  stu1.setName("张小强");
  stu1.setSex(1);
  stu1.setAddress("北京市海淀区知春路");
  studentList.add(stu1);
  stu1 = new Student();
  stu1.setId("104");
  stu1.setName("刘小米");
  stu1.setSex(0);
  stu1.setAddress("北京市海淀区北苑路");
  studentList.add(stu1);
  client.cloudSolrServer.addBeans(studentList);
  client.cloudSolrServer.commit();
  } catch (SolrServerException e) {
  e.printStackTrace();
  } catch (IOException e) {
  e.printStackTrace();
  }
  }
  @Test
  public void testQueryStudent(){
  try {
  String strQuery = "stu_name:小米";//q表示查询的内容
  SolrQuery query = new SolrQuery(strQuery);
  QueryResponse resp = client.cloudSolrServer.query(query);
  SolrDocumentList sdList = resp.getResults();
  long totalResults = sdList.getNumFound();//命中的总记录数
  
            
  System.out.println("totalResults-->"+totalResults);
  for(SolrDocument sd:sdList){
  Student student = client.cloudSolrServer.getBinder().getBean(Student.class, sd);
  System.out.println(student);
  }
  } catch (SolrServerException e) {
  e.printStackTrace();
  }
  }
  @Test
  public void testQuery(){
  try {
  String strQuery = "stu_address:nanchang";//q表示查询的内容
  SolrQuery query = new SolrQuery();
  query.set("q",strQuery);
  QueryResponse resp = client.cloudSolrServer.query(query);
  SolrDocumentList sdList = resp.getResults();
  long totalResults = sdList.getNumFound();//命中的总记录数
  
            
  System.out.println("totalResults-->"+totalResults);
  for(SolrDocument sd:sdList){
  System.out.print("id:" + sd.getFieldValue("id") +
  " " + " stu_name:" + sd.getFieldValue("stu_name") +
  " " + " stu_address:" + sd.getFieldValue("stu_address"));
  }
  } catch (SolrServerException e) {
  e.printStackTrace();
  }
  }
  @Test
  public void testDelete(){
  try {
  client.cloudSolrServer.deleteById("200");
  client.cloudSolrServer.commit();
  } catch (Exception e) {
  e.printStackTrace();
  }
  }
  

  
}
页: [1]
查看完整版本: 13.4SolrCloud集群使用手册之CRUD