springboot整合Elasticsearch 入门实战

桃扇骨 2023-07-13 03:04 2阅读 0赞

**附：Elasticsearch 环境搭建，可参考：[Linux Centos7.5 安装elasticsearch(单机/集群)][Linux Centos7.5 _elasticsearch]**

**1.demo结构： **

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM3NDk1Nzg2_size_16_color_FFFFFF_t_70][]

**2.pom.xml**

<parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.2.5.RELEASE</version>
        <relativePath/> 
     </parent>
    
    .....
    
     <dependency>
         <groupId>org.springframework.boot</groupId>
         <artifactId>spring-boot-starter-data-elasticsearch</artifactId>
     </dependency>
     <dependency>
         <groupId>org.projectlombok</groupId>
         <artifactId>lombok</artifactId>
         <version>1.18.12</version>
         <scope>provided</scope>
      </dependency>

**3.实体类： **

package com.lucifer.esdemo.pojo;
    
    import lombok.Data;
    import org.springframework.data.annotation.Id;
    import org.springframework.data.elasticsearch.annotations.Document;
    import org.springframework.data.elasticsearch.annotations.Field;
    import org.springframework.data.elasticsearch.annotations.FieldType;
    
    /**
     * @author: Lucifer
     * @date: 2020/3/9
     * @description: 实体类
     */
    @Data
    @Document(indexName = "stu")
    public class Stu {
    
        @Id
        private Long stuId;
    
        @Field(store = true)
        private String name;
    
        @Field(store = true)
        private Integer age;
    
        @Field(store = true)
        private Float money;
    
        @Field(store = true,type = FieldType.Keyword)
        private String sign;
    
        @Field(store = true)
        private String description;
    
    }

**4.application.yml: **

spring:
      data:
        elasticsearch:
          cluster-name: lucifer-es-cluster
          cluster-nodes: 192.168.160.130:9300,192.168.160.133:9300,192.168.160.134:9300

**ps:cluster-name;cluster-nodes** 两个属性废弃了

**5.测试类：**

package com.lucifer.esdemo;
    
    import com.lucifer.esdemo.pojo.Stu;
    
    import org.elasticsearch.action.index.IndexRequest;
    import org.elasticsearch.action.search.SearchResponse;
    import org.elasticsearch.action.update.UpdateResponse;
    import org.elasticsearch.index.query.QueryBuilders;
    import org.elasticsearch.search.SearchHit;
    import org.elasticsearch.search.SearchHits;
    import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;
    import org.elasticsearch.search.fetch.subphase.highlight.HighlightField;
    import org.elasticsearch.search.sort.FieldSortBuilder;
    import org.elasticsearch.search.sort.SortBuilder;
    import org.elasticsearch.search.sort.SortOrder;
    import org.junit.jupiter.api.Test;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.boot.test.context.SpringBootTest;
    import org.springframework.data.domain.PageRequest;
    import org.springframework.data.domain.Pageable;
    import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;
    import org.springframework.data.elasticsearch.core.SearchResultMapper;
    import org.springframework.data.elasticsearch.core.aggregation.AggregatedPage;
    import org.springframework.data.elasticsearch.core.aggregation.impl.AggregatedPageImpl;
    import org.springframework.data.elasticsearch.core.query.*;
    
    import java.util.*;
    
    
    @SpringBootTest
    public class EsDemoApplicationTests {
    
        @Autowired
        private ElasticsearchTemplate elasticsearchTemplate;
    
        /**
         * 生成测试数据
         */
        @Test
        public void createStu() {
    
            List<IndexQuery> indexQueryList = new ArrayList<>();
            for (int i = 0; i < 10; i++) {
                Stu stu = new Stu();
                stu.setStuId((long) new Random().nextInt(1500));
                stu.setName("王五" + i);
                stu.setAge(new Random().nextInt(100));
                stu.setMoney(new Random().nextFloat() + 10);
                stu.setSign("隔壁老王" + i);
                stu.setDescription("不可描述" + i);
                indexQueryList.add(new IndexQueryBuilder().withObject(stu).build());
            }
            elasticsearchTemplate.bulkIndex(indexQueryList);
        }
    
    
        //==========================================【索引】====================================================
    
        /**
         * 创建索引，不过不建议使用ElasticsearchTemplate 代码的方式 创建
         */
        @Test
        public void createIndexStu() {
            Stu stu = new Stu();
            stu.setStuId(1002L);
            stu.setName("张三");
            stu.setAge(25);
            stu.setMoney(20.5F);
            stu.setSign("lucifer");
            stu.setDescription("hello world");
    
            IndexQuery indexQuery = new IndexQueryBuilder().withObject(stu).build();
            String index = elasticsearchTemplate.index(indexQuery);
            System.out.println("index:" + index);
        }
    
    
        /**
         * 删除索引
         */
        @Test
        public void deleteIndexStu() {
            boolean b = elasticsearchTemplate.deleteIndex(Stu.class);
            System.out.println("result:" + b);
        }
    
    
        //==========================================【修改文档】====================================================
    
        /**
         * 修改文档
         */
        @Test
        public void updateStuDoc() {
            Map<String, Object> sourceMap = new HashMap<>();
            sourceMap.put("money", 33.04f);
            sourceMap.put("age", 18);
    
            IndexRequest indexRequest = new IndexRequest();
            indexRequest.source(sourceMap);
    
            //    类似sql语法：    update stu set sign='abc',age=33,money=88.6 where docId='1002'
            UpdateQuery updateQuery = new UpdateQueryBuilder().withClass(Stu.class).withId("1002").withIndexRequest(indexRequest).build();
    
            UpdateResponse update = elasticsearchTemplate.update(updateQuery);
            System.out.println("result:" + update);
        }
    
        //=============================================【查询文档】=================================================
    
        /**
         * 查询文档
         */
        @Test
        public void getStuDoc() {
            GetQuery query = new GetQuery();
            query.setId("1002");
    
            Stu stu = elasticsearchTemplate.queryForObject(query, Stu.class);
            System.out.println("result:" + stu);
        }
    
        //==========================================【删除文档】======================================================
    
        /**
         * 删除文档
         */
        @Test
        public void deleteStuDoc() {
            elasticsearchTemplate.delete(Stu.class, "1002");
        }
    
        //==========================================【分页查询文档】======================================================
        @Test
        public void searchStuDoc() {
            PageRequest pageRequest = PageRequest.of(0, 10);
    
            SearchQuery searchQuery = new NativeSearchQueryBuilder().
                    withQuery(QueryBuilders.matchQuery("description", "hello 不可")).
                    withPageable(pageRequest).build();
    
            AggregatedPage<Stu> stus = elasticsearchTemplate.queryForPage(searchQuery, Stu.class);
            System.out.println("检索后总分页数目为:" + stus.getTotalPages());
            for (Stu stu : stus) {
                System.out.println(stu);
            }
        }
    
    
        //==========================================【高亮】======================================================
        @Test
        public void highlightStuDoc() {
            String preTag = "<font color='red'>";
            String postTags = "</font>";
    
            //按年龄排序（倒序）
            SortBuilder sortBuilder = new FieldSortBuilder("age");
            sortBuilder.order(SortOrder.DESC);
    
            PageRequest pageRequest = PageRequest.of(0, 5);
            SearchQuery searchQuery = new NativeSearchQueryBuilder().
                    withQuery(QueryBuilders.matchQuery("description", "hello 不可")).
                    withHighlightFields(new HighlightBuilder.Field("description").
                            preTags(preTag).
                            postTags(postTags)).
                    withSort(sortBuilder).
                    withPageable(pageRequest).build();
    
            AggregatedPage<Stu> stus = elasticsearchTemplate.queryForPage(searchQuery, Stu.class, new SearchResultMapper() {
                @Override
                public <T> AggregatedPage<T> mapResults(SearchResponse response, Class<T> clazz, Pageable pageable) {
    
                    List<Stu> hitsHighlight = new ArrayList<>();
                    SearchHits hits = response.getHits();
                    for (SearchHit hit : hits) {
                        Map<String, HighlightField> highlightFields = hit.getHighlightFields();
                        HighlightField highlightField = highlightFields.get("description");
                        String description = highlightField.getFragments()[0].toString();
    
                        Object stuId = hit.getSourceAsMap().get("stuId");
                        String name = (String) hit.getSourceAsMap().get("name");
                        Integer age = (Integer) hit.getSourceAsMap().get("age");
                        String sign = (String) hit.getSourceAsMap().get("sign");
                        Object money = hit.getSourceAsMap().get("money");
    
                        Stu stu = new Stu();
                        stu.setStuId(Long.valueOf(stuId.toString()));
                        stu.setName(name);
                        stu.setAge(age);
                        stu.setSign(sign);
                        stu.setMoney(Float.valueOf(money.toString()));
                        stu.setDescription(description);
    
                        hitsHighlight.add(stu);
                    }
                    if (hitsHighlight.size() > 0) {
                        return new AggregatedPageImpl<>((List<T>) hitsHighlight);
                    }
                    return null;
                }
    
                @Override
                public <T> T mapSearchHit(SearchHit searchHit, Class<T> type) {
                    return null;
                }
            });
            System.out.println("检索后总分页数目为:" + stus.getTotalPages());
            for (Stu stu : stus) {
                System.out.println(stu);
            }
        }
    
    
    }

[Linux Centos7.5 _elasticsearch]: https://blog.csdn.net/qq_37495786/article/details/104735472
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM3NDk1Nzg2_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200309175357711.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM3NDk1Nzg2,size_16,color_FFFFFF,t_70