SpringBoot + Elasticsearch7.6实现查询及高亮分词查询
需要先安装Elasticsearch7.6环境。
1、导入spring-data-elasticsearch依赖。(版本需要与Elasticsearch一致,还需要注意自己的springboot版本是否支持)
<dependency>
<groupId>org.springframework.data</groupId>
<artifactId>spring-data-elasticsearch</artifactId>
<version>7.6.2.RELEASE</version>
</dependency>
2、引入elasticsearch配置文件
yml
elasticSearch:
url:127.0.0.1:9000
@Configuration
public class EsConf {
@Value("${elasticSearch.url}")
private String url;
@Bean
RestHighLevelClient client() {
ClientConfiguration clientConfiguration = ClientConfiguration.builder()
.connectedTo(url)//elasticsearch地址
.build();
return RestClients.create(clientConfiguration).rest();
}
}
3、创建实体类(可以按照自己的需求创建,注意注释)
@Data
@Document(indexName = "user")//索引名称 建议与实体类一致
public class User {
@Id
private Integer id;
@Field(type = FieldType.Auto)//自动检测类型
private Integer age;
@Field(type = FieldType.Keyword)//手动设置为keyword,但同时也就不能分词
private String name;
@Field(type = FieldType.Text,analyzer = "ik_smart",searchAnalyzer = "ik_max_word")//设置为text设置的为ik_max_word
private String info;
}
4、创建Elasticsearch服务(创建方法可以默认提示)
public interface EsUserService extends ElasticsearchRepository<User,Integer> {
//根据name查询
List<User> findByName(String name);
//根据name和info查询
List<User> findByNameAndInfo(String name,String info);
}
5、简单查询
这里直接使用假数据并没有连接数据库,还需要注意几个点:
-
elasticsearchTemplate需要采用ElasticsearchRestTemplate 原来的已经过时了
-
elasticsearch的mapping没有自动生成,这导致了我们在实体类中指定的分词器没有生效,所以我在导入数据的同时,手动导入了mapping
-
LogAnnotation是我自定义的注解,大家可以直接去掉
controller如下:
@RestController public class EsController { @Autowired private ElasticsearchRestTemplate elasticsearchTemplate; @Autowired private EsUserService esUserService; private String[] names={"诸葛亮","曹操","李白","韩信","赵云","小乔","狄仁杰","李四","诸小明","王五"}; private String[] infos={"我来自中国的一个小乡村,地处湖南省","我来自中国的一个大城市,名叫上海,人们称作魔都" ,"我来自东北,家住大囤里,一口大碴子话"}; @LogAnnotation(requestRemark = "存数据") @GetMapping("saveUser") public ResultVO saveUser(){ //添加索引mapping 索引会自动创建但mapping自只用默认的这会导致分词器不生效 所以这里我们手动导入mapping elasticsearchTemplate.putMapping(User.class); Random random = new Random(); List<User> users = new ArrayList<>(); for (int i=0;i<20;i++){ User user = new User(); user.setId(i); user.setName(names[random.nextInt(9)]); user.setAge(random.nextInt(40)+i); user.setInfo(infos[random.nextInt(2)]); users.add(user); } Iterable<User> users1 = esUserService.saveAll(users); return new ResultVO(users1); } @LogAnnotation(requestRemark = "根据id查询数据") @GetMapping("getDataById") public ResultVO getDataById(Integer id){ return new ResultVO(esUserService.findById(id)); } @LogAnnotation(requestRemark = "分页查询所有数据") @GetMapping("getAllDataByPage") public ResultVO getAllDataByPage(){ //本该传入page和size,这里为了方便就直接写死了 Pageable page = PageRequest.of(0,10, Sort.Direction.ASC,"id"); Page<User> all = esUserService.findAll(page); return new ResultVO(all.getContent()); } @LogAnnotation(requestRemark = "根据名字查询") @GetMapping("getDataByName") public ResultVO getDataByName(String name){ return new ResultVO(esUserService.findByName(name)); } @LogAnnotation(requestRemark = "根据名字和介绍查询") @GetMapping("getDataByNameAndInfo") public ResultVO getDataByNameAndInfo(String name,String info){ //这里是查询两个字段取交集,即代表两个条件需要同时满足 return new ResultVO(esUserService.findByNameAndInfo(name,info)); } }
6、分词高亮查询
@LogAnnotation(requestRemark = "查询高亮显示")
@GetMapping("getHightByUser")
public ResultVO getHightByUser(String value){
//根据一个值查询多个字段 并高亮显示 这里的查询是取并集,即多个字段只需要有一个字段满足即可
//需要查询的字段
BoolQueryBuilder boolQueryBuilder= QueryBuilders.boolQuery()
.should(QueryBuilders.matchQuery("info",value))
.should(QueryBuilders.matchQuery("name",value));
//构建高亮查询
NativeSearchQuery searchQuery = new NativeSearchQueryBuilder()
.withQuery(boolQueryBuilder)
.withHighlightFields(
new HighlightBuilder.Field("info")
,new HighlightBuilder.Field("name"))
.withHighlightBuilder(new HighlightBuilder().preTags("<span style='color:red'>").postTags("</span>"))
.build();
//查询
SearchHits<User> search = elasticsearchTemplate.search(searchQuery, User.class);
//得到查询返回的内容
List<SearchHit<User>> searchHits = search.getSearchHits();
//设置一个最后需要返回的实体类集合
List<User> users = new ArrayList<>();
//遍历返回的内容进行处理
for(SearchHit<User> searchHit:searchHits){
//高亮的内容
Map<String, List<String>> highlightFields = searchHit.getHighlightFields();
//将高亮的内容填充到content中
searchHit.getContent().setName(highlightFields.get("name")==null ? searchHit.getContent().getName():highlightFields.get("name").get(0));
searchHit.getContent().setInfo(highlightFields.get("info")==null ? searchHit.getContent().getInfo():highlightFields.get("info").get(0));
//放到实体类中
users.add(searchHit.getContent());
}
return new ResultVO(users);
}