Spring Boot集成ElasticSearch实现简单的增删查改接口

Love The Way You Lie 2022-02-22 14:12 190阅读 0赞

pringBoot集成ElasticSearch

pom.xml文件中，依赖的各jar包版本如下：

> org.springframework.boot spring-boot-starter-parent 2.0.3.RELEASE

在工程中新建一个config包，在该包中创建一个ESConfig配置类，用于构造es的客户端实例对象。代码如下：

> package org.zero01.elasticsearch.demo.config;
> 
> import org.elasticsearch.client.transport.TransportClient; import  
> org.elasticsearch.common.settings.Settings; import  
> org.elasticsearch.common.transport.InetSocketTransportAddress; import  
> org.elasticsearch.transport.client.PreBuiltTransportClient; import  
> org.springframework.context.annotation.Bean; import  
> org.springframework.context.annotation.Configuration;
> 
> import java.net.InetAddress; import java.net.UnknownHostException;
> 
> /\*\* \* @program: es-demo \* @description: es配置类 \* @author: 01 \*  
> @create: 2018-06-28 22:32 \*\*/ @Configuration public class ESConfig \{
> 
>     @Bean
>     public TransportClient client() throws UnknownHostException {
>         // 9300是es的tcp服务端口
>         InetSocketTransportAddress node = new InetSocketTransportAddress(
>                 InetAddress.getByName("192.168.190.129"),
>                 9300);
>     
>         // 设置es节点的配置信息
>         Settings settings = Settings.builder()
>                 .put("cluster.name", "es")
>                 .build();
>     
>         // 实例化es的客户端对象
>         TransportClient client = new PreBuiltTransportClient(settings);
>         client.addTransportAddress(node);
>     
>         return client;
>     } }

查询接口开发

我现在有一个结构化的索引如下：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70]  
该索引中有一些文档数据，如下：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 1]  
在工程中新建一个controller包，在该包中新建一个 BookCrudController 类，用于演示es增删查改接口demo。我们首先来开发查询接口，代码如下：

> package org.zero01.elasticsearch.demo.controller;
> 
> import org.elasticsearch.action.get.GetResponse; import  
> org.elasticsearch.client.transport.TransportClient; import  
> org.springframework.beans.factory.annotation.Autowired; import  
> org.springframework.http.HttpStatus; import  
> org.springframework.http.ResponseEntity; import  
> org.springframework.web.bind.annotation.GetMapping; import  
> org.springframework.web.bind.annotation.RequestMapping; import  
> org.springframework.web.bind.annotation.RequestParam; import  
> org.springframework.web.bind.annotation.RestController;
> 
> /\*\* \* @program: es-demo \* @description: es增删查改接口demo \* @author: 01
> 
>  *  @create: 2018-07-01 10:42 \*\*/ @RestController @RequestMapping("/es/demo") public class BookCrudController \{
>     
>     @Autowired  
>     private TransportClient client;
>     
>     /\*\*
>     
>      *  按id查询
>      *  @param id
>      *  @return  
>         \*/  
>         @GetMapping("/get/book/novel")  
>         public ResponseEntity searchById(@RequestParam(“id”) String id) \{  
>         if (id.isEmpty()) \{  
>         return new ResponseEntity(HttpStatus.NOT\_FOUND);  
>         \}
>         
>         // 通过索引、类型、id向es进行查询数据  
>         GetResponse response = client.prepareGet(“book”, “novel”, id).get();
>         
>         if (!response.isExists()) \{  
>         return new ResponseEntity(HttpStatus.NOT\_FOUND);  
>         \}
>         
>         // 返回查询到的数据  
>         return new ResponseEntity(response.getSource(), HttpStatus.OK);  
>         \} \}

启动SpringBoot工程，使用postman进行测试，查询结果如下：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 2]  
新增接口开发

在 BookCrudController 类中开发新增接口，代码如下：

> /\*\* \* 添加书籍数据 \* \* @param title 书籍标题 \* @param author 书籍作者
> 
>  *  @param wordCount 书籍字数 \* @param publishDate 发行时间 \* @return \*/ @PostMapping("/add/book/novel") public ResponseEntity  
>     add(@RequestParam(“title”) String title,  
>     @RequestParam(“author”) String author,  
>     @RequestParam(“word\_count”) int wordCount,  
>     @RequestParam(“publish\_date”)  
>     @DateTimeFormat(pattern = “yyy-MM-dd HH:mm:ss”)  
>     Date publishDate) \{  
>     try \{  
>     // 将参数build成一个json对象  
>     XContentBuilder content = XContentFactory.jsonBuilder()  
>     .startObject()  
>     .field(“title”, title)  
>     .field(“author”, author)  
>     .field(“word\_count”, wordCount)  
>     .field(“publish\_date”, publishDate.getTime())  
>     .endObject();
>     
>         IndexResponse response = client.prepareIndex("book", "novel")
>                   .setSource(content)
>                   .get();
>         
>           return new ResponseEntity(response.getId(), HttpStatus.OK);
>     
>     \} catch (IOException e) \{  
>     e.printStackTrace();  
>     return new ResponseEntity(HttpStatus.INTERNAL\_SERVER\_ERROR);  
>     \} \}

重启SpringBoot工程，使用postman进行测试，测试结果如下：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 3]  
使用返回的id去查询我们刚刚添加的书籍数据，结果如下：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 4]  
删除接口开发

代码如下：

> /\*\* \* 按id删除数据 \* \* @param id \* @return \*/  
> @DeleteMapping("/delete/book/novel") public ResponseEntity  
> delete(@RequestParam(“id”) String id) \{  
> DeleteResponse response = client.prepareDelete(“book”, “novel”, id).get();
> 
>     return new ResponseEntity(response.getResult(), HttpStatus.OK); }

重启SpringBoot工程，使用postman进行测试，删除数据成功：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 5]  
更新接口开发

代码如下：

> \*\* \* 根据文档id更新某个文档的数据 \* \* @param id \* @param title \* @param author \* @param wordCount \* @param publishDate \* @return \*/ @PutMapping("/update/book/novel") public ResponseEntity  
> update(@RequestParam(“id”) String id,  
> @RequestParam(value = “title”, required = false) String title,  
> @RequestParam(value = “author”, required = false) String author,  
> @RequestParam(value = “word\_count”, required = false) Integer wordCount,  
> @RequestParam(value = “publish\_date”, required = false)  
> @DateTimeFormat(pattern = “yyy-MM-dd HH:mm:ss”)  
> Date publishDate) \{  
> UpdateRequest update = new UpdateRequest(“book”, “novel”, id);  
> try \{  
> XContentBuilder builder = XContentFactory.jsonBuilder()  
> .startObject();
> 
>     if (title != null) {
>             builder.field("title", title);
>         }
>         if (author != null) {
>             builder.field("author", author);
>         }
>         if (wordCount != null) {
>             builder.field("word_count", wordCount);
>         }
>         if (publishDate != null) {
>             builder.field("publish_date", publishDate.getTime());
>         }
>     
>         builder.endObject();
>         update.doc(builder);
>     
>         UpdateResponse response = client.update(update).get();
>     
>         return new ResponseEntity(response.getResult().toString(), HttpStatus.OK);
>     } catch (Exception e) {
>         e.printStackTrace();
>         return new ResponseEntity(HttpStatus.INTERNAL_SERVER_ERROR);
>     } }

例如我们要修改文档id为AWQ-N\_XeWWbfsfQzkrTh的书籍数据：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 6]  
修改书籍的标题和作者：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 7]  
修改成功：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 8]  
复合查询接口开发

代码如下：

**
     * 复合查询接口
     *
     * @param title
     * @param author
     * @param wordCount
     * @param publishDate
     * @param gtWordCount
     * @param ltWordCount
     * @return
     */
    @PostMapping("/query/book/novel")
    public ResponseEntity query(@RequestParam(value = "title", required = false) String title,
                                @RequestParam(value = "author", required = false) String author,
                                @RequestParam(value = "word_count", required = false) Integer wordCount,
                                @RequestParam(value = "publish_date", required = false)
                                @DateTimeFormat(pattern = "yyy-MM-dd HH:mm:ss")
                                        Date publishDate,
                                @RequestParam(value = "gt_word_count", defaultValue = "0") Integer gtWordCount,
                                @RequestParam(value = "lt_word_count", required = false) Integer ltWordCount) {
    
        // 组装查询条件
        BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
        if (title != null) {
            boolQuery.must(QueryBuilders.matchQuery("title", title));
        }
        if (author != null) {
            boolQuery.must(QueryBuilders.matchQuery("author", author));
        }
        if (wordCount != null) {
            boolQuery.must(QueryBuilders.matchQuery("word_count", wordCount));
        }
        if (publishDate != null) {
            boolQuery.must(QueryBuilders.matchQuery("publish_date", publishDate));
        }
        // 以word_count作为条件范围
        RangeQueryBuilder rangeQuery = QueryBuilders.rangeQuery("word_count").from(gtWordCount);
        if (ltWordCount != null && ltWordCount > 0) {
            rangeQuery.to(ltWordCount);
        }
        boolQuery.filter(rangeQuery);
    
        // 组装查询请求
        SearchRequestBuilder requestBuilder = client.prepareSearch("book")
                .setTypes("novel")
                .setSearchType(SearchType.DFS_QUERY_THEN_FETCH)
                .setQuery(boolQuery)
                .setFrom(0)
                .setSize(10);
    
        // 发送查询请求
        SearchResponse response = requestBuilder.get();
    
        // 组装查询到的数据集
        List<Map<String, Object>> result = new ArrayList<>();
        for (SearchHit searchHitFields : response.getHits()) {
            result.add(searchHitFields.getSource());
        }
    
        return new ResponseEntity(result, HttpStatus.OK);
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70]: /images/20220222/9530e997648043c59c73ce0c00547818.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 1]: /images/20220222/28f67a91150f4b7e9fae947894d2b112.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 2]: /images/20220222/12d99190494043d2951ee12c02cd95ca.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 3]: /images/20220222/6ae91a40c9ec499f8ef7c66c5b941412.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 4]: /images/20220222/9bf0b7ea78bb4367ac100d7a85227a05.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 5]: /images/20220222/ae6ce9f9338d4c78a3d444b38b21e519.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 6]: /images/20220222/4df3fa3fd79f492d9fbf505e18775935.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 7]: /images/20220222/7dda7fd26be74f32b5ee8e53e73ded01.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 8]: /images/20220222/ba66c0195b88453ea67cc43d3c8a4ed1.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3podWd1YW5nMTA_size_16_color_FFFFFF_t_70 9]: /images/20220222/793f7e1c2a5845b29c7df80cf165207b.png