Spring Boot 集成 ElasticSearch

心已赠人 2021-07-31 18:09 357阅读 0赞

点击上方 **蓝色字体**，选择“标星公众号”

优质文章，第一时间送达

![640?wx\_fmt=jpeg][640_wx_fmt_jpeg]

作者 | ZeroOne01

链接 | blog.51cto.com/zero01/2134718

###     ###

### **Spring Boot集成ElasticSearch** ###

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

<parent>	
        <groupId>org.springframework.boot</groupId>	
        <artifactId>spring-boot-starter-parent</artifactId>	
        <version>2.0.3.RELEASE</version>	
        <relativePath/> 	
    </parent>	
    
    	
    <properties>	
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>	
        <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>	
        <java.version>1.8</java.version>	
        <elasticsearch.version>5.5.2</elasticsearch.version>	
    </properties>	
    
    	
    <dependencies>	
        <dependency>	
            <groupId>org.springframework.boot</groupId>	
            <artifactId>spring-boot-starter-web</artifactId>	
        </dependency>	
        <dependency>	
            <groupId>org.projectlombok</groupId>	
            <artifactId>lombok</artifactId>	
            <optional>true</optional>	
        </dependency>	
        <dependency>	
            <groupId>org.springframework.boot</groupId>	
            <artifactId>spring-boot-starter-test</artifactId>	
            <scope>test</scope>	
        </dependency>	
        <dependency>	
            <groupId>org.elasticsearch.client</groupId>	
            <artifactId>transport</artifactId>	
            <version>${elasticsearch.version}</version>	
        </dependency>	
    </dependencies>

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

@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;	
        }	
    }

**查询接口开发**

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

![640][]

该索引中有一些文档数据，如下：

![640][640 1]

在工程中新建一个controller包，在该包中新建一个 BookCrudController 类，用于演示es增删查改接口demo。我们首先来开发查询接口，代码如下：

@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进行测试，查询结果如下：

![640][640 2]

**新增接口开发**

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

@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进行测试，测试结果如下：

![640][640 3]

使用返回的id去查询我们刚刚添加的书籍数据，结果如下：

![640][640 4]

**删除接口开发**

代码如下：

@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进行测试，删除数据成功：

![640][640 5]

**更新接口开发**

代码如下：

@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的书籍数据：

![640][640 6]

修改书籍的标题和作者：

![640][640 7]

修改成功：

![640][640 8]

**复合查询接口开发**

代码如下：

@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);	
    }

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

![640][640 9]

如果喜欢本篇文章，欢迎转发、点赞。关注订阅号「Web项目聚集地」，回复「进群」即可进入无广告技术交流。

--------------------

**推荐阅读**

***1.***

***2. ***

***3.***

***4. ***

***5.***

***6.***

![640?wx\_fmt=jpeg][640_wx_fmt_jpeg 1]

**喜欢文章，点个** **在看** ** ** ![640?wx\_fmt=jpeg][640_wx_fmt_jpeg 2]

[640_wx_fmt_jpeg]: /images/20210731/5dbb9194fcbb43fca91068acde588841.png?wx_fmt=jpeg
[640]: /images/20210731/5f60af94032448329ee54d36ad5d8463.png
[640 1]: /images/20210731/0636bd21192d4f3b86a5c433591b15e9.png
[640 2]: /images/20210731/5a2dbca4c3694e74b818e5a0b92c67ed.png
[640 3]: /images/20210731/f4f00c92781f4b21bf301d67e3a030cc.png
[640 4]: /images/20210731/df9c7146bd274192962e40ef712efaa7.png
[640 5]: /images/20210731/b3682698787d4e649a550953b5ac710b.png
[640 6]: /images/20210731/d85b96f5c1544bbea9483d82c6f11e8a.png
[640 7]: /images/20210731/2be9f34c3b094efb96e96ddd3bf7c62c.png
[640 8]: /images/20210731/69a7683264f745d485064f912a86883d.png
[640 9]: /images/20210731/cac4d1c6a27f465685b568b479def4a1.png
[640_wx_fmt_jpeg 1]: /images/20210731/b4b7c9e5a9f34201b45148b1a29920ba.png?wx_fmt=jpeg
[640_wx_fmt_jpeg 2]: /images/20210731/8d2fbb2b5a1042cea2cd2d09eeffd692.png?wx_fmt=jpeg