JestService.java[html] view plain copy 在CODE上查看代码片派生到我的代码片
public class JestService {  /**  * 获取JestClient对象  * @return  */  public JestClient getJestClient() {    JestClientFactory factory = new JestClientFactory();  factory.setHttpClientConfig(new HttpClientConfig  .Builder("http://localhost:9200")  .gson(new GsonBuilder().setDateFormat("yyyy-MM-dd'T'hh:mm:ss").create())  .connTimeout(1500)  .readTimeout(3000)  .multiThreaded(true)  .build());  return factory.getObject();  }  /**  * 创建索引  * @param jestClient  * @param indexName  * @return  * @throws Exception  */  public boolean createIndex(JestClient jestClient, String indexName) throws Exception {  JestResult jr = jestClient.execute(new CreateIndex.Builder(indexName).build());  return jr.isSucceeded();  }  /**  * Put映射  * @param jestClient  * @param indexName  * @param typeName  * @param source  * @return  * @throws Exception  */  public boolean createIndexMapping(JestClient jestClient, String indexName, String typeName, String source) throws Exception {  PutMapping putMapping = new PutMapping.Builder(indexName, typeName, source).build();  JestResult jr = jestClient.execute(putMapping);  return jr.isSucceeded();  }  /**  * Get映射  * @param jestClient  * @param indexName  * @param typeName  * @return  * @throws Exception  */  public String getIndexMapping(JestClient jestClient, String indexName, String typeName) throws Exception {  GetMapping getMapping = new GetMapping.Builder().addIndex(indexName).addType(typeName).build();  JestResult jr = jestClient.execute(getMapping);  return jr.getJsonString();  }  /**  * 索引文档  * @param jestClient  * @param indexName  * @param typeName  * @param objs  * @return  * @throws Exception  */  public boolean index(JestClient jestClient, String indexName, String typeName, List<Object> objs) throws Exception {  Bulk.Builder bulk = new Bulk.Builder().defaultIndex(indexName).defaultType(typeName);  for (Object obj : objs) {  Index index = new Index.Builder(obj).build();  bulk.addAction(index);  }  BulkResult br = jestClient.execute(bulk.build());  return br.isSucceeded();  }  /**  * 搜索文档  * @param jestClient  * @param indexName  * @param typeName  * @param query  * @return  * @throws Exception  */  public SearchResult search(JestClient jestClient, String indexName, String typeName, String query) throws Exception {  Search search = new Search.Builder(query)  .addIndex(indexName)  .addType(typeName)  .build();  return jestClient.execute(search);  }  /**  * Count文档  * @param jestClient  * @param indexName  * @param typeName  * @param query  * @return  * @throws Exception  */  public Double count(JestClient jestClient, String indexName, String typeName, String query) throws Exception {  Count count = new Count.Builder()  .addIndex(indexName)  .addType(typeName)  .query(query)  .build();  CountResult results = jestClient.execute(count);   return results.getCount();  }  /**  * Get文档  * @param jestClient  * @param indexName  * @param typeName  * @param id  * @return  * @throws Exception  */  public JestResult get(JestClient jestClient, String indexName, String typeName, String id) throws Exception {  Get get = new Get.Builder(indexName, id).type(typeName).build();  return jestClient.execute(get);  }  /**  * Delete索引  * @param jestClient  * @param indexName  * @return  * @throws Exception  */  public boolean delete(JestClient jestClient, String indexName) throws Exception {  JestResult jr = jestClient.execute(new DeleteIndex.Builder(indexName).build());  return jr.isSucceeded();  }  /**  * Delete文档  * @param jestClient  * @param indexName  * @param typeName  * @param id  * @return  * @throws Exception  */  public boolean delete(JestClient jestClient, String indexName, String typeName, String id) throws Exception {  DocumentResult dr = jestClient.execute(new Delete.Builder(id).index(indexName).type(typeName).build());  return dr.isSucceeded();  }  /**  * 关闭JestClient客户端  * @param jestClient  * @throws Exception  */  public void closeJestClient(JestClient jestClient) throws Exception {  if (jestClient != null) {  jestClient.shutdownClient();  }  }
}
（3）UserTest.java[html] view plain copy 在CODE上查看代码片派生到我的代码片
public class UserTest {  private JestService jestService;  private JestClient jestClient;  private String indexName = "hwd";  private String typeName = "user";  @Before  public void setUp() throws Exception {  jestService = new JestService();  jestClient = jestService.getJestClient();  }  @After  public void tearDown() throws Exception {  jestService.closeJestClient(jestClient);  }  @Test  public void createIndex() throws Exception {  boolean result = jestService.createIndex(jestClient, indexName);  System.out.println(result);  }  @Test  public void createIndexMapping() throws Exception {  String source = "{\"" + typeName + "\":{\"properties\":{"  + "\"id\":{\"type\":\"integer\"}"  + ",\"name\":{\"type\":\"string\",\"index\":\"not_analyzed\"}"  + ",\"birth\":{\"type\":\"date\",\"format\":\"strict_date_optional_time||epoch_millis\"}"  + "}}}";  System.out.println(source);  boolean result = jestService.createIndexMapping(jestClient, indexName, typeName, source);  System.out.println(result);  }  @Test  public void getIndexMapping() throws Exception {  String result = jestService.getIndexMapping(jestClient, indexName, typeName);  System.out.println(result);  }  @Test  public void index() throws Exception {  List<Object> objs = new ArrayList<Object>();  objs.add(new User(1, "T:o\"m-", new Date()));  objs.add(new User(2, "J,e{r}r;y:", new Date()));  boolean result = jestService.index(jestClient, indexName, typeName, objs);  System.out.println(result);  }  @Test  public void termQuery() throws Exception {  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();  QueryBuilder queryBuilder = QueryBuilders  .termQuery("name", "T:o\"m-");//单值完全匹配查询  searchSourceBuilder.query(queryBuilder);  searchSourceBuilder.size(10);  searchSourceBuilder.from(0);  String query = searchSourceBuilder.toString();   System.out.println(query);  SearchResult result = jestService.search(jestClient, indexName, typeName, query);  List<Hit<User, Void>> hits = result.getHits(User.class);  System.out.println("Size:" + hits.size());  for (Hit<User, Void> hit : hits) {  User user = hit.source;  System.out.println(user.toString());  }  }  @Test  public void termsQuery() throws Exception {  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();  QueryBuilder queryBuilder = QueryBuilders  .termsQuery("name", new String[]{ "T:o\"m-", "J,e{r}r;y:" });//多值完全匹配查询  searchSourceBuilder.query(queryBuilder);  searchSourceBuilder.size(10);  searchSourceBuilder.from(0);  String query = searchSourceBuilder.toString();   System.out.println(query);  SearchResult result = jestService.search(jestClient, indexName, typeName, query);  List<Hit<User, Void>> hits = result.getHits(User.class);  System.out.println("Size:" + hits.size());  for (Hit<User, Void> hit : hits) {  User user = hit.source;  System.out.println(user.toString());  }  }  @Test  public void wildcardQuery() throws Exception {  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();  QueryBuilder queryBuilder = QueryBuilders  .wildcardQuery("name", "*:*");//通配符和正则表达式查询  searchSourceBuilder.query(queryBuilder);  searchSourceBuilder.size(10);  searchSourceBuilder.from(0);  String query = searchSourceBuilder.toString();      System.out.println(query);  SearchResult result = jestService.search(jestClient, indexName, typeName, query);  List<Hit<User, Void>> hits = result.getHits(User.class);  System.out.println("Size:" + hits.size());  for (Hit<User, Void> hit : hits) {  User user = hit.source;  System.out.println(user.toString());  }  }  @Test  public void prefixQuery() throws Exception {  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();  QueryBuilder queryBuilder = QueryBuilders  .prefixQuery("name", "T:o");//前缀查询  searchSourceBuilder.query(queryBuilder);  searchSourceBuilder.size(10);  searchSourceBuilder.from(0);  String query = searchSourceBuilder.toString();      System.out.println(query);  SearchResult result = jestService.search(jestClient, indexName, typeName, query);  List<Hit<User, Void>> hits = result.getHits(User.class);  System.out.println("Size:" + hits.size());  for (Hit<User, Void> hit : hits) {  User user = hit.source;  System.out.println(user.toString());  }  }  @Test  public void rangeQuery() throws Exception {  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();  QueryBuilder queryBuilder = QueryBuilders  .rangeQuery("birth")  .gte("2016-09-01T00:00:00")  .lte("2016-10-01T00:00:00")  .includeLower(true)  .includeUpper(true);//区间查询  searchSourceBuilder.query(queryBuilder);  searchSourceBuilder.size(10);  searchSourceBuilder.from(0);  String query = searchSourceBuilder.toString();  System.out.println(query);  SearchResult result = jestService.search(jestClient, indexName, typeName, query);  List<Hit<User, Void>> hits = result.getHits(User.class);  System.out.println("Size:" + hits.size());  for (Hit<User, Void> hit : hits) {  User user = hit.source;  System.out.println(user.toString());  }  }  @Test  public void queryString() throws Exception {  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();  QueryBuilder queryBuilder = QueryBuilders  .queryString(QueryParser.escape("T:o\""));//文本检索，应该是将查询的词先分成词库中存在的词，然后分别去检索，存在任一存在的词即返回，查询词分词后是OR的关系。需要转义特殊字符  searchSourceBuilder.query(queryBuilder);  searchSourceBuilder.size(10);  searchSourceBuilder.from(0);  String query = searchSourceBuilder.toString();   System.out.println(query);  SearchResult result = jestService.search(jestClient, indexName, typeName, query);  List<Hit<User, Void>> hits = result.getHits(User.class);  System.out.println("Size:" + hits.size());  for (Hit<User, Void> hit : hits) {  User user = hit.source;  System.out.println(user.toString());  }  }  @Test  public void count() throws Exception {  String[] name = new String[]{ "T:o\"m-", "Jerry" };  String from = "2016-09-01T00:00:00";  String to = "2016-10-01T00:00:00";  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();  QueryBuilder queryBuilder = QueryBuilders.boolQuery()  .must(QueryBuilders.termsQuery("name", name))  .must(QueryBuilders.rangeQuery("birth").gte(from).lte(to));  searchSourceBuilder.query(queryBuilder);  String query = searchSourceBuilder.toString();   System.out.println(query);  Double count = jestService.count(jestClient, indexName, typeName, query);  System.out.println("Count:" + count);  }  @Test  public void get() throws Exception {  String id = "2";  JestResult result = jestService.get(jestClient, indexName, typeName, id);  if (result.isSucceeded()) {  User user = result.getSourceAsObject(User.class);  System.out.println(user.toString());  }  }  @Test  public void deleteIndexDocument() throws Exception {  String id = "2";  boolean result = jestService.delete(jestClient, indexName, typeName, id);  System.out.println(result);  }  @Test  public void deleteIndex() throws Exception {  boolean result = jestService.delete(jestClient, indexName);  System.out.println(result);  }
}