ElasticSearch Java API

POM依赖

<dependencies>
    <dependency>
        <groupId>org.elasticsearch</groupId>
        <artifactId>elasticsearch</artifactId>
        <version>7.8.0</version>
    </dependency>
    <!-- elasticsearch 的客户端 -->
    <dependency>
        <groupId>org.elasticsearch.client</groupId>
        <artifactId>elasticsearch-rest-high-level-client</artifactId>
        <version>7.8.0</version>
    </dependency>
    <!-- elasticsearch 依赖 2.x 的 log4j -->
    <dependency>
        <groupId>org.apache.logging.log4j</groupId>
        <artifactId>log4j-api</artifactId>
        <version>2.8.2</version>
    </dependency>
    <dependency>
        <groupId>org.apache.logging.log4j</groupId>
        <artifactId>log4j-core</artifactId>
        <version>2.8.2</version>
    </dependency>
    <dependency>
        <groupId>com.fasterxml.jackson.core</groupId>
        <artifactId>jackson-databind</artifactId>
        <version>2.9.9</version>
    </dependency>
    <!-- junit 单元测试 -->
    <dependency>
        <groupId>junit</groupId>
        <artifactId>junit</artifactId>
        <version>4.12</version>
    </dependency>
</dependencies>

API示例

package com.zcc.elasticsearch;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.sun.org.apache.bcel.internal.generic.NEW;
import org.apache.http.HttpHost;
import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest;
import org.elasticsearch.action.bulk.BulkItemResponse;
import org.elasticsearch.action.bulk.BulkRequest;
import org.elasticsearch.action.bulk.BulkResponse;
import org.elasticsearch.action.delete.DeleteRequest;
import org.elasticsearch.action.delete.DeleteResponse;
import org.elasticsearch.action.get.GetRequest;
import org.elasticsearch.action.get.GetResponse;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.action.index.IndexResponse;
import org.elasticsearch.action.search.SearchRequest;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.action.support.master.AcknowledgedResponse;
import org.elasticsearch.action.update.UpdateRequest;
import org.elasticsearch.action.update.UpdateResponse;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.client.indices.CreateIndexRequest;
import org.elasticsearch.client.indices.CreateIndexResponse;
import org.elasticsearch.client.indices.GetIndexRequest;
import org.elasticsearch.client.indices.GetIndexResponse;
import org.elasticsearch.common.unit.Fuzziness;
import org.elasticsearch.common.xcontent.XContentType;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.FuzzyQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHits;
import org.elasticsearch.search.aggregations.Aggregation;
import org.elasticsearch.search.aggregations.AggregationBuilder;
import org.elasticsearch.search.aggregations.AggregationBuilders;
import org.elasticsearch.search.aggregations.Aggregations;
import org.elasticsearch.search.aggregations.bucket.terms.Terms;
import org.elasticsearch.search.aggregations.metrics.Max;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;
import org.elasticsearch.search.sort.SortOrder;
import org.junit.After;
import org.junit.Test;

import java.io.IOException;

public class EsClientTest {

    private RestHighLevelClient esClient = new RestHighLevelClient(
            RestClient.builder(new HttpHost("localhost", 9200, "http")));

    private ObjectMapper objectMapper = new ObjectMapper();

    //表示在任意使用@Test注解标注的public void方法执行之后执行
    @After
    public void close() throws IOException {
        esClient.close();
    }

    //1.创建索引
    @Test
    public void createIndex() throws IOException {
        //创建索引
        CreateIndexRequest request = new CreateIndexRequest("user");
        CreateIndexResponse createIndexResponse = esClient.indices().create(request, RequestOptions.DEFAULT);
        //响应
        boolean acknowledged = createIndexResponse.isAcknowledged();
        System.out.println(acknowledged == true ? "创建索引成功" : "创建索引失败");
    }

    //2.查看索引
    @Test
    public void findIndex() throws IOException {
        GetIndexRequest request = new GetIndexRequest("user");
        GetIndexResponse response = esClient.indices().get(request, RequestOptions.DEFAULT);

        System.out.println(response.getAliases());
        System.out.println(response.getMappings()); //查看映射
        System.out.println(response.getSettings());
    }

    //3.删除索引
    @Test
    public void deleteIndex() throws IOException {
        DeleteIndexRequest request = new DeleteIndexRequest("user");
        AcknowledgedResponse response = esClient.indices().delete(request, RequestOptions.DEFAULT);

        boolean acknowledged = response.isAcknowledged();
        System.out.println(acknowledged == true ? "删除索引成功" : "删除索引失败");
    }

    //4.创建文档
    @Test
    public void createDocument() throws IOException {
        IndexRequest indexRequest = new IndexRequest();
        indexRequest.index("user").id("1001");
        User user = new User("zccheng", "男", 27);
        //向ES插入数据，必须将数据转换位JSON格式
        indexRequest.source(objectMapper.writeValueAsString(user), XContentType.JSON);
        IndexResponse response = esClient.index(indexRequest, RequestOptions.DEFAULT);
    }

    //5.修改文档
    @Test
    public void updateDocument() throws IOException {
        UpdateRequest updateRequest = new UpdateRequest();
        updateRequest.index("user").id("1001");
        updateRequest.doc(XContentType.JSON, "sex", "女");
        UpdateResponse response = esClient.update(updateRequest, RequestOptions.DEFAULT);
        System.out.println(response.getResult());
    }

    //6.查看文档
    @Test
    public void findDocument() throws IOException {
        GetRequest getRequest = new GetRequest();
        getRequest.index("user").id("1001");
        GetResponse response = esClient.get(getRequest, RequestOptions.DEFAULT);
        //获取查询的结果字符串
        System.out.println(response.getSourceAsString());
    }

    //7.删除文档
    @Test
    public void deleteDocument() throws IOException {
        DeleteRequest deleteRequest = new DeleteRequest();
        deleteRequest.index("user").id("1001");
        DeleteResponse response = esClient.delete(deleteRequest, RequestOptions.DEFAULT);
        System.out.println(response.getResult());
    }

    //8.批量插入文档
    @Test
    public void insertDocumentBatch() throws IOException {
        BulkRequest bulkRequest = new BulkRequest();
        bulkRequest.add(new IndexRequest().index("user").id("1001").source(XContentType.JSON, "name", "zccheng", "age", 20, "sex", "男"));
        bulkRequest.add(new IndexRequest().index("user").id("1002").source(XContentType.JSON, "name", "lisi", "age", 30, "sex", "男"));
        bulkRequest.add(new IndexRequest().index("user").id("1003").source(XContentType.JSON, "name", "wangwu", "age", 40, "sex", "女"));
        bulkRequest.add(new IndexRequest().index("user").id("1004").source(XContentType.JSON, "name", "zhaoliu", "age", 50, "sex", "男"));
        bulkRequest.add(new IndexRequest().index("user").id("1005").source(XContentType.JSON, "name", "wangba", "age", 50, "sex", "女"));
        bulkRequest.add(new IndexRequest().index("user").id("1006").source(XContentType.JSON, "name", "huangjiu", "age", 20, "sex", "男"));
        BulkResponse response = esClient.bulk(bulkRequest, RequestOptions.DEFAULT);
        for (BulkItemResponse item : response.getItems()) {
            if (item.isFailed()) {
                System.out.println("id = " + item.getId() + "插入失败");
            } else {
                System.out.println("id = " + item.getId() + "插入成功");
            }
        }
    }

    //9.批量删除文档
    @Test
    public void deleteDocumentBatch() throws IOException {
        BulkRequest bulkRequest = new BulkRequest();
        bulkRequest.add(new DeleteRequest().index("user").id("1001"));
        bulkRequest.add(new DeleteRequest().index("user").id("1002"));
        bulkRequest.add(new DeleteRequest().index("user").id("1003"));
        bulkRequest.add(new DeleteRequest().index("user").id("1004"));
        bulkRequest.add(new DeleteRequest().index("user").id("1005"));
        bulkRequest.add(new DeleteRequest().index("user").id("1006"));
        BulkResponse response = esClient.bulk(bulkRequest, RequestOptions.DEFAULT);
        if (response.hasFailures()) {
            System.out.println("批量删除失败");
        } else {
            System.out.println("批量删除成功");
        }
    }

    //10.高级查询-全量查询
    @Test
    public void queryDocumentAll() throws IOException {
        SearchRequest searchRequest = new SearchRequest();
        searchRequest.indices("user").source(new SearchSourceBuilder().query(QueryBuilders.matchAllQuery()));
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });
    }

    //11.高级查询-条件查询
    @Test
    public void queryDocument() throws IOException {
        SearchRequest searchRequest = new SearchRequest();
        searchRequest.indices("user").source(new SearchSourceBuilder().query(QueryBuilders.termQuery("sex", "男"))); //条件查询
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });

    }

    //12.高级查询-分页查询
    @Test
    public void queryDocumentPage() throws IOException {
        int pageNum = 3;
        int pageSize = 2;
        SearchRequest searchRequest = new SearchRequest();
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder()
                .query(QueryBuilders.matchAllQuery())
                .from(pageSize * (pageNum - 1)) //分页查询
                .size(pageSize);
        searchRequest.indices("user").source(searchSourceBuilder);
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });

    }

    //12.高级查询-排序查询
    @Test
    public void queryDocumentSort() throws IOException {
        SearchRequest searchRequest = new SearchRequest();
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder()
                .query(QueryBuilders.matchAllQuery())
                .sort("age", SortOrder.DESC); //排序查询,按照年龄降序排序
        searchRequest.indices("user").source(searchSourceBuilder);
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });
    }

    //13.高级查询-过滤字段
    @Test
    public void queryFilterFields() throws IOException {
        SearchRequest searchRequest = new SearchRequest();
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder()
                .query(QueryBuilders.matchAllQuery())
                .fetchSource(new String[]{}, new String[]{"age"}); //第一个数组参数是包含的字段,第二个数组参数是排除的字段
        searchRequest.indices("user").source(searchSourceBuilder);
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });
    }

    //14.高级查询-组合查询&&范围查询
    @Test
    public void queryConditions() throws IOException {
        SearchRequest searchRequest = new SearchRequest();

        BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
        boolQueryBuilder.must(QueryBuilders.termQuery("sex", "男"));
        //boolQueryBuilder.must(QueryBuilders.termQuery("name","zccheng"));
        //boolQueryBuilder.should(QueryBuilders.termQuery("sex","女"));
        boolQueryBuilder.mustNot(QueryBuilders.rangeQuery("age").gte(30)); //范围查询 表示年龄一定不能超过30
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder()
                .query(boolQueryBuilder);

        searchRequest.indices("user").source(searchSourceBuilder);
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });
    }

    //15.高级查询-模糊查询&&高亮查询
    @Test
    public void queryFuzzy() throws IOException {
        SearchRequest searchRequest = new SearchRequest();
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
        FuzzyQueryBuilder fuzzyQueryBuilder = QueryBuilders.fuzzyQuery("name", "zcchen").fuzziness(Fuzziness.ONE);
        HighlightBuilder highlightBuilder = new HighlightBuilder();
        highlightBuilder.preTags("<font color='red'>");
        highlightBuilder.postTags("</font>");
        highlightBuilder.field("name");


        searchSourceBuilder.query(fuzzyQueryBuilder);
        searchSourceBuilder.highlighter(highlightBuilder);

        searchRequest.indices("user").source(searchSourceBuilder);
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });
    }

    //16.高级查询-聚合查询
    @Test
    public void queryAggregation() throws IOException {
        SearchRequest searchRequest = new SearchRequest();
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
        searchSourceBuilder.query(QueryBuilders.matchAllQuery());

        AggregationBuilder maxAgeBuilder = AggregationBuilders.max("max_age_agg").field("age");
        AggregationBuilder sexGroupByBuilder = AggregationBuilders.terms("group_by_age").field("age");
        searchSourceBuilder.aggregation(maxAgeBuilder);
        searchSourceBuilder.aggregation(sexGroupByBuilder);

        searchRequest.indices("user").source(searchSourceBuilder);
        SearchResponse response = esClient.search(searchRequest, RequestOptions.DEFAULT);
        SearchHits hits = response.getHits();

        System.out.println(response.getTook()); //一共花费多长时间
        System.out.println(hits.getTotalHits()); //一共命中多少条记录
        //遍历显示每一条数据
        hits.forEach(hit -> {
            System.out.println(hit.getSourceAsString());
        });

        //获取聚合数据
        System.out.println("---------聚合数据--------");
        Aggregations aggregations = response.getAggregations();
        //因为是Max的聚合，所以用Max去接收
        Max max_age_agg = aggregations.get("max_age_agg");
        System.out.println("年龄最大值 = " + max_age_agg.getValue());

        //获取性别分组的聚合数据
        Terms group_by_age = aggregations.get("group_by_age");
        for (Terms.Bucket bucket : group_by_age.getBuckets()) {
            System.out.println("年龄:" + bucket.getKeyAsString() + "========> 人数:" + bucket.getDocCount());
        }

    }
}