十一.全文检索ElasticSearch经典入门-SpringBoot整合ElasticSearch7

前言

你可以通过官网文档看到ElasticSearch7和6的变化，本文的重点在于在SpringBoot环境中使用ElasticSearch7。

软件安装

从官网下载ElasticSearch ，我这里使用的是7.9.3版本 ， 下载Kibana也是7.9.3，以及IK分词器，下载好之后把ElasticSearch进行解压： 把IK分词器解压到plugins/ik目录中，如下： 解压Kibana，目录结构如下

启动ElasticSearch ：elasticsearch-7.9.3\bin\elasticsearch.bat， 然后启动Kibana ：\kibana-7.9.3-windows-x86_64\bin\kibana.bat,访问kibana:http://localhost:5601

Spring整合ES7

第一步：导入依赖,SpringBoot使用的是2.4；需要注意版本兼容7.9.3

<parent> <groupId>org.springframework.boot @Id @Field(type = FieldType.Keyword) private String id; @Field(type = FieldType.Long) private Long orderNo; @Field(type = FieldType.Text, analyzer = "ik_smart", searchAnalyzer = "ik_max_word") private String title; //用来封装高亮的结果 private Map<String, List<String>> highlights; public OrderDoc(String id, Long orderNo ,String title){ this.id = id; this.orderNo = orderNo; this.title =title; } } 

第四步：编写Repository接口

@Repository //泛型是Doc对象和ID的类型 public interface OrderRepository extends ElasticsearchRepository<OrderDoc, String> { } 

第五步：编写Service接口,定义基本的CRUD方法

public interface OrderDocService { void saveAll(List<OrderDoc> orderDocs); OrderDoc findById(String id); void deleteById(String id); void updateById(OrderDoc orderDoc); PageResponse<OrderDoc> findList(OrderDoc orderDoc, Integer pageIndex, Integer pageSize); PageResponse<OrderDoc> findAll(Integer pageIndex, Integer pageSize); PageResponse<OrderDoc> findHighlight(OrderDoc orderDoc, Integer pageIndex, Integer pageSize); } 

下面是实现类，注入OrderRepository和ElasticsearchRestTemplate实现CRUD工作

@Service public class OrderDocServiceImpl implements OrderDocService { @Autowired private OrderRepository orderRepository; @Autowired private ElasticsearchRestTemplate template; @Override public void saveAll(List<OrderDoc> orderDocs) { orderRepository.saveAll(orderDocs); } @Override public OrderDoc findById(String id) { return orderRepository.findById(id).orElse(null); } @Override public void deleteById(String id) { orderRepository.deleteById(id); } @Override public void updateById(OrderDoc orderDoc) { orderRepository.save(orderDoc); } @Override public PageResponse<OrderDoc> findList(OrderDoc orderDoc, Integer pageIndex, Integer pageSize) { CriteriaQuery criteriaQuery = new CriteriaQuery( new Criteria().and("title").contains(orderDoc.getTitle()),PageRequest.of(pageIndex,pageSize)); SearchHits<OrderDoc> searchHits = template.search(criteriaQuery, OrderDoc.class); List<OrderDoc> orderDocList = searchHits.getSearchHits().stream() .map(orderDocSearchHit -> orderDocSearchHit.getContent()).collect(Collectors.toList()); return new PageResponse<OrderDoc>(searchHits.getTotalHits() ,orderDocList); } @Override public PageResponse<OrderDoc> findAll(Integer pageIndex, Integer pageSize) { Page<OrderDoc> page = orderRepository.findAll(PageRequest.of(pageIndex, pageSize)); return new PageResponse<OrderDoc>(page.getTotalElements(),page.getContent()); } @Override public PageResponse<OrderDoc> findHighlight(OrderDoc orderDoc, Integer pageIndex, Integer pageSize) { //查询条件 CriteriaQuery criteriaQuery = new CriteriaQuery( new Criteria().and("title").contains(orderDoc.getTitle()), PageRequest.of(pageIndex,pageSize)); //高亮字段 HighlightBuilder highlightBuilder = new HighlightBuilder(); highlightBuilder.field("title").preTags("").postTags(""); criteriaQuery.setHighlightQuery(new HighlightQuery(highlightBuilder)); //搜索结果 SearchHits<OrderDoc> searchHits = template.search(criteriaQuery, OrderDoc.class); List<OrderDoc> orderDocList = searchHits.get().map(searchHit -> { //把高亮结果添加到doc中中 OrderDoc doc = searchHit.getContent(); //获取高亮 doc.setHighlights(searchHit.getHighlightFields()); return doc; }).collect(Collectors.toList()); //处理结果 return new PageResponse<>(searchHits.getTotalHits() , orderDocList); } } 

下面是需要的结果对象

@Data @AllArgsConstructor @NoArgsConstructor public class PageResponse<T> { private Long total; private List<T> list; } 

第六步：编写单元测试

@RunWith(SpringRunner.class) @SpringBootTest(classes = ApplicationStarter.class) public class OrderDocServiceTest { @Autowired private OrderDocService orderDocService; @Test public void saveAll() { orderDocService.saveAll(Arrays.asList( new OrderDoc("1",100000l,"苹果电脑"), new OrderDoc("2",100000l,"苹果电脑"), new OrderDoc("3",100000l,"苹果电脑"), new OrderDoc("4",100000l,"苹果电脑"), new OrderDoc("5",100000l,"苹果电脑"), new OrderDoc("6",100000l,"苹果电脑"), new OrderDoc("7",100000l,"苹果电脑") )); } @Test public void findById() { System.out.println(orderDocService.findById("1")); } @Test public void deleteById() { orderDocService.deleteById("1"); } @Test public void updateById() { OrderDoc orderDoc = orderDocService.findById("1"); orderDoc.setTitle("华为电脑"); orderDocService.updateById(orderDoc); } @Test public void findList() { OrderDoc orderDoc = new OrderDoc(null, 100l, "电脑"); PageResponse<OrderDoc> response = orderDocService.findList(orderDoc, 0, 10); System.out.println(response.getTotal()); response.getList().forEach(System.out::println); } @Test public void findAll() { PageResponse<OrderDoc> response = orderDocService.findAll( 0, 10); System.out.println(response.getTotal()); response.getList().forEach(System.out::println); } @Test public void findHighlight() { OrderDoc orderDoc = new OrderDoc(null, 100l, "电脑",null); PageResponse<OrderDoc> response = orderDocService.findHighlight( orderDoc,0, 10); System.out.println(response.getTotal()); response.getList().forEach(System.out::println); } }