springboot集成elasticsearch，实现搜索提示补全功能_spring boot整合 es自动补全

springboot集成elasticsearch，通过实体类创建索引，实现搜索提示补全功能

---

注：在此之前需要下载elasticsearch以及拼音分词器。可以查看这篇文章，免费下载，下载完直接解压就行。https://download.csdn.net/download/weixin_47874230/86514890

一、引入pom依赖

spring-data-elasticsearch版本需要与springboot版本对应，此处使用springboot版本为2.2.13.RELEASE

        <dependency>
            <groupId>org.springframework.data</groupId>
            <artifactId>spring-data-elasticsearch</artifactId>
            <version>3.2.12.RELEASE</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
1
2
3
4
5
6
7
8
9
10

二、修改yml文件

server:
  port: 8080
spring:
  elasticsearch:
    rest:
      uris:
        - http://localhost:9200

1
2
3
4
5
6
7
8

三、创建实体类GoodsDoc.java

import com.fasterxml.jackson.annotation.JsonFormat;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.data.annotation.Id;
import org.springframework.data.elasticsearch.annotations.*;
import org.springframework.data.elasticsearch.core.completion.Completion;
import java.util.Date;


@Data
@AllArgsConstructor
@NoArgsConstructor
@Document(indexName = "goods", shards = 1, replicas = 0)  //type = "_doc" ,
public class GoodsDoc {

    @Id
    @Field(type = FieldType.Keyword)
    private String id;

    @CompletionField(analyzer = "ik_smart", searchAnalyzer = "ik_smart", maxInputLength = 100)
    private Completion title;

    @Field(type = FieldType.Text)       //测试用，当name和title值相同时，检索结果区别,title查出结果数据较少，name为正常查询
    private String name;

    @Field(type = FieldType.Float)
    private float price;

    @Field(type = FieldType.Text)
    private String desc;

    @Field(type = FieldType.Date, format = DateFormat.custom, pattern = "yyyy-MM-dd HH:mm:ss")
    @JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
    private Date createdTime;

    @Field(type = FieldType.Date, format = DateFormat.custom, pattern = "yyyy-MM-dd HH:mm:ss")
    @JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
    private Date modifiedTime;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

四、创建GoodsMapper.java

import com.example.elasticsearch7demo.entity.GoodsDoc;
import org.springframework.data.elasticsearch.repository.ElasticsearchRepository;
import org.springframework.stereotype.Repository;

@Repository
public interface GoodsMapper extends ElasticsearchRepository<GoodsDoc, String> {

}

1
2
3
4
5
6
7
8
9

五、创建GoodsController.java

包含创建索引，添加数据，测试功能
1

import com.example.elasticsearch7demo.mapper.GoodsMapper;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.search.suggest.Suggest;
import org.elasticsearch.search.suggest.SuggestBuilder;
import org.elasticsearch.search.suggest.SuggestBuilders;
import org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.elasticsearch.core.ElasticsearchRestTemplate;
import org.springframework.data.elasticsearch.core.completion.Completion;
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;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

@RestController
@RequestMapping
public class GoodsController {

    @Autowired
    private ElasticsearchRestTemplate restTemplate;
    @Autowired
    private GoodsMapper goodsMapper;


    //创建索引和映射
    @GetMapping("/create")
    public void testCreateIndex(){
        restTemplate.createIndex(GoodsDoc.class);
        restTemplate.putMapping(GoodsDoc.class);

    }

    //添加数据
    @GetMapping("/insert")
    public void addData() throws Exception {
        List<GoodsDoc> orderDocs = new ArrayList<>();

        List<String> suggestList =  new ArrayList<>();
        String title = "笔记本电脑";
        suggestList.add(title); //可以把多个内容作为suggest的数据源
        Completion suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        GoodsDoc orderDoc = new GoodsDoc("1", suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        suggestList =  new ArrayList<>();
        title = "笔记本";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("2", suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        suggestList =  new ArrayList<>();
        title = "笔记本支架";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("3",  suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        suggestList =  new ArrayList<>();
        title = "笔记本内存条";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("4",  suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        suggestList =  new ArrayList<>();
        title = "联想笔记本";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("5",  suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);



        suggestList =  new ArrayList<>();
        title = "华硕笔记本";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("6",  suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        suggestList =  new ArrayList<>();
        title = "联想台式机";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("7",  suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        suggestList =  new ArrayList<>();
        title = "笔记本电脑是联想牌子的";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("8",  suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        suggestList =  new ArrayList<>();
        title = "你喜欢什么牌子的笔记本？";
        suggestList.add(title);
        suggest = new Completion(suggestList.toArray(new String[suggestList.size()]));
        orderDoc = new GoodsDoc("9", suggest,title, 100,"笔记本电脑非常方便",new Date(),new Date());
        orderDocs.add(orderDoc);

        goodsMapper.saveAll(orderDocs);
    }

    //测试
    @GetMapping("/test")
    public List<String> search(@RequestParam("keywords") String keywords){
        // 使用suggest进行标题联想
        CompletionSuggestionBuilder suggest = SuggestBuilders.completionSuggestion("title")
                //根据什么前缀来联想
//                .prefix("笔记本")
                .prefix(keywords)
                // 跳过重复过滤
                .skipDuplicates(true)
                // 匹配数量
                .size(10);
        SuggestBuilder suggestBuilder = new SuggestBuilder();
        suggestBuilder.addSuggestion("title-suggest",suggest);

        //执行查询
        SearchResponse suggestResp = restTemplate.suggest(suggestBuilder, GoodsDoc.class);

        //拿到Suggest结果
        Suggest.Suggestion<? extends Suggest.Suggestion.Entry<? extends Suggest.Suggestion.Entry.Option>> orderSuggest = suggestResp
                .getSuggest().getSuggestion("title-suggest");

        // 处理返回结果
        List<String> suggests = orderSuggest.getEntries().stream()
                .map(x -> x.getOptions().stream()
                        .map(y->y.getText().toString())
                        .collect(Collectors.toList())).findFirst().get();

        // 输出内容
        for (String str : suggests) {
            System.out.println("自动补全 = " + str);
        }
        return suggests;
    }

}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148

六、结果

总结

	以上就是记录的es实体类的创建以及搜索提示补全功能内容。
1