1.项目地址

https://github.com/GuardFTC/elasticsearch-test.git

2.创建实体类

  1. import lombok.Data;
  3. import java.util.List;
  5. /**
  6.  * @author: 冯铁城 [17615007230@163.com]
  7.  * @date: 2022-08-22 15:54:07
  8.  * @describe: 查询测试使用Student实体类
  9.  */
  10. @Data
  11. public class SearchStudent {
  13.     /**
  14.      * 名称
  15.      */
  16.     private String name;
  18.     /**
  19.      * 年龄
  20.      */
  21.     private Integer age;
  23.     /**
  24.      * 成绩
  25.      */
  26.     private Double grade;
  28.     /**
  29.      * 描述
  30.      */
  31.     private String des;
  33.     /**
  34.      * 标签
  35.      */
  36.     private List<String> tags;
  37. }

3.查询Api命令整合

term

  1. @Test
  2. void term() throws IOException {
  4.     //1.查询名称为宋可心的同学
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.constantScore(c -> c.filter(f -> f.term(t -> t
  8.                     .field("name")
  9.                     .value(FieldValue.of("宋可心"))
  10.             )))), SearchStudent.class
  11.     );
  13.     //2.验证
  14.     Assert.isTrue(1 == search.hits().hits().size());
  15.     Assert.isTrue("宋可心".equals(search.hits().hits().get(0).source().getName()));
  16. }

terms

  1. @Test
  2. void terms() throws IOException {
  4.     //1.查询名称为宋可心和王乾雯的同学
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.constantScore(c -> c.filter(f -> f.terms(t -> t
  8.                     .field("name")
  9.                     .terms(terms -> terms.value(CollUtil.newArrayList(
  10.                             FieldValue.of("宋可心"),
  11.                             FieldValue.of("王乾雯")
  12.                     )))
  13.             )))), SearchStudent.class
  14.     );
  16.     //2.校验
  17.     Assert.isTrue(2 == search.hits().hits().size());
  18.     List<String> names = CollUtil.newArrayList(
  19.             search.hits().hits().get(0).source().getName(),
  20.             search.hits().hits().get(1).source().getName()
  21.     );
  22.     Assert.isTrue(names.contains("王乾雯"));
  23.     Assert.isTrue(names.contains("宋可心"));
  24. }

range

  1. @Test
  2. void range() throws IOException {
  4.     //1.查询年龄小于18岁的同学
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.range(r -> r
  8.                     .field("age")
  9.                     .lt(JsonData.of(18))
  10.             )), SearchStudent.class
  11.     );
  12.     Assert.isTrue(1 == search.hits().hits().size());
  13.     Assert.isTrue(17 == search.hits().hits().get(0).source().getAge());
  15.     //2.查询年龄大于17，小于23的同学
  16.     search = primaryClient.search(s -> s
  17.             .index(INDEX_NAME)
  18.             .query(q -> q.range(r -> r
  19.                     .field("age")
  20.                     .gt(JsonData.of(17))
  21.                     .lt(JsonData.of(23))
  22.             )), SearchStudent.class
  23.     );
  24.     Assert.isTrue(4 == search.hits().hits().size());
  25. }

exist

  1. @Test
  2. void exist() throws IOException {
  4.     //1.判定文档不包含对应字段
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.constantScore(c -> c.filter(f -> f.exists(e -> e
  8.                     .field("qqqqq")
  9.             )))), SearchStudent.class
  10.     );
  11.     Assert.isTrue(0 == search.hits().hits().size());
  13.     //2.判定文档包含对应字段
  14.     search = primaryClient.search(s -> s
  15.             .index(INDEX_NAME)
  16.             .query(q -> q.constantScore(c -> c.filter(f -> f.exists(e -> e
  17.                     .field("age")
  18.             )))), SearchStudent.class
  19.     );
  20.     Assert.isTrue(7 == search.hits().hits().size());
  21. }

match

  1. @Test
  2. void match() throws IOException {
  4.     //1.查询年龄为17的同学
  5.     SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.constantScore(c -> c.filter(f -> f.match(m -> m
  8.                     .field("age")
  9.                     .query(query -> query.longValue(17))
  10.             )))), SearchStudent.class
  11.     );
  12.     Assert.isTrue(1 == search.hits().hits().size());
  13.     Assert.isTrue(17 == search.hits().hits().get(0).source().getAge());
  15.     //2.查询描述中包含性感的同学
  16.     search = secondaryClient.search(s -> s
  17.             .index(INDEX_NAME)
  18.             .query(q -> q.match(m -> m
  19.                     .field("des")
  20.                     .query(query -> query.stringValue("性感"))
  21.             )), SearchStudent.class
  22.     );
  23.     Assert.isTrue(2 == search.hits().hits().size());
  24.     Assert.isTrue(search.hits().hits().get(0).source().getDes().contains("性感"));
  25.     Assert.isTrue(search.hits().hits().get(1).source().getDes().contains("性感"));
  27.     //3.查询描述中包含性感小姐姐的同学
  28.     search = secondaryClient.search(s -> s
  29.             .index(INDEX_NAME)
  30.             .query(q -> q.match(m -> m
  31.                     .field("des")
  32.                     .query(query -> query.stringValue("性感小姐姐"))
  33.                     .operator(Operator.And)
  34.             )), SearchStudent.class
  35.     );
  36.     Assert.isTrue(2 == search.hits().hits().size());
  37.     Assert.isTrue(
  38.             search.hits().hits().get(0).source().getDes().contains("性感") ||
  39.                     search.hits().hits().get(0).source().getDes().contains("小姐") ||
  40.                     search.hits().hits().get(0).source().getDes().contains("姐姐")
  41.     );
  42.     Assert.isTrue(
  43.             search.hits().hits().get(1).source().getDes().contains("性感") ||
  44.                     search.hits().hits().get(1).source().getDes().contains("小姐") ||
  45.                     search.hits().hits().get(1).source().getDes().contains("姐姐")
  46.     );
  48.     //4.查询描述中包含性感小姐姐的同学,匹配百分比为100%
  49.     search = secondaryClient.search(s -> s
  50.             .index(INDEX_NAME)
  51.             .query(q -> q.match(m -> m
  52.                     .field("des")
  53.                     .query(query -> query.stringValue("性感小姐姐"))
  54.                     .minimumShouldMatch("100%")
  55.             )), SearchStudent.class
  56.     );
  57.     Assert.isTrue(2 == search.hits().hits().size());
  58.     Assert.isTrue(
  59.             search.hits().hits().get(0).source().getDes().contains("性感") ||
  60.                     search.hits().hits().get(0).source().getDes().contains("小姐") ||
  61.                     search.hits().hits().get(0).source().getDes().contains("姐姐")
  62.     );
  63.     Assert.isTrue(
  64.             search.hits().hits().get(1).source().getDes().contains("性感") ||
  65.                     search.hits().hits().get(1).source().getDes().contains("小姐") ||
  66.                     search.hits().hits().get(1).source().getDes().contains("姐姐")
  67.     );
  68. }

multiMatch

  1. @Test
  2. void multiMatch() throws IOException {
  4.     //1.查询描述中包含性感或标签中包含性感的同学
  5.     SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.multiMatch(m -> m
  8.                     .query("性感")
  9.                     .fields("des", "tags")
  10.             )), SearchStudent.class
  11.     );
  12.     Assert.isTrue(4 == search.hits().hits().size());
  13.     Assert.isTrue(
  14.             search.hits().hits().get(0).source().getDes().contains("性感") ||
  15.                     JSONUtil.toJsonStr(search.hits().hits().get(0).source().getTags()).contains("性感")
  16.     );
  17.     Assert.isTrue(
  18.             search.hits().hits().get(1).source().getDes().contains("性感") ||
  19.                     JSONUtil.toJsonStr(search.hits().hits().get(1).source().getTags()).contains("性感")
  20.     );
  21.     Assert.isTrue(
  22.             search.hits().hits().get(2).source().getDes().contains("性感") ||
  23.                     JSONUtil.toJsonStr(search.hits().hits().get(2).source().getTags()).contains("性感")
  24.     );
  25.     Assert.isTrue(
  26.             search.hits().hits().get(3).source().getDes().contains("性感") ||
  27.                     JSONUtil.toJsonStr(search.hits().hits().get(3).source().getTags()).contains("性感")
  28.     );
  29. }

matchPhrase

  1. @Test
  2. void matchPhrase() throws IOException {
  4.     //1.查询描述中包含好看的小姐姐的同学
  5.     SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.matchPhrase(m -> m
  8.                     .field("des")
  9.                     .query("好看的小姐姐")
  10.             )), SearchStudent.class
  11.     );
  12.     Assert.isTrue(2 == search.hits().hits().size());
  13.     Assert.isTrue(search.hits().hits().get(0).source().getDes().contains("好看的小姐姐"));
  14.     Assert.isTrue(search.hits().hits().get(1).source().getDes().contains("好看的小姐姐"));
  16.     //2.查询描述中包含好看的小姐姐的同学,步长为20
  17.     search = secondaryClient.search(s -> s
  18.             .index(INDEX_NAME)
  19.             .query(q -> q.matchPhrase(m -> m
  20.                     .field("des")
  21.                     .query("好看的小姐姐")
  22.                     .slop(20)
  23.             )), SearchStudent.class
  24.     );
  25.     Assert.isTrue(3 == search.hits().hits().size());
  26. }

prefix

  1. @Test
  2. void prefix() throws IOException {
  4.     //1.查询所有名称为王的同学
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.constantScore(c -> c.filter(f -> f.prefix(p -> p
  8.                     .field("name")
  9.                     .value("王")
  10.             )))), SearchStudent.class
  11.     );
  13.     //2.校验
  14.     Assert.isTrue(3 == search.hits().hits().size());
  15.     Assert.isTrue(search.hits().hits().get(0).source().getName().startsWith("王"));
  16.     Assert.isTrue(search.hits().hits().get(1).source().getName().startsWith("王"));
  17.     Assert.isTrue(search.hits().hits().get(2).source().getName().startsWith("王"));
  18. }

wildcard

  1. @Test
  2. void wildcard() throws IOException {
  4.     //1.查询所有名称为王的同学
  5.     SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.wildcard(w -> w
  8.                     .field("des")
  9.                     .value("好看")
  10.             )), SearchStudent.class
  11.     );
  13.     //2.校验
  14.     Assert.isTrue(3 == search.hits().hits().size());
  15.     Assert.isTrue(search.hits().hits().get(0).source().getDes().contains("好看"));
  16.     Assert.isTrue(search.hits().hits().get(1).source().getDes().contains("好看"));
  17.     Assert.isTrue(search.hits().hits().get(2).source().getDes().contains("好看"));
  18. }

must

  1. @Test
  2. void must() throws IOException {
  4.     //1.查询姓氏为王并且年龄为18的同学
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.bool(b -> b
  8.                     .must(m -> m.prefix(p -> p
  9.                             .field("name")
  10.                             .value("王")
  11.                     ))
  12.                     .must(m -> m.term(t -> t
  13.                             .field("age")
  14.                             .value(FieldValue.of(18))
  15.                     ))
  16.             )), SearchStudent.class
  17.     );
  19.     //2.校验
  20.     Assert.isTrue(1 == search.hits().hits().size());
  21.     Assert.isTrue(search.hits().hits().get(0).source().getName().startsWith("王"));
  22.     Assert.isTrue(18 == search.hits().hits().get(0).source().getAge());
  23. }

mustNot

  1. @Test
  2. void mustNot() throws IOException {
  4.     //1.查询姓氏不为王并且年龄不为18的同学
  5.     SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.bool(b -> b
  8.                     .mustNot(m -> m.prefix(p -> p
  9.                             .field("name")
  10.                             .value("王")
  11.                     ))
  12.                     .mustNot(m -> m.term(t -> t
  13.                             .field("age")
  14.                             .value(FieldValue.of(18))
  15.                     ))
  16.             )), SearchStudent.class
  17.     );
  19.     //2.校验
  20.     Assert.isTrue(3 == search.hits().hits().size());
  21.     Assert.isTrue(!search.hits().hits().get(0).source().getName().startsWith("王"));
  22.     Assert.isTrue(search.hits().hits().get(0).source().getAge() != 18);
  23.     Assert.isTrue(!search.hits().hits().get(1).source().getName().startsWith("王"));
  24.     Assert.isTrue(search.hits().hits().get(1).source().getAge() != 18);
  25.     Assert.isTrue(!search.hits().hits().get(2).source().getName().startsWith("王"));
  26.     Assert.isTrue(search.hits().hits().get(2).source().getAge() != 18);
  27. }

shouldWithOutMust

  1. @Test
  2. void shouldWithoutMust() throws IOException {
  4.     //1.查询年龄为18或标签中包含性感的同学
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.bool(b -> b
  8.                     .should(should -> should.term(t -> t
  9.                             .field("age")
  10.                             .value(FieldValue.of(18))
  11.                     ))
  12.                     .should(should -> should.term(t -> t
  13.                             .field("tags")
  14.                             .value(FieldValue.of("性感"))
  15.                     ))
  16.             )), SearchStudent.class
  17.     );
  19.     //2.校验
  20.     Assert.isTrue(4 == search.hits().hits().size());
  21.     Assert.isTrue(
  22.             search.hits().hits().get(0).source().getAge() == 18 ||
  23.                     search.hits().hits().get(0).source().getTags().contains("性感")
  24.     );
  25.     Assert.isTrue(
  26.             search.hits().hits().get(1).source().getAge() == 18 ||
  27.                     search.hits().hits().get(1).source().getTags().contains("性感")
  28.     );
  29.     Assert.isTrue(
  30.             search.hits().hits().get(2).source().getAge() == 18 ||
  31.                     search.hits().hits().get(2).source().getTags().contains("性感")
  32.     );
  33.     Assert.isTrue(
  34.             search.hits().hits().get(3).source().getAge() == 18 ||
  35.                     search.hits().hits().get(3).source().getTags().contains("性感")
  36.     );
  37. }

shouldWithMust

  1. @Test
  2. void shouldWithMust() throws IOException {
  4.     //1.查询描述匹配好看的小姐姐短语且年龄为18的同学，同时标签中包含风骚的同学优先级高
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.bool(b -> b
  8.                     .must(m -> m.match(match -> match
  9.                             .field("des")
  10.                             .query(query -> query.stringValue("好看的小姐姐"))
  11.                     ))
  12.                     .must(m -> m.term(t -> t
  13.                             .field("age")
  14.                             .value(FieldValue.of(18))
  15.                     ))
  16.                     .should(should -> should.term(t -> t
  17.                             .field("tags")
  18.                             .value(FieldValue.of("风骚"))
  19.                     ))
  20.             )), SearchStudent.class
  21.     );
  23.     //2.校验
  24.     Assert.isTrue(2 == search.hits().hits().size());
  25.     Assert.isTrue(search.hits().hits().get(0).source().getTags().contains("风骚"));
  26.     Assert.isTrue(!search.hits().hits().get(1).source().getTags().contains("风骚"));
  27. }

mustAndShould

  1. @Test
  2. void mustAndShould() throws IOException {
  4.     //1.查询姓氏为王，并且标签包含性感或可爱的同学
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.bool(b -> b
  8.                     .filter(f -> f.prefix(p -> p
  9.                             .field("name")
  10.                             .value("王")
  11.                     ))
  12.                     .filter(f -> f.bool(fb -> fb.should(fbs -> fbs.terms(t -> t
  13.                             .field("tags")
  14.                             .terms(tt -> tt.value(CollUtil.newArrayList(
  15.                                     FieldValue.of("性感"),
  16.                                     FieldValue.of("可爱")
  17.                             )))
  18.                     ))))
  19.             )), SearchStudent.class
  20.     );
  22.     //2.校验
  23.     Assert.isTrue(2 == search.hits().hits().size());
  24.     Assert.isTrue(search.hits().hits().get(0).source().getName().startsWith("王"));
  25.     Assert.isTrue(
  26.             search.hits().hits().get(0).source().getTags().contains("性感") ||
  27.                     search.hits().hits().get(0).source().getTags().contains("可爱")
  28.     );
  29.     Assert.isTrue(search.hits().hits().get(1).source().getName().startsWith("王"));
  30.     Assert.isTrue(
  31.             search.hits().hits().get(1).source().getTags().contains("性感") ||
  32.                     search.hits().hits().get(1).source().getTags().contains("可爱")
  33.     );
  35.     //3.查询年龄为18或标签中包含长腿和性感的同学
  36.     search = primaryClient.search(s -> s
  37.             .index(INDEX_NAME)
  38.             .query(q -> q.bool(b -> b
  39.                     .should(should -> should.term(t -> t
  40.                             .field("age")
  41.                             .value(FieldValue.of(18))
  42.                     ))
  43.                     .should(should -> should.bool(sb -> sb.must(sbm -> sbm.terms(t -> t
  44.                             .field("tags")
  45.                             .terms(terms -> terms.value(CollUtil.newArrayList(
  46.                                     FieldValue.of("长腿"),
  47.                                     FieldValue.of("性感")
  48.                             )))
  49.                     ))))
  50.             )), SearchStudent.class
  51.     );
  53.     //4.校验
  54.     Assert.isTrue(4 == search.hits().hits().size());
  55.     Assert.isTrue(
  56.             18 == search.hits().hits().get(0).source().getAge() || (
  57.                     search.hits().hits().get(0).source().getTags().contains("长腿") &&
  58.                             search.hits().hits().get(0).source().getTags().contains("性感")
  59.             ));
  60.     Assert.isTrue(
  61.             18 == search.hits().hits().get(1).source().getAge() || (
  62.                     search.hits().hits().get(1).source().getTags().contains("长腿") &&
  63.                             search.hits().hits().get(1).source().getTags().contains("性感")
  64.             ));
  65.     Assert.isTrue(
  66.             18 == search.hits().hits().get(2).source().getAge() || (
  67.                     search.hits().hits().get(2).source().getTags().contains("长腿") &&
  68.                             search.hits().hits().get(2).source().getTags().contains("性感")
  69.             ));
  70.     Assert.isTrue(
  71.             18 == search.hits().hits().get(3).source().getAge() || (
  72.                     search.hits().hits().get(3).source().getTags().contains("长腿") &&
  73.                             search.hits().hits().get(3).source().getTags().contains("性感")
  74.             ));
  75. }

matchAll

  1. @Test
  2. void matchAll() throws IOException {
  4.     //1.全量查询
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .query(q -> q.matchAll(m -> m)), SearchStudent.class
  8.     );
  10.     //2.校验
  11.     Assert.isTrue(7 == search.hits().hits().size());
  12. }

sort

  1. @Test
  2. void sort() throws IOException {
  4.     //1.按照年龄排序
  5.     SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  6.                     .index(INDEX_NAME)
  7.                     .sort(sort -> sort.field(f -> f.field("age").order(SortOrder.Asc)))
  8.                     .sort(sort -> sort.field(f -> f.field("grade").order(SortOrder.Asc))),
  9.             SearchStudent.class
  10.     );
  12.     //2.验证
  13.     Assert.isTrue(7 == search.hits().hits().size());
  14.     Assert.isTrue(17 == search.hits().hits().get(0).source().getAge());
  15.     Assert.isTrue(28 == search.hits().hits().get(6).source().getAge());
  16.     Assert.isTrue(16.7 == search.hits().hits().get(1).source().getGrade());
  17.     Assert.isTrue(22.7 == search.hits().hits().get(2).source().getGrade());
  18. }

fromSize

  1. @Test
  2. void fromSize() throws IOException {
  4.     //1.每页3个，查询第3页数据
  5.     SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  6.             .index(INDEX_NAME)
  7.             .from(6)
  8.             .size(3), SearchStudent.class
  9.     );
  11.     //2.验证
  12.     Assert.isTrue(1 == search.hits().hits().size());
  13. }

scroll

  1. @Test
  2. void scroll() throws IOException {
  4.     //1.定义结果集
  5.     List<SearchStudent> students = CollUtil.newArrayList();
  7.     //2.查询快照数据
  8.     SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  9.             .index(INDEX_NAME)
  10.             .scroll(scroll -> scroll.time("10m"))
  11.             .sort(sort -> sort.field(f -> f
  12.                     .field("grade")
  13.                     .order(SortOrder.Asc)
  14.             ))
  15.             .size(2), SearchStudent.class
  16.     );
  18.     //3.初始数据加入结果集
  19.     students.addAll(CollStreamUtil.toList(search.hits().hits(), Hit::source));
  21.     //4.定义count值
  22.     int count = Integer.MIN_VALUE;
  23.     String scrollId = search.scrollId();
  25.     //5.循环从快照获取值
  26.     while (count != 0) {
  28.         //6.查询快照数据
  29.         search = secondaryClient.scroll(s -> s
  30.                 .scrollId(scrollId)
  31.                 .scroll(scroll -> scroll.time("10m")), SearchStudent.class
  32.         );
  34.         //7.重新复制count
  35.         count = search.hits().hits().size();
  37.         //8.数据继续加入结果集
  38.         students.addAll(CollStreamUtil.toList(search.hits().hits(), Hit::source));
  39.     }
  41.     //9.校验
  42.     Assert.isTrue(7 == students.size());
  43. }

4.完整版单元测试

  1. import cn.hutool.core.collection.CollStreamUtil;
  2. import cn.hutool.core.collection.CollUtil;
  3. import cn.hutool.core.lang.Assert;
  4. import cn.hutool.core.util.ObjectUtil;
  5. import cn.hutool.json.JSONUtil;
  6. import co.elastic.clients.elasticsearch.ElasticsearchClient;
  7. import co.elastic.clients.elasticsearch._types.FieldValue;
  8. import co.elastic.clients.elasticsearch._types.Refresh;
  9. import co.elastic.clients.elasticsearch._types.SortOrder;
  10. import co.elastic.clients.elasticsearch._types.query_dsl.Operator;
  11. import co.elastic.clients.elasticsearch.core.BulkRequest;
  12. import co.elastic.clients.elasticsearch.core.BulkResponse;
  13. import co.elastic.clients.elasticsearch.core.DeleteByQueryResponse;
  14. import co.elastic.clients.elasticsearch.core.SearchResponse;
  15. import co.elastic.clients.elasticsearch.core.search.Hit;
  16. import co.elastic.clients.elasticsearch.indices.CreateIndexResponse;
  17. import co.elastic.clients.json.JsonData;
  18. import com.ftc.elasticsearchtest.entity.SearchStudent;
  19. import org.junit.jupiter.api.BeforeEach;
  20. import org.junit.jupiter.api.Test;
  21. import org.springframework.beans.factory.annotation.Qualifier;
  22. import org.springframework.boot.test.context.SpringBootTest;
  24. import javax.annotation.Resource;
  25. import java.io.IOException;
  26. import java.util.List;
  28. /**
  29.  * @author: 冯铁城 [17615007230@163.com]
  30.  * @date: 2022-08-22 15:32:17
  31.  * @describe: ElasticSearch查询单元测试
  32.  */
  33. @SpringBootTest
  34. public class ElasticSearchSearchTest {
  36.     @Resource
  37.     @Qualifier("primaryElasticsearchClient")
  38.     private ElasticsearchClient primaryClient;
  40.     @Resource
  41.     @Qualifier("secondaryElasticsearchClient")
  42.     private ElasticsearchClient secondaryClient;
  44.     /**
  45.      * 测试Mock索引名称常量
  46.      */
  47.     private static final String INDEX_NAME = "test_index";
  49.     /**
  50.      * 测试数据
  51.      */
  52.     private static final String TEST_DATA = "[{\"name\":\"王乾雯\",\"age\":\"18\",\"grade\":16.7,\"des\":\"好看性感的小姐姐\",\"tags\":[\"性感\",\"风骚\",\"很好上\"]},{\"name\":\"宋可心\",\"age\":\"18\",\"grade\":22.7,\"des\":\"她也是一个性感小姐姐\",\"tags\":[\"性感\",\"声音好听\",\"美脚\"]},{\"name\":\"郭楠\",\"age\":\"28\",\"grade\":15.2,\"des\":\"长腿小姐姐\",\"tags\":[\"腿长\"]},{\"name\":\"王可菲\",\"age\":\"17\",\"grade\":68.2,\"des\":\"清纯好看的小姐姐\",\"tags\":[\"高颜值\"]},{\"name\":\"王彦舒\",\"age\":\"22\",\"grade\":11.2,\"des\":\"闷骚的小姐姐\",\"tags\":[\"可爱\",\"闷骚\"]},{\"name\":\"宁娜\",\"age\":\"25\",\"grade\":44.4,\"des\":\"特别骚的一个小姐姐\",\"tags\":[\"骚\",\"长腿\",\"我想上她\",\"性感\"]},{\"name\":\"lulu\",\"age\":\"19\",\"grade\":16.4,\"des\":\"脚特别好看的小姐姐\",\"tags\":[\"骚\",\"长腿\",\"我想上她\",\"极品美脚\",\"性感\"]}]";
  54.     @BeforeEach
  55.     void createIndexAndAddData() throws IOException {
  57.         //1.判定索引是否存在
  58.         boolean exist = primaryClient.indices().exists(e -> e.index(INDEX_NAME)).value();
  60.         //2.不存在创建索引
  61.         if (!exist) {
  63.             //3.创建索引
  64.             CreateIndexResponse createIndexResponse = primaryClient.indices().create(c -> c
  65.                     .index(INDEX_NAME + "_1")
  66.                     .aliases(INDEX_NAME, a -> a)
  67.                     .mappings(m -> m
  68.                             .properties("name", p -> p.keyword(k -> k))
  69.                             .properties("age", p -> p.integer(in -> in))
  70.                             .properties("grade", p -> p.halfFloat(s -> s))
  71.                             .properties("des", p -> p.text(t -> t.analyzer("ik_max_word")))
  72.                             .properties("tags", p -> p.keyword(k -> k))
  73.                     )
  74.                     .settings(s -> s
  75.                             .refreshInterval(t -> t.time("1s"))
  76.                             .numberOfReplicas("1")
  77.                             .numberOfShards("3")
  78.                     )
  79.             );
  80.             Assert.isTrue(Boolean.TRUE.equals(createIndexResponse.acknowledged()));
  81.         }
  83.         //4.获取索引中文档数量
  84.         long count = primaryClient.count(c -> c.index(INDEX_NAME)).count();
  85.         if (0 != count) {
  87.             //5.文档数据不为空，全量删除
  88.             DeleteByQueryResponse deleteByQueryResponse = primaryClient.deleteByQuery(d -> d
  89.                     .index(INDEX_NAME)
  90.                     .query(q -> q.matchAll(m -> m))
  91.                     .refresh(true)
  92.             );
  93.             Assert.isTrue(ObjectUtil.isNotNull(deleteByQueryResponse.deleted()));
  94.         }
  96.         //6.准备测试数据
  97.         List<SearchStudent> students = JSONUtil.toList(TEST_DATA, SearchStudent.class);
  98.         BulkRequest.Builder builder = new BulkRequest.Builder();
  99.         builder.index(INDEX_NAME);
  100.         students.forEach(s -> builder.operations(o -> o.create(c -> c.document(s))));
  101.         builder.refresh(Refresh.True);
  103.         //7.存入数据
  104.         BulkResponse bulk = primaryClient.bulk(builder.build());
  105.         Assert.isFalse(bulk.errors());
  106.         Assert.isTrue(7 == bulk.items().size());
  107.     }
  109.     @Test
  110.     void term() throws IOException {
  112.         //1.查询名称为宋可心的同学
  113.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  114.                 .index(INDEX_NAME)
  115.                 .query(q -> q.constantScore(c -> c.filter(f -> f.term(t -> t
  116.                         .field("name")
  117.                         .value(FieldValue.of("宋可心"))
  118.                 )))), SearchStudent.class
  119.         );
  121.         //2.验证
  122.         Assert.isTrue(1 == search.hits().hits().size());
  123.         Assert.isTrue("宋可心".equals(search.hits().hits().get(0).source().getName()));
  124.     }
  126.     @Test
  127.     void terms() throws IOException {
  129.         //1.查询名称为宋可心和王乾雯的同学
  130.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  131.                 .index(INDEX_NAME)
  132.                 .query(q -> q.constantScore(c -> c.filter(f -> f.terms(t -> t
  133.                         .field("name")
  134.                         .terms(terms -> terms.value(CollUtil.newArrayList(
  135.                                 FieldValue.of("宋可心"),
  136.                                 FieldValue.of("王乾雯")
  137.                         )))
  138.                 )))), SearchStudent.class
  139.         );
  141.         //2.校验
  142.         Assert.isTrue(2 == search.hits().hits().size());
  143.         List<String> names = CollUtil.newArrayList(
  144.                 search.hits().hits().get(0).source().getName(),
  145.                 search.hits().hits().get(1).source().getName()
  146.         );
  147.         Assert.isTrue(names.contains("王乾雯"));
  148.         Assert.isTrue(names.contains("宋可心"));
  149.     }
  151.     @Test
  152.     void range() throws IOException {
  154.         //1.查询年龄小于18岁的同学
  155.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  156.                 .index(INDEX_NAME)
  157.                 .query(q -> q.range(r -> r
  158.                         .field("age")
  159.                         .lt(JsonData.of(18))
  160.                 )), SearchStudent.class
  161.         );
  162.         Assert.isTrue(1 == search.hits().hits().size());
  163.         Assert.isTrue(17 == search.hits().hits().get(0).source().getAge());
  165.         //2.查询年龄大于17，小于23的同学
  166.         search = primaryClient.search(s -> s
  167.                 .index(INDEX_NAME)
  168.                 .query(q -> q.range(r -> r
  169.                         .field("age")
  170.                         .gt(JsonData.of(17))
  171.                         .lt(JsonData.of(23))
  172.                 )), SearchStudent.class
  173.         );
  174.         Assert.isTrue(4 == search.hits().hits().size());
  175.     }
  177.     @Test
  178.     void exist() throws IOException {
  180.         //1.判定文档不包含对应字段
  181.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  182.                 .index(INDEX_NAME)
  183.                 .query(q -> q.constantScore(c -> c.filter(f -> f.exists(e -> e
  184.                         .field("qqqqq")
  185.                 )))), SearchStudent.class
  186.         );
  187.         Assert.isTrue(0 == search.hits().hits().size());
  189.         //2.判定文档包含对应字段
  190.         search = primaryClient.search(s -> s
  191.                 .index(INDEX_NAME)
  192.                 .query(q -> q.constantScore(c -> c.filter(f -> f.exists(e -> e
  193.                         .field("age")
  194.                 )))), SearchStudent.class
  195.         );
  196.         Assert.isTrue(7 == search.hits().hits().size());
  197.     }
  199.     @Test
  200.     void match() throws IOException {
  202.         //1.查询年龄为17的同学
  203.         SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  204.                 .index(INDEX_NAME)
  205.                 .query(q -> q.constantScore(c -> c.filter(f -> f.match(m -> m
  206.                         .field("age")
  207.                         .query(query -> query.longValue(17))
  208.                 )))), SearchStudent.class
  209.         );
  210.         Assert.isTrue(1 == search.hits().hits().size());
  211.         Assert.isTrue(17 == search.hits().hits().get(0).source().getAge());
  213.         //2.查询描述中包含性感的同学
  214.         search = secondaryClient.search(s -> s
  215.                 .index(INDEX_NAME)
  216.                 .query(q -> q.match(m -> m
  217.                         .field("des")
  218.                         .query(query -> query.stringValue("性感"))
  219.                 )), SearchStudent.class
  220.         );
  221.         Assert.isTrue(2 == search.hits().hits().size());
  222.         Assert.isTrue(search.hits().hits().get(0).source().getDes().contains("性感"));
  223.         Assert.isTrue(search.hits().hits().get(1).source().getDes().contains("性感"));
  225.         //3.查询描述中包含性感小姐姐的同学
  226.         search = secondaryClient.search(s -> s
  227.                 .index(INDEX_NAME)
  228.                 .query(q -> q.match(m -> m
  229.                         .field("des")
  230.                         .query(query -> query.stringValue("性感小姐姐"))
  231.                         .operator(Operator.And)
  232.                 )), SearchStudent.class
  233.         );
  234.         Assert.isTrue(2 == search.hits().hits().size());
  235.         Assert.isTrue(
  236.                 search.hits().hits().get(0).source().getDes().contains("性感") ||
  237.                         search.hits().hits().get(0).source().getDes().contains("小姐") ||
  238.                         search.hits().hits().get(0).source().getDes().contains("姐姐")
  239.         );
  240.         Assert.isTrue(
  241.                 search.hits().hits().get(1).source().getDes().contains("性感") ||
  242.                         search.hits().hits().get(1).source().getDes().contains("小姐") ||
  243.                         search.hits().hits().get(1).source().getDes().contains("姐姐")
  244.         );
  246.         //4.查询描述中包含性感小姐姐的同学,匹配百分比为100%
  247.         search = secondaryClient.search(s -> s
  248.                 .index(INDEX_NAME)
  249.                 .query(q -> q.match(m -> m
  250.                         .field("des")
  251.                         .query(query -> query.stringValue("性感小姐姐"))
  252.                         .minimumShouldMatch("100%")
  253.                 )), SearchStudent.class
  254.         );
  255.         Assert.isTrue(2 == search.hits().hits().size());
  256.         Assert.isTrue(
  257.                 search.hits().hits().get(0).source().getDes().contains("性感") ||
  258.                         search.hits().hits().get(0).source().getDes().contains("小姐") ||
  259.                         search.hits().hits().get(0).source().getDes().contains("姐姐")
  260.         );
  261.         Assert.isTrue(
  262.                 search.hits().hits().get(1).source().getDes().contains("性感") ||
  263.                         search.hits().hits().get(1).source().getDes().contains("小姐") ||
  264.                         search.hits().hits().get(1).source().getDes().contains("姐姐")
  265.         );
  266.     }
  268.     @Test
  269.     void multiMatch() throws IOException {
  271.         //1.查询描述中包含性感或标签中包含性感的同学
  272.         SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  273.                 .index(INDEX_NAME)
  274.                 .query(q -> q.multiMatch(m -> m
  275.                         .query("性感")
  276.                         .fields("des", "tags")
  277.                 )), SearchStudent.class
  278.         );
  279.         Assert.isTrue(4 == search.hits().hits().size());
  280.         Assert.isTrue(
  281.                 search.hits().hits().get(0).source().getDes().contains("性感") ||
  282.                         JSONUtil.toJsonStr(search.hits().hits().get(0).source().getTags()).contains("性感")
  283.         );
  284.         Assert.isTrue(
  285.                 search.hits().hits().get(1).source().getDes().contains("性感") ||
  286.                         JSONUtil.toJsonStr(search.hits().hits().get(1).source().getTags()).contains("性感")
  287.         );
  288.         Assert.isTrue(
  289.                 search.hits().hits().get(2).source().getDes().contains("性感") ||
  290.                         JSONUtil.toJsonStr(search.hits().hits().get(2).source().getTags()).contains("性感")
  291.         );
  292.         Assert.isTrue(
  293.                 search.hits().hits().get(3).source().getDes().contains("性感") ||
  294.                         JSONUtil.toJsonStr(search.hits().hits().get(3).source().getTags()).contains("性感")
  295.         );
  296.     }
  298.     @Test
  299.     void matchPhrase() throws IOException {
  301.         //1.查询描述中包含好看的小姐姐的同学
  302.         SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  303.                 .index(INDEX_NAME)
  304.                 .query(q -> q.matchPhrase(m -> m
  305.                         .field("des")
  306.                         .query("好看的小姐姐")
  307.                 )), SearchStudent.class
  308.         );
  309.         Assert.isTrue(2 == search.hits().hits().size());
  310.         Assert.isTrue(search.hits().hits().get(0).source().getDes().contains("好看的小姐姐"));
  311.         Assert.isTrue(search.hits().hits().get(1).source().getDes().contains("好看的小姐姐"));
  313.         //2.查询描述中包含好看的小姐姐的同学,步长为20
  314.         search = secondaryClient.search(s -> s
  315.                 .index(INDEX_NAME)
  316.                 .query(q -> q.matchPhrase(m -> m
  317.                         .field("des")
  318.                         .query("好看的小姐姐")
  319.                         .slop(20)
  320.                 )), SearchStudent.class
  321.         );
  322.         Assert.isTrue(3 == search.hits().hits().size());
  323.     }
  325.     @Test
  326.     void prefix() throws IOException {
  328.         //1.查询所有名称为王的同学
  329.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  330.                 .index(INDEX_NAME)
  331.                 .query(q -> q.constantScore(c -> c.filter(f -> f.prefix(p -> p
  332.                         .field("name")
  333.                         .value("王")
  334.                 )))), SearchStudent.class
  335.         );
  337.         //2.校验
  338.         Assert.isTrue(3 == search.hits().hits().size());
  339.         Assert.isTrue(search.hits().hits().get(0).source().getName().startsWith("王"));
  340.         Assert.isTrue(search.hits().hits().get(1).source().getName().startsWith("王"));
  341.         Assert.isTrue(search.hits().hits().get(2).source().getName().startsWith("王"));
  342.     }
  344.     @Test
  345.     void wildcard() throws IOException {
  347.         //1.查询所有名称为王的同学
  348.         SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  349.                 .index(INDEX_NAME)
  350.                 .query(q -> q.wildcard(w -> w
  351.                         .field("des")
  352.                         .value("好看")
  353.                 )), SearchStudent.class
  354.         );
  356.         //2.校验
  357.         Assert.isTrue(3 == search.hits().hits().size());
  358.         Assert.isTrue(search.hits().hits().get(0).source().getDes().contains("好看"));
  359.         Assert.isTrue(search.hits().hits().get(1).source().getDes().contains("好看"));
  360.         Assert.isTrue(search.hits().hits().get(2).source().getDes().contains("好看"));
  361.     }
  363.     @Test
  364.     void must() throws IOException {
  366.         //1.查询姓氏为王并且年龄为18的同学
  367.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  368.                 .index(INDEX_NAME)
  369.                 .query(q -> q.bool(b -> b
  370.                         .must(m -> m.prefix(p -> p
  371.                                 .field("name")
  372.                                 .value("王")
  373.                         ))
  374.                         .must(m -> m.term(t -> t
  375.                                 .field("age")
  376.                                 .value(FieldValue.of(18))
  377.                         ))
  378.                 )), SearchStudent.class
  379.         );
  381.         //2.校验
  382.         Assert.isTrue(1 == search.hits().hits().size());
  383.         Assert.isTrue(search.hits().hits().get(0).source().getName().startsWith("王"));
  384.         Assert.isTrue(18 == search.hits().hits().get(0).source().getAge());
  385.     }
  387.     @Test
  388.     void mustNot() throws IOException {
  390.         //1.查询姓氏不为王并且年龄不为18的同学
  391.         SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  392.                 .index(INDEX_NAME)
  393.                 .query(q -> q.bool(b -> b
  394.                         .mustNot(m -> m.prefix(p -> p
  395.                                 .field("name")
  396.                                 .value("王")
  397.                         ))
  398.                         .mustNot(m -> m.term(t -> t
  399.                                 .field("age")
  400.                                 .value(FieldValue.of(18))
  401.                         ))
  402.                 )), SearchStudent.class
  403.         );
  405.         //2.校验
  406.         Assert.isTrue(3 == search.hits().hits().size());
  407.         Assert.isTrue(!search.hits().hits().get(0).source().getName().startsWith("王"));
  408.         Assert.isTrue(search.hits().hits().get(0).source().getAge() != 18);
  409.         Assert.isTrue(!search.hits().hits().get(1).source().getName().startsWith("王"));
  410.         Assert.isTrue(search.hits().hits().get(1).source().getAge() != 18);
  411.         Assert.isTrue(!search.hits().hits().get(2).source().getName().startsWith("王"));
  412.         Assert.isTrue(search.hits().hits().get(2).source().getAge() != 18);
  413.     }
  415.     @Test
  416.     void shouldWithoutMust() throws IOException {
  418.         //1.查询年龄为18或标签中包含性感的同学
  419.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  420.                 .index(INDEX_NAME)
  421.                 .query(q -> q.bool(b -> b
  422.                         .should(should -> should.term(t -> t
  423.                                 .field("age")
  424.                                 .value(FieldValue.of(18))
  425.                         ))
  426.                         .should(should -> should.term(t -> t
  427.                                 .field("tags")
  428.                                 .value(FieldValue.of("性感"))
  429.                         ))
  430.                 )), SearchStudent.class
  431.         );
  433.         //2.校验
  434.         Assert.isTrue(4 == search.hits().hits().size());
  435.         Assert.isTrue(
  436.                 search.hits().hits().get(0).source().getAge() == 18 ||
  437.                         search.hits().hits().get(0).source().getTags().contains("性感")
  438.         );
  439.         Assert.isTrue(
  440.                 search.hits().hits().get(1).source().getAge() == 18 ||
  441.                         search.hits().hits().get(1).source().getTags().contains("性感")
  442.         );
  443.         Assert.isTrue(
  444.                 search.hits().hits().get(2).source().getAge() == 18 ||
  445.                         search.hits().hits().get(2).source().getTags().contains("性感")
  446.         );
  447.         Assert.isTrue(
  448.                 search.hits().hits().get(3).source().getAge() == 18 ||
  449.                         search.hits().hits().get(3).source().getTags().contains("性感")
  450.         );
  451.     }
  453.     @Test
  454.     void shouldWithMust() throws IOException {
  456.         //1.查询描述匹配好看的小姐姐短语且年龄为18的同学，同时标签中包含风骚的同学优先级高
  457.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  458.                 .index(INDEX_NAME)
  459.                 .query(q -> q.bool(b -> b
  460.                         .must(m -> m.match(match -> match
  461.                                 .field("des")
  462.                                 .query(query -> query.stringValue("好看的小姐姐"))
  463.                         ))
  464.                         .must(m -> m.term(t -> t
  465.                                 .field("age")
  466.                                 .value(FieldValue.of(18))
  467.                         ))
  468.                         .should(should -> should.term(t -> t
  469.                                 .field("tags")
  470.                                 .value(FieldValue.of("风骚"))
  471.                         ))
  472.                 )), SearchStudent.class
  473.         );
  475.         //2.校验
  476.         Assert.isTrue(2 == search.hits().hits().size());
  477.         Assert.isTrue(search.hits().hits().get(0).source().getTags().contains("风骚"));
  478.         Assert.isTrue(!search.hits().hits().get(1).source().getTags().contains("风骚"));
  479.     }
  481.     @Test
  482.     void mustAndShould() throws IOException {
  484.         //1.查询姓氏为王，并且标签包含性感或可爱的同学
  485.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  486.                 .index(INDEX_NAME)
  487.                 .query(q -> q.bool(b -> b
  488.                         .filter(f -> f.prefix(p -> p
  489.                                 .field("name")
  490.                                 .value("王")
  491.                         ))
  492.                         .filter(f -> f.bool(fb -> fb.should(fbs -> fbs.terms(t -> t
  493.                                 .field("tags")
  494.                                 .terms(tt -> tt.value(CollUtil.newArrayList(
  495.                                         FieldValue.of("性感"),
  496.                                         FieldValue.of("可爱")
  497.                                 )))
  498.                         ))))
  499.                 )), SearchStudent.class
  500.         );
  502.         //2.校验
  503.         Assert.isTrue(2 == search.hits().hits().size());
  504.         Assert.isTrue(search.hits().hits().get(0).source().getName().startsWith("王"));
  505.         Assert.isTrue(
  506.                 search.hits().hits().get(0).source().getTags().contains("性感") ||
  507.                         search.hits().hits().get(0).source().getTags().contains("可爱")
  508.         );
  509.         Assert.isTrue(search.hits().hits().get(1).source().getName().startsWith("王"));
  510.         Assert.isTrue(
  511.                 search.hits().hits().get(1).source().getTags().contains("性感") ||
  512.                         search.hits().hits().get(1).source().getTags().contains("可爱")
  513.         );
  515.         //3.查询年龄为18或标签中包含长腿和性感的同学
  516.         search = primaryClient.search(s -> s
  517.                 .index(INDEX_NAME)
  518.                 .query(q -> q.bool(b -> b
  519.                         .should(should -> should.term(t -> t
  520.                                 .field("age")
  521.                                 .value(FieldValue.of(18))
  522.                         ))
  523.                         .should(should -> should.bool(sb -> sb.must(sbm -> sbm.terms(t -> t
  524.                                 .field("tags")
  525.                                 .terms(terms -> terms.value(CollUtil.newArrayList(
  526.                                         FieldValue.of("长腿"),
  527.                                         FieldValue.of("性感")
  528.                                 )))
  529.                         ))))
  530.                 )), SearchStudent.class
  531.         );
  533.         //4.校验
  534.         Assert.isTrue(4 == search.hits().hits().size());
  535.         Assert.isTrue(
  536.                 18 == search.hits().hits().get(0).source().getAge() || (
  537.                         search.hits().hits().get(0).source().getTags().contains("长腿") &&
  538.                                 search.hits().hits().get(0).source().getTags().contains("性感")
  539.                 ));
  540.         Assert.isTrue(
  541.                 18 == search.hits().hits().get(1).source().getAge() || (
  542.                         search.hits().hits().get(1).source().getTags().contains("长腿") &&
  543.                                 search.hits().hits().get(1).source().getTags().contains("性感")
  544.                 ));
  545.         Assert.isTrue(
  546.                 18 == search.hits().hits().get(2).source().getAge() || (
  547.                         search.hits().hits().get(2).source().getTags().contains("长腿") &&
  548.                                 search.hits().hits().get(2).source().getTags().contains("性感")
  549.                 ));
  550.         Assert.isTrue(
  551.                 18 == search.hits().hits().get(3).source().getAge() || (
  552.                         search.hits().hits().get(3).source().getTags().contains("长腿") &&
  553.                                 search.hits().hits().get(3).source().getTags().contains("性感")
  554.                 ));
  555.     }
  557.     @Test
  558.     void matchAll() throws IOException {
  560.         //1.全量查询
  561.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  562.                 .index(INDEX_NAME)
  563.                 .query(q -> q.matchAll(m -> m)), SearchStudent.class
  564.         );
  566.         //2.校验
  567.         Assert.isTrue(7 == search.hits().hits().size());
  568.     }
  570.     @Test
  571.     void sort() throws IOException {
  573.         //1.按照年龄排序
  574.         SearchResponse<SearchStudent> search = primaryClient.search(s -> s
  575.                         .index(INDEX_NAME)
  576.                         .sort(sort -> sort.field(f -> f.field("age").order(SortOrder.Asc)))
  577.                         .sort(sort -> sort.field(f -> f.field("grade").order(SortOrder.Asc))),
  578.                 SearchStudent.class
  579.         );
  581.         //2.验证
  582.         Assert.isTrue(7 == search.hits().hits().size());
  583.         Assert.isTrue(17 == search.hits().hits().get(0).source().getAge());
  584.         Assert.isTrue(28 == search.hits().hits().get(6).source().getAge());
  585.         Assert.isTrue(16.7 == search.hits().hits().get(1).source().getGrade());
  586.         Assert.isTrue(22.7 == search.hits().hits().get(2).source().getGrade());
  587.     }
  589.     @Test
  590.     void fromSize() throws IOException {
  592.         //1.每页3个，查询第3页数据
  593.         SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  594.                 .index(INDEX_NAME)
  595.                 .from(6)
  596.                 .size(3), SearchStudent.class
  597.         );
  599.         //2.验证
  600.         Assert.isTrue(1 == search.hits().hits().size());
  601.     }
  603.     @Test
  604.     void scroll() throws IOException {
  606.         //1.定义结果集
  607.         List<SearchStudent> students = CollUtil.newArrayList();
  609.         //2.查询快照数据
  610.         SearchResponse<SearchStudent> search = secondaryClient.search(s -> s
  611.                 .index(INDEX_NAME)
  612.                 .scroll(scroll -> scroll.time("10m"))
  613.                 .sort(sort -> sort.field(f -> f
  614.                         .field("grade")
  615.                         .order(SortOrder.Asc)
  616.                 ))
  617.                 .size(2), SearchStudent.class
  618.         );
  620.         //3.初始数据加入结果集
  621.         students.addAll(CollStreamUtil.toList(search.hits().hits(), Hit::source));
  623.         //4.定义count值
  624.         int count = Integer.MIN_VALUE;
  625.         String scrollId = search.scrollId();
  627.         //5.循环从快照获取值
  628.         while (count != 0) {
  630.             //6.查询快照数据
  631.             search = secondaryClient.scroll(s -> s
  632.                     .scrollId(scrollId)
  633.                     .scroll(scroll -> scroll.time("10m")), SearchStudent.class
  634.             );
  636.             //7.重新复制count
  637.             count = search.hits().hits().size();
  639.             //8.数据继续加入结果集
  640.             students.addAll(CollStreamUtil.toList(search.hits().hits(), Hit::source));
  641.         }
  643.         //9.校验
  644.         Assert.isTrue(7 == students.size());
  645.     }
  646. }