工作中ES的使用

1
@Document(indexName = "product")  // 索引名称
2
@Data
3
public class Product {
4

5
    @Id
6
    private Long id;
7

8
    @Field(type = FieldType.Text, analyzer = "ik_max_word")
9
    private String name;  // 商品名称，支持分词
10

11
    @Field(type = FieldType.Keyword)
12
    private String category;  // 分类，精确匹配
13

14
    @Field(type = FieldType.Double)
15
    private Double price;  // 价格
16

17
    @Field(type = FieldType.Integer)
18
    private Integer stock;  // 库存
19
}

1
@Repository
2
public interface ProductRepository extends ElasticsearchRepository<Product, Long> {
3

4
    // 按名称模糊查询
5
    List<Product> findByName(String name);
6

7
    // 按分类查询
8
    List<Product> findByCategory(String category);
9

10
    // 价格区间查询
11
    List<Product> findByPriceBetween(Double min, Double max);
12

13
    // 自定义查询：名称包含关键字
14
    @Query("{\"match\": {\"name\": \"?0\"}}")
15
    List<Product> searchByName(String keyword);
16
}

1
@Service
2
@Slf4j
3
public class ProductService {
4

5
    @Autowired
6
    private ProductRepository productRepository;
7

8
    // 1. 新增/更新文档
9
    public void save(Product product) {
10
        productRepository.save(product);
11
        log.info("保存成功");
12
    }
13

14
    // 2. 批量新增
15
    public void saveAll(List<Product> products) {
16
        productRepository.saveAll(products);
17
    }
18

19
    // 3. 根据ID查询
20
    public Product getById(Long id) {
21
        return productRepository.findById(id).orElse(null);
22
    }
23

24
    // 4. 查询所有
25
    public List<Product> getAll() {
26
        Iterable<Product> all = productRepository.findAll();
27
        List<Product> list = new ArrayList<>();
28
        all.forEach(list::add);
29
        return list;
30
    }
31

32
    // 5. 删除
33
    public void delete(Long id) {
34
        productRepository.deleteById(id);
35
    }
36

37
    // 6. 按名称搜索
38
    public List<Product> searchByName(String name) {
39
        return productRepository.findByName(name);
40
    }
41

42
    // 7. 价格区间查询
43
    public List<Product> searchByPriceRange(Double min, Double max) {
44
        return productRepository.findByPriceBetween(min, max);
45
    }
46
}

1
@Service
2
@Slf4j
3
public class ProductSearchService {
4

5
    @Autowired
6
    private ElasticsearchRestTemplate esTemplate;
7

8
    /**
9
     * 多条件搜索
10
     * 支持关键字搜索 + 分类筛选 + 价格区间
11
     */
12
    public List<Product> search(String keyword, String category, Double minPrice, Double maxPrice) {
13
        // 1. 构建查询条件
14
        NativeSearchQueryBuilder queryBuilder = new NativeSearchQueryBuilder();
15

16
        // bool查询（组合查询）
17
        BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
18

19
        // 关键字搜索（must：必须满足）
20
        if (StringUtils.isNotBlank(keyword)) {
21
            boolQuery.must(QueryBuilders.matchQuery("name", keyword));
22
        }
23

24
        // 分类筛选（filter：过滤，不参与评分）
25
        if (StringUtils.isNotBlank(category)) {
26
            boolQuery.filter(QueryBuilders.termQuery("category", category));
27
        }
28

29
        // 价格区间
30
        if (minPrice != null && maxPrice != null) {
31
            boolQuery.filter(QueryBuilders.rangeQuery("price")
32
                    .gte(minPrice)
33
                    .lte(maxPrice));
34
        }
35

36
        queryBuilder.withQuery(boolQuery);
37

38
        // 2. 执行查询
39
        SearchHits<Product> hits = esTemplate.search(queryBuilder.build(), Product.class);
40

41
        // 3. 处理结果
42
        return hits.getSearchHits().stream()
43
                .map(SearchHit::getContent)
44
                .collect(Collectors.toList());
45
    }
46

47
    /**
48
     * 分页查询
49
     */
50
    public Page<Product> searchPage(String keyword, int page, int size) {
51
        NativeSearchQueryBuilder queryBuilder = new NativeSearchQueryBuilder();
52

53
        // 关键字搜索
54
        if (StringUtils.isNotBlank(keyword)) {
55
            queryBuilder.withQuery(QueryBuilders.matchQuery("name", keyword));
56
        }
57

58
        // 分页
59
        queryBuilder.withPageable(PageRequest.of(page, size));
60

61
        // 排序（按价格升序）
62
        queryBuilder.withSort(SortBuilders.fieldSort("price").order(SortOrder.ASC));
63

64
        SearchHits<Product> hits = esTemplate.search(queryBuilder.build(), Product.class);
65

66
        // 转换结果
67
        List<Product> list = hits.getSearchHits().stream()
68
                .map(SearchHit::getContent)
69
                .collect(Collectors.toList());
70

71
        return new PageImpl<>(list, PageRequest.of(page, size), hits.getTotalHits());
72
    }
73
}

1
@Service
2
@Transactional
3
public class ProductServiceImpl implements ProductService {
4

5
    @Autowired
6
    private ProductMapper productMapper;  // MySQL
7

8
    @Autowired
9
    private ProductRepository productRepository;  // ES
10

11
    /**
12
     * 新增商品
13
     */
14
    public void addProduct(Product product) {
15
        // 1. 存MySQL
16
        productMapper.insert(product);
17

18
        // 2. 存ES
19
        productRepository.save(product);
20
    }
21

22
    /**
23
     * 更新商品
24
     */
25
    public void updateProduct(Product product) {
26
        // 1. 更新MySQL
27
        productMapper.updateById(product);
28

29
        // 2. 更新ES
30
        productRepository.save(product);  // ES的save是新增或更新
31
    }
32

33
    /**
34
     * 删除商品
35
     */
36
    public void deleteProduct(Long id) {
37
        // 1. 删除MySQL
38
        productMapper.deleteById(id);
39

40
        // 2. 删除ES
41
        productRepository.deleteById(id);
42
    }
43
}