@SpringBootTest
class EsdemoApplicationTests {

    private final Logger log = LoggerFactory.getLogger(this.getClass());

    @Autowired
    private RestHighLevelClient restHighLevelClient;

    public final static String ES_INDEX = "test_index";

    /**
     * api 调用
     */
    @Test
    public void testCreateIndex() throws IOException {
        // 测试索引是否存在
        GetIndexRequest getIndexRequest = new GetIndexRequest(ES_INDEX);
        boolean exists = restHighLevelClient.indices().exists(getIndexRequest, RequestOptions.DEFAULT);

        log.info("索引是否存在:{}", exists);

        if (!exists) {
            // 创建索引请求
            CreateIndexRequest request = new CreateIndexRequest(ES_INDEX);
            // 客户端执行请求
            CreateIndexResponse createIndexResponse = restHighLevelClient.indices().create(request, RequestOptions.DEFAULT);
            log.info("索引创建成功:{}", createIndexResponse);
        }

        /* 文档操作 */
        //创建文档
        User user = new User("张三", 7);
        IndexRequest index = new IndexRequest(ES_INDEX);
        // 规则 put/index/_doc/1
        index.id("1");
        index.timeout(TimeValue.timeValueSeconds(1));
        // 将数据放入请求
        index.source(JSONUtil.toJsonStr(user), XContentType.JSON);
        // 发送请求 ,获取响应结果
        IndexResponse indexResponse = restHighLevelClient.index(index, RequestOptions.DEFAULT);
        log.info("响应结果:{},响应状态:{}", indexResponse, indexResponse.status());

        // 获取文档
        GetRequest getRequest = new GetRequest("test_index", "1");
        // 不获取返回的 _source 的上下文
        getRequest.fetchSourceContext(new FetchSourceContext(false));
        getRequest.storedFields("_none_");
        boolean exists1 = restHighLevelClient.exists(getRequest, RequestOptions.DEFAULT);
        log.info("查看文档是否存在:{}", exists1);
        GetRequest getRequest2 = new GetRequest("test_index", "1");
        GetResponse documentFields = restHighLevelClient.get(getRequest2, RequestOptions.DEFAULT);
        String sourceAsString = documentFields.getSourceAsString();
        log.info("文档内容:{}", sourceAsString);

        // 更新文档
        UpdateRequest updateRequest = new UpdateRequest("test_index", "1");
        User user1 = new User("李四", 8);
        updateRequest.doc(JSONUtil.toJsonStr(user1), XContentType.JSON);
        UpdateResponse update = restHighLevelClient.update(updateRequest, RequestOptions.DEFAULT);
        log.info("更新文档:{},更新状态:{}", update, update.status());

        // 批量操作
        BulkRequest test_index = new BulkRequest();
        for (int i = 0; i < 20; i++) {
            test_index.add(new IndexRequest(ES_INDEX).id(i + 2 + "").source(JSONUtil.toJsonStr(new User(RandomUtil.randomString(8), i + 2)), XContentType.JSON));
        }
        BulkResponse bulk = restHighLevelClient.bulk(test_index, RequestOptions.DEFAULT);
        log.info("批量执行失败状态:{}", bulk.hasFailures());

        // 查询
        SearchRequest searchRequest = new SearchRequest(ES_INDEX);
        // 构建搜索
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
        // 构建查询对象
        TermQueryBuilder termQueryBuilder = QueryBuilders.termQuery("age", 8);
        searchSourceBuilder.query(termQueryBuilder);
        searchRequest.source(searchSourceBuilder);
        SearchResponse search = restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT);
        // 结果
        log.info("搜索返回结果:{}", JSONUtil.toJsonStr(search.getHits()));
        for (SearchHit hit : search.getHits().getHits()) {
            log.info("结果:{}", hit.getSourceAsMap());
        }

        // 删除文档
        DeleteRequest deleteRequest = new DeleteRequest("test_index", "1");
        DeleteResponse delete = restHighLevelClient.delete(deleteRequest, RequestOptions.DEFAULT);
        log.info("删除文档:{},删除状态:{}", delete, delete.status());

        // 删除索引
        DeleteIndexRequest deleteIndexRequest = new DeleteIndexRequest(ES_INDEX);
        AcknowledgedResponse deleteResponse = restHighLevelClient.indices().delete(deleteIndexRequest, RequestOptions.DEFAULT);
        log.info("是否删除成功:{}", deleteResponse.isAcknowledged());
    }

}