SpringBoot整合Elasticsearch

SpringBoot整合Elasticsearch

文章目录

一、环境搭建二、测试类三、特别说明

---

一、环境搭建

项目结构

pom文件

    4.0.0

    com.mei
    SpringBootElasticsearch
    1.0-SNAPSHOT

    
        org.springframework.boot
        spring-boot-starter-parent
        2.3.6.RELEASE
        
    

    
        
            org.springframework.boot
            spring-boot-starter-test
            test
        

        
            org.springframework.boot
            spring-boot-test
        
        
            junit
            junit
        
        
            org.springframework
            spring-test
        

        
            org.springframework.boot
            spring-boot-starter-data-elasticsearch
        

        
            org.springframework.boot
            spring-boot-starter-web
        

        
            org.projectlombok
            lombok
            true
        

    

实体类

package com.mei.entity;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;
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;

@Data
@NoArgsConstructor
@AllArgsConstructor
@ToString
@document(indexName = "shopping", shards = 3, replicas = 1) // 创建一个叫shopping的索引，给它设置三个分片，每个分片设置一个副本
public class Product {
 //必须有 id,这里的 id 是全局唯一的标识，等同于 es 中的"_id"
 @Id
 private Long id;//商品唯一标识

 
 @Field(type = FieldType.Text, analyzer = "ik_max_word")
 private String title;//商品名称

 @Field(type = FieldType.Keyword)
 private String category;//分类名称

 @Field(type = FieldType.Double)
 private Double price;//商品价格

 @Field(type = FieldType.Keyword, index = false)
 private String images;//图片地址
}

DAO层

package com.mei.dao;

import com.mei.entity.Product;
import org.springframework.data.elasticsearch.repository.ElasticsearchRepository;
import org.springframework.stereotype.Repository;

@Repository
public interface ProductDao extends ElasticsearchRepository {
}

配置文件

elasticsearch:
  host: 47.96.107.33
  port: 9200

配置类

1.ElasticsearchRestTemplate是spring-data-elasticsearch项目中的一个类,和其他spring项目中的 template类似。
2.在新版的spring-data-elasticsearch 中，ElasticsearchRestTemplate 代替了原来的ElasticsearchTemplate,原因是ElasticsearchTemplate基于TransportClient，TransportClient即将在8.x 以后的版本中移除。所以，我们推荐使用ElasticsearchRestTemplate。
3.ElasticsearchRestTemplate基于RestHighLevelClient客户端的。需要自定义配置类，继承AbstractElasticsearchConfiguration，并实现elasticsearchClient()抽象方法，创建RestHighLevelClient对象。

package com.mei.config;

import lombok.Data;
import org.apache.http.HttpHost;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestClientBuilder;
import org.elasticsearch.client.RestHighLevelClient;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.elasticsearch.config.AbstractElasticsearchConfiguration;

@ConfigurationProperties(prefix = "elasticsearch")
@Configuration
@Data
public class ElasticsearchConfig extends AbstractElasticsearchConfiguration {
 private String host ;
 private Integer port ;

 @Override
 public RestHighLevelClient elasticsearchClient() {
  RestClientBuilder builder = RestClient.builder(new HttpHost(host, port));
  RestHighLevelClient restHighLevelClient = new RestHighLevelClient(builder);
  return restHighLevelClient;
 }
}

二、测试类

代码如下：

package com.mei.test;

import com.mei.dao.ProductDao;
import com.mei.entity.Product;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.index.query.TermQueryBuilder;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.ArrayList;
import java.util.List;

@RunWith(SpringRunner.class)
@SpringBootTest
public class SpringDataESProductDaoTest {
 @Autowired
 private ProductDao productDao;
 
 @Test
 public void save(){
  Product product = new Product();
  product.setId(2L);
  product.setTitle("华为手机");
  product.setCategory("手机");
  product.setPrice(2999.0);
  product.setImages("http://www.atguigu/hw.jpg");
  productDao.save(product);
 }
 //POSTMAN, GET http://localhost:9200/product/_doc/2

 //修改
 @Test
 public void update(){
  Product product = new Product();
  product.setId(2L);
  product.setTitle("小米 2 手机");
  product.setCategory("手机");
  product.setPrice(9999.0);
  product.setImages("http://www.atguigu/xm.jpg");
  productDao.save(product);
 }
 //POSTMAN, GET http://localhost:9200/product/_doc/2


 //根据 id 查询
 @Test
 public void findById(){
  Product product = productDao.findById(2L).get();
  System.out.println(product);
 }

 @Test
 public void findAll(){
  Iterable products = productDao.findAll();
  for (Product product : products) {
   System.out.println(product);
  }
 }

 //删除
 @Test
 public void delete(){
  Product product = new Product();
  product.setId(2L);
  productDao.delete(product);
 }
 //POSTMAN, GET http://localhost:9200/product/_doc/2

 //批量新增
 @Test
 public void saveAll(){
  List productList = new ArrayList<>();
  for (int i = 0; i < 10; i++) {
   Product product = new Product();
   product.setId(Long.valueOf(i));
   product.setTitle("["+i+"]小米手机");
   product.setCategory("手机");
   product.setPrice(1999.0 + i);
   product.setImages("http://www.atguigu/xm.jpg");
   productList.add(product);
  }
  productDao.saveAll(productList);
 }

 //分页查询
 @Test
 public void findByPageable(){
  //设置排序(排序方式，正序还是倒序，排序的 id)
  Sort sort = Sort.by(Sort.Direction.DESC,"id");
  int currentPage=0;//当前页，第一页从 0 开始， 1 表示第二页
  int pageSize = 5;//每页显示多少条
  //设置查询分页
  PageRequest pageRequest = PageRequest.of(currentPage, pageSize,sort);
  //分页查询
  Page productPage = productDao.findAll(pageRequest);
  for (Product Product : productPage.getContent()) {
   System.out.println(Product);
  }
 }

 
 @Test
 public void termQuery(){
  //  TermQueryBuilder termQueryBuilder = QueryBuilders.termQuery("title", "小米");
  MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery("title", "小米");
  Iterable products = productDao.search(termQueryBuilder);
  for (Product product : products) {
   System.out.println(product);
  }
 }
 
 @Test
 public void termQueryByPage(){
  int currentPage= 0 ;
  int pageSize = 5;
  //设置查询分页
  PageRequest pageRequest = PageRequest.of(currentPage, pageSize);
  TermQueryBuilder termQueryBuilder = QueryBuilders.termQuery("title", "小米");
  Iterable products =
   productDao.search(termQueryBuilder,pageRequest);
  for (Product product : products) {
   System.out.println(product);
  }
 }

}

三、特别说明

当系统初始化时，与实体类对应的索引会自动创建。