public class IndexTest {

    String ES_URL = "http://192.168.207.129:9200";

    private RestHighLevelClient client;

    @BeforeEach
    void setUp() {
        this.client = new RestHighLevelClient(RestClient.builder(
                HttpHost.create(ES_URL)
        ));
        System.out.println("setUp success");
    }

    @Test
    void testConnect() {
        System.out.println(client);
    }

    @Test
    void testCreateIndex() throws IOException {
        // 1.创建Request对象
        CreateIndexRequest request = new CreateIndexRequest("user");
        // 2.准备请求参数
        request.source(MAPPING_TEMPLATE, XContentType.JSON);
        // 3.发送请求
        client.indices()
                .create(request, RequestOptions.DEFAULT);
    }

    @Test
    void testDeleteIndex() throws IOException {
        // 1.创建Request对象
        DeleteIndexRequest request = new DeleteIndexRequest("user");
        // 2.发送请求
        AcknowledgedResponse response = client.indices()
                .delete(request, RequestOptions.DEFAULT);
        System.out.println(response.isAcknowledged());
    }

    @Test
    void testGetIndex() throws IOException {
        // 1.创建Request对象
        GetIndexRequest request = new GetIndexRequest("user");
        // 2.发送请求
        GetIndexResponse getIndexResponse = client.indices()
                .get(request, RequestOptions.DEFAULT);
        // 3.输出
        System.err.println(getIndexResponse.getAliases());
        System.err.println(getIndexResponse.getMappings());
        System.err.println(getIndexResponse.getSettings());
    }

    @Test
    void testExistsIndex() throws IOException {
        // 1.创建Request对象
        GetIndexRequest request = new GetIndexRequest("user");
        // 2.发送请求
        boolean exists = client.indices()
                .exists(request, RequestOptions.DEFAULT);
        // 3.输出
        System.err.println(exists ? "索引库已经存在！" : "索引库不存在！");
    }

    static final String MAPPING_TEMPLATE = "{\n" +
            "  \"mappings\": {\n" +
            "    \"properties\": {\n" +
            "      \"id\": {\n" +
            "        \"type\": \"long\"\n" +
            "      },\n" +
            "      \"user_account\": {\n" +
            "        \"type\": \"keyword\"\n" +
            "      },\n" +
            "      \"user_password\": {\n" +
            "        \"type\": \"keyword\",\n" +
            "        \"index\": false\n" +
            "      },\n" +
            "      \"user_name\": {\n" +
            "        \"type\": \"text\",\n" +
            "        \"analyzer\": \"ik_max_word\",\n" +
            "        \"fields\": {\n" +
            "          \"keyword\": {\n" +
            "            \"type\": \"keyword\",\n" +
            "            \"ignore_above\": 256\n" +
            "          }\n" +
            "        }\n" +
            "      },\n" +
            "      \"user_avatar\": {\n" +
            "        \"type\": \"keyword\",\n" +
            "        \"index\": false\n" +
            "      },\n" +
            "      \"gender\": {\n" +
            "        \"type\": \"integer\"\n" +
            "      },\n" +
            "      \"phone\": {\n" +
            "        \"type\": \"keyword\"\n" +
            "      },\n" +
            "      \"email\": {\n" +
            "        \"type\": \"keyword\"\n" +
            "      },\n" +
            "      \"user_profile\": {\n" +
            "        \"type\": \"text\",\n" +
            "        \"analyzer\": \"ik_max_word\"\n" +
            "      },\n" +
            "      \"user_role\": {\n" +
            "        \"type\": \"keyword\"\n" +
            "      },\n" +
            "      \"create_user\": {\n" +
            "        \"type\": \"long\"\n" +
            "      },\n" +
            "      \"create_time\": {\n" +
            "        \"type\": \"date\",\n" +
            "        \"format\": \"yyyy-MM-dd HH:mm:ss||epoch_millis\"\n" +
            "      },\n" +
            "      \"update_user\": {\n" +
            "        \"type\": \"long\"\n" +
            "      },\n" +
            "      \"update_time\": {\n" +
            "        \"type\": \"date\",\n" +
            "        \"format\": \"yyyy-MM-dd HH:mm:ss||epoch_millis\"\n" +
            "      },\n" +
            "      \"is_deleted\": {\n" +
            "        \"type\": \"byte\"\n" +
            "      },\n" +
            "      \"tenant_id\": {\n" +
            "        \"type\": \"long\"\n" +
            "      }\n" +
            "    }\n" +
            "  }\n" +
            "}";

    @AfterEach
    void tearDown() throws IOException {
        this.client.close();
        System.out.println("tearDown success");
    }
}

public class DocumentTest {

    String ES_URL = "http://192.168.207.129:9200";

    private RestHighLevelClient client;

    @BeforeEach
    void setUp() {
        this.client = new RestHighLevelClient(RestClient.builder(
                HttpHost.create(ES_URL)
        ));
        System.out.println("setUp success");
    }

    @Test
    void testInsertDocument() throws IOException {
        // 1.准备Request对象
        IndexRequest request = new IndexRequest();
        request.index("user").id("1001");
        // 2.准备Json文档，向ES插入数据，必须将数据转换位JSON格式
        UserDoc userDoc = new UserDoc();
        userDoc.setUserName("zhangsan");
        userDoc.setGender(1);
        request.source(JSONUtil.toJsonStr(userDoc), XContentType.JSON);
        // 3.发送请求
        IndexResponse response = client.index(request, RequestOptions.DEFAULT);

        System.out.println("res=" + response.getResult());
    }

    @Test
    void testGetDocumentById() throws IOException {
        // 1.准备Request对象
        GetRequest request = new GetRequest("user", "1001");
        // 2.发送请求
        GetResponse response = client.get(request, RequestOptions.DEFAULT);
        // 3.获取响应结果中的source
        String json = response.getSourceAsString();

        UserDoc userDoc = JSONUtil.toBean(json, UserDoc.class);
        System.out.println("userDoc= " + userDoc);
    }

    @Test
    void testDeleteDocument() throws IOException {
        // 1.准备Request，两个参数，第一个是索引库名，第二个是文档id
        DeleteRequest request = new DeleteRequest("user", "1001");
        // 2.发送请求
        client.delete(request, RequestOptions.DEFAULT);
    }

    @Test
    void testUpdateDocument() throws IOException {
        // 1.准备Request
        UpdateRequest request = new UpdateRequest("user", "1001");
        // 2.准备请求参数
        request.doc(
                "user_name", "王五",
                "gender", 0
        );
        // 3.发送请求
        UpdateResponse updateResponse = client.update(request, RequestOptions.DEFAULT);

        System.out.println("updateResponse.getResult() = " + updateResponse.getResult());
    }

    @Test
    void testBulk() throws IOException {
        ArrayList<UserDoc> list = new ArrayList<>();
        for (int i = 0; i < 100; i++) {
            UserDoc userDoc = new UserDoc();
            userDoc.setUserName("zhangsan_" + i);
            userDoc.setGender(1);
            list.add(userDoc);
        }

        // 1.创建Request
        BulkRequest request = new BulkRequest();
        // 2.准备请求参数
        for (int i = 0; i < list.size(); i++) {
            UserDoc userDoc = list.get(i);
            request.add(new IndexRequest("user")
                    .id("" + (i + 1))
                    .source(JSONUtil.toJsonStr(userDoc), XContentType.JSON));
        }
        // 3.发送请求
        BulkResponse bulkResponse = client.bulk(request, RequestOptions.DEFAULT);
        System.out.println("bulkResponse.getTook() = " + bulkResponse.getTook());
        System.out.println(Arrays.toString(bulkResponse.getItems()));
    }

    @AfterEach
    void tearDown() throws IOException {
        this.client.close();
        System.out.println("tearDown success");
    }
}