1.项目地址

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

2.文档API命令整合

创建实体类

  1. import lombok.Data;
  3. /**
  4.  * @author: 冯铁城 [17615007230@163.com]
  5.  * @date: 2022-08-22 11:16:43
  6.  * @describe: 学生类
  7.  */
  8. @Data
  9. public class Student {
  11.     /**
  12.      * 主键ID
  13.      */
  14.     private Integer id;
  16.     /**
  17.      * 名称
  18.      */
  19.     private String name;
  21.     /**
  22.      * 年龄
  23.      */
  24.     private Integer age;
  25. }

存储单个文档

使用.index方法控制

  1. @Test
  2. void createDocSingle() throws IOException {
  4.     //1.创建学生对象
  5.     Student student = new Student();
  6.     student.setId(1);
  7.     student.setName("冯铁城");
  8.     student.setAge(11);
  10.     //2.创建单个文档不指定ID
  11.     IndexResponse indexResponse = primaryClient.index(i -> i
  12.             .index(INDEX_NAME)
  13.             .document(student)
  14.     );
  15.     Result result = indexResponse.result();
  16.     Assert.isTrue("created".equals(result.jsonValue().toLowerCase(Locale.ROOT)));
  18.     //3.创建单个文档指定ID
  19.     indexResponse = primaryClient.index(i -> i
  20.             .index(INDEX_NAME)
  21.             .document(student)
  22.             .id("1")
  23.     );
  24.     result = indexResponse.result();
  25.     Assert.isTrue("created".equals(result.jsonValue().toLowerCase(Locale.ROOT)));
  27.     //4.创建单个文档进行ID唯一性控制
  28.     String errorMessage = null;
  29.     try {
  30.         primaryClient.index(i -> i
  31.                 .index(INDEX_NAME)
  32.                 .document(student)
  33.                 .id("1")
  34.                 .opType(OpType.Create)
  35.         );
  36.     } catch (Exception e) {
  37.         errorMessage = e.getMessage();
  38.     }
  39.     Assert.isTrue(ObjectUtil.isNotNull(errorMessage));
  40.     Assert.isTrue(errorMessage.contains("version_conflict_engine_exception"));
  41. }

存储多个文档

  1. @Test
  2. void createDocBulk() throws IOException {
  4.     //1.创建学生对象
  5.     Student student = new Student();
  6.     student.setId(1);
  7.     student.setName("张钰玲");
  8.     student.setAge(11);
  10.     //2.批量保存数据不带ID
  11.     BulkResponse bulk = primaryClient.bulk(b -> b
  12.             .index(INDEX_NAME)
  13.             .operations(o -> o.create(c -> c.document(student)))
  14.             .operations(o -> o.create(c -> c.document(student)))
  15.     );
  16.     Assert.isFalse(bulk.errors());
  17.     Assert.isTrue(2 == bulk.items().size());
  19.     //3.批量保存数据带ID
  20.     bulk = primaryClient.bulk(b -> b
  21.             .index(INDEX_NAME)
  22.             .operations(o -> o.create(c -> c.document(student).id("1")))
  23.             .operations(o -> o.create(c -> c.document(student).id("2")))
  24.     );
  25.     Assert.isFalse(bulk.errors());
  26.     Assert.isTrue(2 == bulk.items().size());
  27. }

查询文档

  1. @Test
  2. void getDocSingle() throws IOException {
  4.     //1.创建学生对象
  5.     Student student = new Student();
  6.     student.setId(1);
  7.     student.setName("冯铁城");
  8.     student.setAge(18);
  9.     Student studentBak = new Student();
  10.     student.setId(2);
  11.     student.setName("张钰玲");
  12.     student.setAge(16);
  14.     //2.批量保存数据
  15.     BulkResponse bulk = secondaryClient.bulk(b -> b
  16.             .index(INDEX_NAME)
  17.             .operations(o -> o.create(c -> c.document(student).id("1")))
  18.             .operations(o -> o.create(c -> c.document(studentBak).id("2")))
  19.             .refresh(Refresh.True)
  20.     );
  21.     Assert.isFalse(bulk.errors());
  22.     Assert.isTrue(2 == bulk.items().size());
  24.     //3.根据ID查询单个数据
  25.     Student source = secondaryClient.get(g -> g
  26.             .index(INDEX_NAME)
  27.             .id("1"), Student.class
  28.     ).source();
  29.     Assert.isTrue(JSONUtil.toJsonStr(student).equals(JSONUtil.toJsonStr(source)));
  31.     //4.按照ID批量查询
  32.     List<MultiGetResponseItem<Student>> docs = secondaryClient.mget(m -> m
  33.             .index(INDEX_NAME)
  34.             .ids("1", "2"), Student.class
  35.     ).docs();
  36.     Assert.isTrue(2 == docs.size());
  37.     Student source1 = docs.get(0).result().source();
  38.     Assert.isTrue(JSONUtil.toJsonStr(student).equals(JSONUtil.toJsonStr(source1)));
  39.     Student source2 = docs.get(1).result().source();
  40.     Assert.isTrue(JSONUtil.toJsonStr(studentBak).equals(JSONUtil.toJsonStr(source2)));
  41. }

判定文档是否存在

  1. @Test
  2. void existDoc() throws IOException {
  4.     //1.验证数据存在
  5.     BooleanResponse exists = secondaryClient.exists(e -> e
  6.             .index(INDEX_NAME)
  7.             .id(DEFAULT_ID)
  8.     );
  9.     Assert.isTrue(exists.value());
  11.     //2.验证数据不存在
  12.     exists = secondaryClient.exists(e -> e
  13.             .index(INDEX_NAME)
  14.             .id(DEFAULT_ID + "1")
  15.     );
  16.     Assert.isFalse(exists.value());
  17. }

修改文档单个

  1. @Test
  2. void updateDocSingle() throws IOException {
  4.     //1.查询默认数据
  5.     Student student = primaryClient.get(g -> g
  6.             .index(INDEX_NAME)
  7.             .id(DEFAULT_ID), Student.class
  8.     ).source();
  9.     Assert.isTrue(ObjectUtil.isNotNull(student));
  11.     //2.全量更新单个数据（立即refresh，为了单元测试方便）
  12.     student.setAge(100);
  13.     IndexResponse index = primaryClient.index(i -> i
  14.             .index(INDEX_NAME)
  15.             .document(student)
  16.             .id(DEFAULT_ID)
  17.             .refresh(Refresh.True)
  18.     );
  19.     Assert.isTrue("updated".equals(index.result().jsonValue().toLowerCase(Locale.ROOT)));
  21.     //3.查询校验
  22.     Integer age = primaryClient.get(g -> g
  23.             .index(INDEX_NAME)
  24.             .id(DEFAULT_ID), Student.class
  25.     ).source().getAge();
  26.     Assert.isTrue(100 == age);
  28.     //4.全量更新单个数据并进行版本控制（立即refresh，为了单元测试方便）
  29.     student.setAge(11);
  30.     long primaryTerm = index.primaryTerm();
  31.     long seqNo = index.seqNo();
  32.     index = primaryClient.index(i -> i
  33.             .index(INDEX_NAME)
  34.             .document(student)
  35.             .id(DEFAULT_ID)
  36.             .ifPrimaryTerm(primaryTerm)
  37.             .ifSeqNo(seqNo)
  38.             .refresh(Refresh.True)
  39.     );
  40.     Assert.isTrue("updated".equals(index.result().jsonValue().toLowerCase(Locale.ROOT)));
  42.     //5.查询校验
  43.     age = primaryClient.get(g -> g
  44.             .index(INDEX_NAME)
  45.             .id(DEFAULT_ID), Student.class
  46.     ).source().getAge();
  47.     Assert.isTrue(11 == age);
  49.     //6.全量更新单个数据并进行版本控制（立即refresh，为了单元测试方便）
  50.     String errorMessage = null;
  51.     try {
  52.         primaryClient.index(i -> i
  53.                 .index(INDEX_NAME)
  54.                 .document(student)
  55.                 .id(INDEX_NAME)
  56.                 .ifPrimaryTerm((long) 1)
  57.                 .ifSeqNo((long) 2)
  58.                 .refresh(Refresh.True)
  59.         );
  60.     } catch (Exception e) {
  61.         errorMessage = e.getMessage();
  62.     }
  63.     Assert.isTrue(StrUtil.isNotBlank(errorMessage));
  64.     Assert.isTrue(errorMessage.contains("version_conflict_engine_exception"));
  66.     //7.更新文档部分属性
  67.     Student studentBak = new Student();
  68.     studentBak.setAge(30);
  69.     UpdateResponse<Student> update = primaryClient.update(u -> u
  70.             .index(INDEX_NAME)
  71.             .id(DEFAULT_ID)
  72.             .doc(studentBak)
  73.             .refresh(Refresh.True), Student.class
  74.     );
  75.     Assert.isTrue("updated".equals(update.result().jsonValue().toLowerCase(Locale.ROOT)));
  77.     //8.查询校验
  78.     Student source = primaryClient.get(g -> g
  79.             .index(INDEX_NAME)
  80.             .id(DEFAULT_ID), Student.class
  81.     ).source();
  82.     Assert.isTrue(ObjectUtil.isNotNull(source));
  83.     Assert.isTrue(studentBak.getAge() == source.getAge());
  84.     Assert.isTrue(student.getName().equals(source.getName()));
  85. }

修改文档批量

  1. @Test
  2. void updateDocBulk() throws IOException {
  4.     //1.准备测试数据
  5.     Student student = new Student();
  6.     student.setName("冯铁城");
  7.     student.setAge(18);
  8.     Student studentBak = new Student();
  9.     studentBak.setName("张钰玲");
  10.     studentBak.setAge(16);
  12.     //2.存入测试数据
  13.     BulkResponse bulk = primaryClient.bulk(b -> b
  14.             .index(INDEX_NAME)
  15.             .operations(o -> o.create(c -> c.document(student).id("1")))
  16.             .operations(o -> o.create(c -> c.document(studentBak).id("2")))
  17.     );
  18.     Assert.isFalse(bulk.errors());
  19.     Assert.isTrue(2 == bulk.items().size());
  21.     //3.批量更新部分数据
  22.     Student bulkStudent = new Student();
  23.     bulkStudent.setAge(111);
  24.     bulk = primaryClient.bulk(b -> b
  25.             .index(INDEX_NAME)
  26.             .operations(o -> o.update(u -> u.action(a -> a.doc(bulkStudent)).id("1")))
  27.             .operations(o -> o.update(u -> u.action(a -> a.doc(bulkStudent)).id("2")))
  28.     );
  29.     Assert.isFalse(bulk.errors());
  30.     Assert.isTrue(2 == bulk.items().size());
  32.     //4.查询校验
  33.     Student source = primaryClient.get(g -> g
  34.             .index(INDEX_NAME)
  35.             .id("1"), Student.class
  36.     ).source();
  37.     Assert.isTrue("{\"name\":\"冯铁城\",\"age\":111}".equals(JSONUtil.toJsonStr(source)));
  38.     source = primaryClient.get(g -> g
  39.             .index(INDEX_NAME)
  40.             .id("2"), Student.class
  41.     ).source();
  42.     Assert.isTrue("{\"name\":\"张钰玲\",\"age\":111}".equals(JSONUtil.toJsonStr(source)));
  44.     //5.批量更新全量数据
  45.     Student bulkStudent2 = new Student();
  46.     bulk = primaryClient.bulk(b -> b
  47.             .index(INDEX_NAME)
  48.             .operations(o -> o.index(i -> i.document(bulkStudent2).id("1")))
  49.             .operations(o -> o.index(i -> i.document(bulkStudent2).id("2")))
  50.     );
  51.     Assert.isFalse(bulk.errors());
  52.     Assert.isTrue(2 == bulk.items().size());
  54.     //6.查询验证
  55.     source = primaryClient.get(g -> g
  56.             .index(INDEX_NAME)
  57.             .id("1"), Student.class
  58.     ).source();
  59.     Assert.isTrue("{}".equals(JSONUtil.toJsonStr(source)));
  60.     source = primaryClient.get(g -> g
  61.             .index(INDEX_NAME)
  62.             .id("2"), Student.class
  63.     ).source();
  64.     Assert.isTrue("{}".equals(JSONUtil.toJsonStr(source)));
  65. }

修改文档根据条件

  1. @Test
  2. void updateDocQuery() throws IOException {
  4.     //1.准备测试数据
  5.     Student student = new Student();
  6.     student.setName("冯铁城");
  7.     student.setAge(18);
  8.     Student studentBak = new Student();
  9.     studentBak.setName("张钰玲");
  10.     studentBak.setAge(16);
  12.     //2.存入测试数据
  13.     BulkResponse bulk = primaryClient.bulk(b -> b
  14.             .index(INDEX_NAME)
  15.             .operations(o -> o.create(c -> c.document(student).id("1")))
  16.             .operations(o -> o.create(c -> c.document(studentBak).id("2")))
  17.             .refresh(Refresh.True)
  18.     );
  19.     Assert.isFalse(bulk.errors());
  20.     Assert.isTrue(2 == bulk.items().size());
  22.     //3.更新名称为冯铁城的数据，将其年龄改为100
  23.     UpdateByQueryResponse updateByQueryResponse = primaryClient.updateByQuery(u -> u
  24.             .index(INDEX_NAME)
  25.             .query(q -> q.match(m -> m
  26.                     .field("name")
  27.                     .query(query -> query.stringValue("冯铁城"))
  28.             ))
  29.             .script(s -> s.inline(i -> i
  30.                     .source("ctx._source.age=100")
  31.             ))
  32.             .refresh(true)
  33.     );
  34.     Assert.isTrue(1 == updateByQueryResponse.updated());
  36.     //4.查询校验
  37.     Student source = primaryClient.get(g -> g
  38.             .index(INDEX_NAME)
  39.             .id("1"), Student.class
  40.     ).source();
  41.     Assert.isTrue(100 == source.getAge());
  43.     //5.更新名称为冯铁城的数据，移除age属性
  44.     updateByQueryResponse = primaryClient.updateByQuery(u -> u
  45.             .index(INDEX_NAME)
  46.             .query(q -> q.match(m -> m
  47.                     .field("name")
  48.                     .query(query -> query.stringValue("冯铁城"))
  49.             ))
  50.             .script(s -> s.inline(i -> i
  51.                     .source("ctx._source.remove('age')")
  52.             ))
  53.             .refresh(true)
  54.     );
  55.     Assert.isTrue(1 == updateByQueryResponse.updated());
  57.     //6.查询校验
  58.     source = primaryClient.get(g -> g
  59.             .index(INDEX_NAME)
  60.             .id("1"), Student.class
  61.     ).source();
  62.     Assert.isNull(source.getAge());
  63. }

删除文档单个

  1. @Test
  2. void deleteDocSingle() throws IOException {
  4.     //1.验证默认文档存在
  5.     BooleanResponse exists = secondaryClient.exists(e -> e.index(INDEX_NAME).id(DEFAULT_ID));
  6.     Assert.isTrue(exists.value());
  8.     //2.删除文档
  9.     DeleteResponse delete = secondaryClient.delete(d -> d
  10.             .index(INDEX_NAME)
  11.             .id(DEFAULT_ID)
  12.             .refresh(Refresh.True)
  13.     );
  14.     Assert.isTrue("deleted".equals(delete.result().jsonValue().toLowerCase(Locale.ROOT)));
  16.     //3.再次验证，文档已不存在
  17.     exists = secondaryClient.exists(e -> e.index(INDEX_NAME).id(DEFAULT_ID));
  18.     Assert.isFalse(exists.value());
  19. }

删除文档批量

  1. @Test
  2. void deleteDocBulk() throws IOException {
  4.     //1.存入测试数据
  5.     BulkResponse bulk = primaryClient.bulk(b -> b
  6.             .index(INDEX_NAME)
  7.             .operations(o -> o.create(c -> c.document(new Student()).id("1")))
  8.             .operations(o -> o.create(c -> c.document(new Student()).id("2")))
  9.             .refresh(Refresh.True)
  10.     );
  11.     Assert.isFalse(bulk.errors());
  13.     //2.验证文档已存在
  14.     BooleanResponse exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  15.     Assert.isTrue(exists.value());
  16.     exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  17.     Assert.isTrue(exists.value());
  19.     //3.批量删除
  20.     bulk = primaryClient.bulk(b -> b
  21.             .index(INDEX_NAME)
  22.             .operations(o -> o.delete(d -> d.id("1")))
  23.             .operations(o -> o.delete(d -> d.id("2")))
  24.             .refresh(Refresh.True)
  25.     );
  26.     Assert.isFalse(bulk.errors());
  27.     Assert.isTrue(2 == bulk.items().size());
  29.     //4.验证已不存在
  30.     exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  31.     Assert.isFalse(exists.value());
  32.     exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  33.     Assert.isFalse(exists.value());
  34. }

删除文档根据条件

  1. @Test
  2. void deleteDocQuery() throws IOException {
  4.     //1.准备测试数据
  5.     Student student1 = new Student();
  6.     student1.setAge(11);
  7.     Student student2 = new Student();
  8.     student2.setAge(18);
  10.     //2.存入测试数据
  11.     BulkResponse bulk = primaryClient.bulk(b -> b
  12.             .index(INDEX_NAME)
  13.             .operations(o -> o.create(c -> c.document(student1).id("1")))
  14.             .operations(o -> o.create(c -> c.document(student2).id("2")))
  15.             .refresh(Refresh.True)
  16.     );
  17.     Assert.isFalse(bulk.errors());
  19.     //3.校验是否存在
  20.     BooleanResponse exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  21.     Assert.isTrue(exists.value());
  22.     exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  23.     Assert.isTrue(exists.value());
  25.     //4.删除数据
  26.     DeleteByQueryResponse age = primaryClient.deleteByQuery(d -> d
  27.             .index(INDEX_NAME)
  28.             .query(q -> q.term(t -> t
  29.                     .field("age")
  30.                     .value(v -> v.longValue(11))
  31.             ))
  32.             .refresh(true)
  33.     );
  34.     Assert.isTrue(1 == age.deleted());
  36.     //5.校验是否存在
  37.     exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  38.     Assert.isFalse(exists.value());
  39.     exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  40.     Assert.isTrue(exists.value());
  41. }

完整版单元测试

  1. import cn.hutool.core.lang.Assert;
  2. import cn.hutool.core.util.ObjectUtil;
  3. import cn.hutool.core.util.StrUtil;
  4. import cn.hutool.json.JSONUtil;
  5. import co.elastic.clients.elasticsearch.ElasticsearchClient;
  6. import co.elastic.clients.elasticsearch._types.OpType;
  7. import co.elastic.clients.elasticsearch._types.Refresh;
  8. import co.elastic.clients.elasticsearch._types.Result;
  9. import co.elastic.clients.elasticsearch.core.*;
  10. import co.elastic.clients.elasticsearch.core.mget.MultiGetResponseItem;
  11. import co.elastic.clients.elasticsearch.indices.CreateIndexResponse;
  12. import co.elastic.clients.transport.endpoints.BooleanResponse;
  13. import com.ftc.elasticsearchtest.entity.Student;
  14. import org.junit.jupiter.api.BeforeEach;
  15. import org.junit.jupiter.api.Test;
  16. import org.springframework.beans.factory.annotation.Qualifier;
  17. import org.springframework.boot.test.context.SpringBootTest;
  19. import javax.annotation.Resource;
  20. import java.io.IOException;
  21. import java.util.List;
  22. import java.util.Locale;
  24. /**
  25.  * @author: 冯铁城 [17615007230@163.com]
  26.  * @date: 2022-08-22 11:11:22
  27.  * @describe: ElasticSearch文档API单元测试
  28.  */
  29. @SpringBootTest
  30. public class ElasticsearchDocTest {
  32.     @Resource
  33.     @Qualifier("primaryElasticsearchClient")
  34.     private ElasticsearchClient primaryClient;
  36.     @Resource
  37.     @Qualifier("secondaryElasticsearchClient")
  38.     private ElasticsearchClient secondaryClient;
  40.     /**
  41.      * 默认文档ID
  42.      */
  43.     private static final String DEFAULT_ID = "66";
  45.     /**
  46.      * 测试Mock索引名称常量
  47.      */
  48.     private static final String INDEX_NAME = "test_index";
  50.     @BeforeEach
  51.     void createIndexAndFlushAll() throws IOException {
  53.         //1.判定索引是否存在
  54.         boolean exist = primaryClient
  55.                 .indices()
  56.                 .exists(e -> e.index(INDEX_NAME + "_1"))
  57.                 .value();
  59.         //2.索引不存在创建索引
  60.         if (!exist) {
  61.             CreateIndexResponse createIndexResponse = primaryClient
  62.                     .indices()
  63.                     .create(i -> i
  64.                             .index(INDEX_NAME + "_1")
  65.                             .aliases(INDEX_NAME, a -> a)
  66.                             .mappings(m -> m
  67.                                     .properties("name", p -> p.text(t -> t.analyzer("ik_max_word")))
  68.                                     .properties("age", p -> p.integer(in -> in))
  69.                             )
  70.                             .settings(s -> s
  71.                                     .refreshInterval(t -> t.time("1s"))
  72.                                     .numberOfShards("3")
  73.                                     .numberOfReplicas("1")
  74.                             )
  75.                     );
  76.             Assert.isTrue(Boolean.TRUE.equals(createIndexResponse.acknowledged()));
  77.         }
  79.         //3.清空索引数据
  80.         DeleteByQueryResponse delete = primaryClient.deleteByQuery(d -> d
  81.                 .index(INDEX_NAME)
  82.                 .query(q -> q.matchAll(m -> m))
  83.                 .refresh(true)
  84.         );
  85.         Assert.isTrue(ObjectUtil.isNotNull(delete.deleted()));
  87.         //4.创建最新数据
  88.         Student student = new Student();
  89.         student.setId(66);
  90.         student.setName("宋可心");
  91.         student.setAge(18);
  93.         //5.存入最新数据
  94.         IndexResponse index = primaryClient.index(i -> i
  95.                 .index(INDEX_NAME)
  96.                 .document(student)
  97.                 .id(DEFAULT_ID)
  98.                 .refresh(Refresh.True)
  99.         );
  100.         Assert.isTrue("created".equals(index.result().jsonValue().toLowerCase(Locale.ROOT)));
  101.     }
  103.     @Test
  104.     void createDocSingle() throws IOException {
  106.         //1.创建学生对象
  107.         Student student = new Student();
  108.         student.setId(1);
  109.         student.setName("冯铁城");
  110.         student.setAge(11);
  112.         //2.创建单个文档不指定ID
  113.         IndexResponse indexResponse = primaryClient.index(i -> i
  114.                 .index(INDEX_NAME)
  115.                 .document(student)
  116.         );
  117.         Result result = indexResponse.result();
  118.         Assert.isTrue("created".equals(result.jsonValue().toLowerCase(Locale.ROOT)));
  120.         //3.创建单个文档指定ID
  121.         indexResponse = primaryClient.index(i -> i
  122.                 .index(INDEX_NAME)
  123.                 .document(student)
  124.                 .id("1")
  125.         );
  126.         result = indexResponse.result();
  127.         Assert.isTrue("created".equals(result.jsonValue().toLowerCase(Locale.ROOT)));
  129.         //4.创建单个文档进行ID唯一性控制
  130.         String errorMessage = null;
  131.         try {
  132.             primaryClient.index(i -> i
  133.                     .index(INDEX_NAME)
  134.                     .document(student)
  135.                     .id("1")
  136.                     .opType(OpType.Create)
  137.             );
  138.         } catch (Exception e) {
  139.             errorMessage = e.getMessage();
  140.         }
  141.         Assert.isTrue(ObjectUtil.isNotNull(errorMessage));
  142.         Assert.isTrue(errorMessage.contains("version_conflict_engine_exception"));
  143.     }
  145.     @Test
  146.     void createDocBulk() throws IOException {
  148.         //1.创建学生对象
  149.         Student student = new Student();
  150.         student.setId(1);
  151.         student.setName("张钰玲");
  152.         student.setAge(11);
  154.         //2.批量保存数据不带ID
  155.         BulkResponse bulk = primaryClient.bulk(b -> b
  156.                 .index(INDEX_NAME)
  157.                 .operations(o -> o.create(c -> c.document(student)))
  158.                 .operations(o -> o.create(c -> c.document(student)))
  159.         );
  160.         Assert.isFalse(bulk.errors());
  161.         Assert.isTrue(2 == bulk.items().size());
  163.         //3.批量保存数据带ID
  164.         bulk = primaryClient.bulk(b -> b
  165.                 .index(INDEX_NAME)
  166.                 .operations(o -> o.create(c -> c.document(student).id("1")))
  167.                 .operations(o -> o.create(c -> c.document(student).id("2")))
  168.         );
  169.         Assert.isFalse(bulk.errors());
  170.         Assert.isTrue(2 == bulk.items().size());
  171.     }
  173.     @Test
  174.     void getDocSingle() throws IOException {
  176.         //1.创建学生对象
  177.         Student student = new Student();
  178.         student.setId(1);
  179.         student.setName("冯铁城");
  180.         student.setAge(18);
  181.         Student studentBak = new Student();
  182.         student.setId(2);
  183.         student.setName("张钰玲");
  184.         student.setAge(16);
  186.         //2.批量保存数据
  187.         BulkResponse bulk = secondaryClient.bulk(b -> b
  188.                 .index(INDEX_NAME)
  189.                 .operations(o -> o.create(c -> c.document(student).id("1")))
  190.                 .operations(o -> o.create(c -> c.document(studentBak).id("2")))
  191.                 .refresh(Refresh.True)
  192.         );
  193.         Assert.isFalse(bulk.errors());
  194.         Assert.isTrue(2 == bulk.items().size());
  196.         //3.根据ID查询单个数据
  197.         Student source = secondaryClient.get(g -> g
  198.                 .index(INDEX_NAME)
  199.                 .id("1"), Student.class
  200.         ).source();
  201.         Assert.isTrue(JSONUtil.toJsonStr(student).equals(JSONUtil.toJsonStr(source)));
  203.         //4.按照ID批量查询
  204.         List<MultiGetResponseItem<Student>> docs = secondaryClient.mget(m -> m
  205.                 .index(INDEX_NAME)
  206.                 .ids("1", "2"), Student.class
  207.         ).docs();
  208.         Assert.isTrue(2 == docs.size());
  209.         Student source1 = docs.get(0).result().source();
  210.         Assert.isTrue(JSONUtil.toJsonStr(student).equals(JSONUtil.toJsonStr(source1)));
  211.         Student source2 = docs.get(1).result().source();
  212.         Assert.isTrue(JSONUtil.toJsonStr(studentBak).equals(JSONUtil.toJsonStr(source2)));
  213.     }
  215.     @Test
  216.     void existDoc() throws IOException {
  218.         //1.验证数据存在
  219.         BooleanResponse exists = secondaryClient.exists(e -> e
  220.                 .index(INDEX_NAME)
  221.                 .id(DEFAULT_ID)
  222.         );
  223.         Assert.isTrue(exists.value());
  225.         //2.验证数据不存在
  226.         exists = secondaryClient.exists(e -> e
  227.                 .index(INDEX_NAME)
  228.                 .id(DEFAULT_ID + "1")
  229.         );
  230.         Assert.isFalse(exists.value());
  231.     }
  233.     @Test
  234.     void updateDocSingle() throws IOException {
  236.         //1.查询默认数据
  237.         Student student = primaryClient.get(g -> g
  238.                 .index(INDEX_NAME)
  239.                 .id(DEFAULT_ID), Student.class
  240.         ).source();
  241.         Assert.isTrue(ObjectUtil.isNotNull(student));
  243.         //2.全量更新单个数据（立即refresh，为了单元测试方便）
  244.         student.setAge(100);
  245.         IndexResponse index = primaryClient.index(i -> i
  246.                 .index(INDEX_NAME)
  247.                 .document(student)
  248.                 .id(DEFAULT_ID)
  249.                 .refresh(Refresh.True)
  250.         );
  251.         Assert.isTrue("updated".equals(index.result().jsonValue().toLowerCase(Locale.ROOT)));
  253.         //3.查询校验
  254.         Integer age = primaryClient.get(g -> g
  255.                 .index(INDEX_NAME)
  256.                 .id(DEFAULT_ID), Student.class
  257.         ).source().getAge();
  258.         Assert.isTrue(100 == age);
  260.         //4.全量更新单个数据并进行版本控制（立即refresh，为了单元测试方便）
  261.         student.setAge(11);
  262.         long primaryTerm = index.primaryTerm();
  263.         long seqNo = index.seqNo();
  264.         index = primaryClient.index(i -> i
  265.                 .index(INDEX_NAME)
  266.                 .document(student)
  267.                 .id(DEFAULT_ID)
  268.                 .ifPrimaryTerm(primaryTerm)
  269.                 .ifSeqNo(seqNo)
  270.                 .refresh(Refresh.True)
  271.         );
  272.         Assert.isTrue("updated".equals(index.result().jsonValue().toLowerCase(Locale.ROOT)));
  274.         //5.查询校验
  275.         age = primaryClient.get(g -> g
  276.                 .index(INDEX_NAME)
  277.                 .id(DEFAULT_ID), Student.class
  278.         ).source().getAge();
  279.         Assert.isTrue(11 == age);
  281.         //6.全量更新单个数据并进行版本控制（立即refresh，为了单元测试方便）
  282.         String errorMessage = null;
  283.         try {
  284.             primaryClient.index(i -> i
  285.                     .index(INDEX_NAME)
  286.                     .document(student)
  287.                     .id(INDEX_NAME)
  288.                     .ifPrimaryTerm((long) 1)
  289.                     .ifSeqNo((long) 2)
  290.                     .refresh(Refresh.True)
  291.             );
  292.         } catch (Exception e) {
  293.             errorMessage = e.getMessage();
  294.         }
  295.         Assert.isTrue(StrUtil.isNotBlank(errorMessage));
  296.         Assert.isTrue(errorMessage.contains("version_conflict_engine_exception"));
  298.         //7.更新文档部分属性
  299.         Student studentBak = new Student();
  300.         studentBak.setAge(30);
  301.         UpdateResponse<Student> update = primaryClient.update(u -> u
  302.                 .index(INDEX_NAME)
  303.                 .id(DEFAULT_ID)
  304.                 .doc(studentBak)
  305.                 .refresh(Refresh.True), Student.class
  306.         );
  307.         Assert.isTrue("updated".equals(update.result().jsonValue().toLowerCase(Locale.ROOT)));
  309.         //8.查询校验
  310.         Student source = primaryClient.get(g -> g
  311.                 .index(INDEX_NAME)
  312.                 .id(DEFAULT_ID), Student.class
  313.         ).source();
  314.         Assert.isTrue(ObjectUtil.isNotNull(source));
  315.         Assert.isTrue(studentBak.getAge() == source.getAge());
  316.         Assert.isTrue(student.getName().equals(source.getName()));
  317.     }
  319.     @Test
  320.     void updateDocBulk() throws IOException {
  322.         //1.准备测试数据
  323.         Student student = new Student();
  324.         student.setName("冯铁城");
  325.         student.setAge(18);
  326.         Student studentBak = new Student();
  327.         studentBak.setName("张钰玲");
  328.         studentBak.setAge(16);
  330.         //2.存入测试数据
  331.         BulkResponse bulk = primaryClient.bulk(b -> b
  332.                 .index(INDEX_NAME)
  333.                 .operations(o -> o.create(c -> c.document(student).id("1")))
  334.                 .operations(o -> o.create(c -> c.document(studentBak).id("2")))
  335.         );
  336.         Assert.isFalse(bulk.errors());
  337.         Assert.isTrue(2 == bulk.items().size());
  339.         //3.批量更新部分数据
  340.         Student bulkStudent = new Student();
  341.         bulkStudent.setAge(111);
  342.         bulk = primaryClient.bulk(b -> b
  343.                 .index(INDEX_NAME)
  344.                 .operations(o -> o.update(u -> u.action(a -> a.doc(bulkStudent)).id("1")))
  345.                 .operations(o -> o.update(u -> u.action(a -> a.doc(bulkStudent)).id("2")))
  346.         );
  347.         Assert.isFalse(bulk.errors());
  348.         Assert.isTrue(2 == bulk.items().size());
  350.         //4.查询校验
  351.         Student source = primaryClient.get(g -> g
  352.                 .index(INDEX_NAME)
  353.                 .id("1"), Student.class
  354.         ).source();
  355.         Assert.isTrue("{\"name\":\"冯铁城\",\"age\":111}".equals(JSONUtil.toJsonStr(source)));
  356.         source = primaryClient.get(g -> g
  357.                 .index(INDEX_NAME)
  358.                 .id("2"), Student.class
  359.         ).source();
  360.         Assert.isTrue("{\"name\":\"张钰玲\",\"age\":111}".equals(JSONUtil.toJsonStr(source)));
  362.         //5.批量更新全量数据
  363.         Student bulkStudent2 = new Student();
  364.         bulk = primaryClient.bulk(b -> b
  365.                 .index(INDEX_NAME)
  366.                 .operations(o -> o.index(i -> i.document(bulkStudent2).id("1")))
  367.                 .operations(o -> o.index(i -> i.document(bulkStudent2).id("2")))
  368.         );
  369.         Assert.isFalse(bulk.errors());
  370.         Assert.isTrue(2 == bulk.items().size());
  372.         //6.查询验证
  373.         source = primaryClient.get(g -> g
  374.                 .index(INDEX_NAME)
  375.                 .id("1"), Student.class
  376.         ).source();
  377.         Assert.isTrue("{}".equals(JSONUtil.toJsonStr(source)));
  378.         source = primaryClient.get(g -> g
  379.                 .index(INDEX_NAME)
  380.                 .id("2"), Student.class
  381.         ).source();
  382.         Assert.isTrue("{}".equals(JSONUtil.toJsonStr(source)));
  383.     }
  385.     @Test
  386.     void updateDocQuery() throws IOException {
  388.         //1.准备测试数据
  389.         Student student = new Student();
  390.         student.setName("冯铁城");
  391.         student.setAge(18);
  392.         Student studentBak = new Student();
  393.         studentBak.setName("张钰玲");
  394.         studentBak.setAge(16);
  396.         //2.存入测试数据
  397.         BulkResponse bulk = primaryClient.bulk(b -> b
  398.                 .index(INDEX_NAME)
  399.                 .operations(o -> o.create(c -> c.document(student).id("1")))
  400.                 .operations(o -> o.create(c -> c.document(studentBak).id("2")))
  401.                 .refresh(Refresh.True)
  402.         );
  403.         Assert.isFalse(bulk.errors());
  404.         Assert.isTrue(2 == bulk.items().size());
  406.         //3.更新名称为冯铁城的数据，将其年龄改为100
  407.         UpdateByQueryResponse updateByQueryResponse = primaryClient.updateByQuery(u -> u
  408.                 .index(INDEX_NAME)
  409.                 .query(q -> q.match(m -> m
  410.                         .field("name")
  411.                         .query(query -> query.stringValue("冯铁城"))
  412.                 ))
  413.                 .script(s -> s.inline(i -> i
  414.                         .source("ctx._source.age=100")
  415.                 ))
  416.                 .refresh(true)
  417.         );
  418.         Assert.isTrue(1 == updateByQueryResponse.updated());
  420.         //4.查询校验
  421.         Student source = primaryClient.get(g -> g
  422.                 .index(INDEX_NAME)
  423.                 .id("1"), Student.class
  424.         ).source();
  425.         Assert.isTrue(100 == source.getAge());
  427.         //5.更新名称为冯铁城的数据，移除age属性
  428.         updateByQueryResponse = primaryClient.updateByQuery(u -> u
  429.                 .index(INDEX_NAME)
  430.                 .query(q -> q.match(m -> m
  431.                         .field("name")
  432.                         .query(query -> query.stringValue("冯铁城"))
  433.                 ))
  434.                 .script(s -> s.inline(i -> i
  435.                         .source("ctx._source.remove('age')")
  436.                 ))
  437.                 .refresh(true)
  438.         );
  439.         Assert.isTrue(1 == updateByQueryResponse.updated());
  441.         //6.查询校验
  442.         source = primaryClient.get(g -> g
  443.                 .index(INDEX_NAME)
  444.                 .id("1"), Student.class
  445.         ).source();
  446.         Assert.isNull(source.getAge());
  447.     }
  449.     @Test
  450.     void deleteDocSingle() throws IOException {
  452.         //1.验证默认文档存在
  453.         BooleanResponse exists = secondaryClient.exists(e -> e.index(INDEX_NAME).id(DEFAULT_ID));
  454.         Assert.isTrue(exists.value());
  456.         //2.删除文档
  457.         DeleteResponse delete = secondaryClient.delete(d -> d
  458.                 .index(INDEX_NAME)
  459.                 .id(DEFAULT_ID)
  460.                 .refresh(Refresh.True)
  461.         );
  462.         Assert.isTrue("deleted".equals(delete.result().jsonValue().toLowerCase(Locale.ROOT)));
  464.         //3.再次验证，文档已不存在
  465.         exists = secondaryClient.exists(e -> e.index(INDEX_NAME).id(DEFAULT_ID));
  466.         Assert.isFalse(exists.value());
  467.     }
  469.     @Test
  470.     void deleteDocBulk() throws IOException {
  472.         //1.存入测试数据
  473.         BulkResponse bulk = primaryClient.bulk(b -> b
  474.                 .index(INDEX_NAME)
  475.                 .operations(o -> o.create(c -> c.document(new Student()).id("1")))
  476.                 .operations(o -> o.create(c -> c.document(new Student()).id("2")))
  477.                 .refresh(Refresh.True)
  478.         );
  479.         Assert.isFalse(bulk.errors());
  481.         //2.验证文档已存在
  482.         BooleanResponse exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  483.         Assert.isTrue(exists.value());
  484.         exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  485.         Assert.isTrue(exists.value());
  487.         //3.批量删除
  488.         bulk = primaryClient.bulk(b -> b
  489.                 .index(INDEX_NAME)
  490.                 .operations(o -> o.delete(d -> d.id("1")))
  491.                 .operations(o -> o.delete(d -> d.id("2")))
  492.                 .refresh(Refresh.True)
  493.         );
  494.         Assert.isFalse(bulk.errors());
  495.         Assert.isTrue(2 == bulk.items().size());
  497.         //4.验证已不存在
  498.         exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  499.         Assert.isFalse(exists.value());
  500.         exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  501.         Assert.isFalse(exists.value());
  502.     }
  504.     @Test
  505.     void deleteDocQuery() throws IOException {
  507.         //1.准备测试数据
  508.         Student student1 = new Student();
  509.         student1.setAge(11);
  510.         Student student2 = new Student();
  511.         student2.setAge(18);
  513.         //2.存入测试数据
  514.         BulkResponse bulk = primaryClient.bulk(b -> b
  515.                 .index(INDEX_NAME)
  516.                 .operations(o -> o.create(c -> c.document(student1).id("1")))
  517.                 .operations(o -> o.create(c -> c.document(student2).id("2")))
  518.                 .refresh(Refresh.True)
  519.         );
  520.         Assert.isFalse(bulk.errors());
  522.         //3.校验是否存在
  523.         BooleanResponse exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  524.         Assert.isTrue(exists.value());
  525.         exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  526.         Assert.isTrue(exists.value());
  528.         //4.删除数据
  529.         DeleteByQueryResponse age = primaryClient.deleteByQuery(d -> d
  530.                 .index(INDEX_NAME)
  531.                 .query(q -> q.term(t -> t
  532.                         .field("age")
  533.                         .value(v -> v.longValue(11))
  534.                 ))
  535.                 .refresh(true)
  536.         );
  537.         Assert.isTrue(1 == age.deleted());
  539.         //5.校验是否存在
  540.         exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("1"));
  541.         Assert.isFalse(exists.value());
  542.         exists = primaryClient.exists(e -> e.index(INDEX_NAME).id("2"));
  543.         Assert.isTrue(exists.value());
  544.     }
  545. }