elasticsearch整合Spring用例

package cn.itcast.service.test; import java.util.ArrayList; import java.util.List; import org.elasticsearch.client.Client; import org.elasticsearch.index.query.BoolQueryBuilder; import org.elasticsearch.index.query.QueryBuilder; import org.elasticsearch.index.query.QueryStringQueryBuilder; import org.elasticsearch.index.query.TermQueryBuilder; import org.elasticsearch.index.query.WildcardQueryBuilder; import org.elasticsearch.index.query.QueryStringQueryBuilder.Operator; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.domain.Page; import org.springframework.data.domain.PageRequest; import org.springframework.data.domain.Sort; import org.springframework.data.domain.Pageable; import org.springframework.data.elasticsearch.core.ElasticsearchTemplate; import org.springframework.data.elasticsearch.core.query.NativeSearchQuery; import org.springframework.data.elasticsearch.core.query.SearchQuery; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import cn.itcast.domain.Article; import cn.itcast.service.ArticleService; @RunWith(SpringJUnit4ClassRunner.class) @ContextConfiguration(locations = "classpath:applicationContext.xml") public class ArticleServiceTest { @Autowired private ArticleService articleService; @Autowired private Client client; // 基于原生API @Autowired private ElasticsearchTemplate elasticsearchTemplate; @Test public void createIndex() { // elasticsearchTemplate.createIndex(Article.class); elasticsearchTemplate.putMapping(Article.class); } @Test public void testSave() { Article article = new Article(); article.setId(1001); // article.setTitle("哈哈哈Spring Data Elasticsearch 1.3.1 昨天发布"); // article.setContent("DATAES-171 - 添加失效查询关键字支持 DATAES-194 - 测试可以清理 data 目录 DATAES-179 - 支持 Attachment 字段类型 DATAES-94 " // + "- 更新到最新版本的 elasticsearch 1.7.3 驱动器"); article.setTitle("北京市海淀区建材城西路金燕龙办公楼一层"); article.setContent("DATAES-171 - 添加失效查询关键字支持 DATAES-194 - 测试可以清理 data 目录 DATAES-179 - 支持 Attachment 字段类型 DATAES-94 " + "- 更新到最新版本的 elasticsearch 1.7.3 驱动器"); article.setNum(666); article.setNum2(999L); articleService.save(article); } @Test public void testDelete() { Article article = new Article(); article.setId(1001); articleService.delete(article); } @Test public void testFindOne() { System.out.println(articleService.findOne(1001)); } @Test public void testSaveBatch() { List<Article> articles = new ArrayList<Article>(); for (int i = 1; i <= 50; i++) { Article article = new Article(); article.setId(i); article.setTitle(i + "Spring Data Elasticsearch 1.3.1 昨天发布"); article.setContent(i + "DATAES-171 - 添加失效查询关键字支持 DATAES-194 - 测试可以清理 data 目录 DATAES-179 - 支持 Attachment 字段类型 DATAES-94 - 更新到最新版本的 elasticsearch 1.7.3 驱动器"); articles.add(article); } articleService.saveBatch(articles); } @Test // 排序分页查询 public void testSortAndPaging() { // Iterable<Article> articles = articleService.findAll(); // for (Article article : articles) { // System.out.println(article); // } // Pageable pageable = new PageRequest(0, 100); Pageable pageable = new PageRequest(0, 10001,new Sort(Sort.Direction.ASC, "id")); Page<Article> pageData = articleService.findAll(pageable); for (Article article : pageData.getContent()) { System.out.println(article); } } @Test // 条件查询 public void testConditionQuery() { // 查询 标题中含有 “昨天” List<Article> articles = articleService.findByTitle("昨天"); // List<Article> articles = articleService.findByNum(666); for (Article article : articles) { System.out.println(article); } // 查询 标题中含有 “昨天” 1-10条 // Pageable pageable = new PageRequest(0, 10); // Page<Article> pageData = articleService.findByTitle("北京市海淀区建材城西路金燕龙办公楼", pageable); // System.out.println("总记录数：" + pageData.getTotalElements()); // for (Article article : pageData.getContent()) { // System.out.println(article); // } } @Test // 条件查询 public void testConditionQuery2() { // select * from t where a!=xxx // must 条件必须成立 and // must not 条件必须不成立 not // should 条件可以成立 or BoolQueryBuilder query = new BoolQueryBuilder(); // 布尔查询 ，多条件组合查询 //TermQueryBuilder查询内容要精准匹配词条 // QueryBuilder termQuery = new TermQueryBuilder("title","建"); // query.must(termQuery); //WildcardQueryBuilder查询内容要<=一个词条 // QueryBuilder wildcardQuery = new WildcardQueryBuilder("title", "建*"); // query.should(wildcardQuery); //QueryStringQueryBuilder先拆分查询内容成多个词条，这些词条要和索引库中内容做匹配，拆分成的多个词条如果用Operator.AND， //表示必须全部匹配，如果用Operator.OR，表示多个词条必须至少匹配一个 QueryBuilder queryStringQueryBuilder = new QueryStringQueryBuilder("海淀期限") //北京市海淀区建材城西路金燕龙办公楼一层 .field("title").defaultOperator(Operator.AND); query.should(queryStringQueryBuilder); SearchQuery searchQuery = new NativeSearchQuery(query); Pageable pageable = new PageRequest(0, 10); searchQuery.setPageable(pageable); // 分页效果 Page<Article> pageData = articleService.testSearch(searchQuery); System.out.println("总记录数：" + pageData.getTotalElements()); for (Article article : pageData.getContent()) { System.out.println(article); } } }