Elasticsearch 从 0 到 1（四）：Spring Boot 集成与商品搜索实战#

1
<dependency>
2
    <groupId>co.elastic.clients</groupId>
3
    <artifactId>elasticsearch-java</artifactId>
4
    <version>8.14.0</version>
5
</dependency>

1
@Configuration
2
public class ElasticsearchConfig {
3

4
    @Bean
5
    public ElasticsearchClient elasticsearchClient() {
6
        RestClient restClient = RestClient.builder(
7
                new HttpHost("localhost", 9200)
8
        ).build();
9

10
        ElasticsearchTransport transport = new RestClientTransport(
11
                restClient,
12
                new JacksonJsonpMapper()
13
        );
14

15
        return new ElasticsearchClient(transport);
16
    }
17
}

1
public class ProductDocument {
2

3
    private Long productId;
4
    private String title;
5
    private String brand;
6
    private String category;
7
    private Integer price;
8
    private Integer sales;
9
    private Integer stock;
10
    private String status;
11
    private LocalDateTime createdAt;
12

13
    // getter setter
14
}

1
@Service
2
public class ProductSearchService {
3

4
    private static final String INDEX = "products";
5

6
    private final ElasticsearchClient client;
7

8
    public ProductSearchService(ElasticsearchClient client) {
9
        this.client = client;
10
    }
11

12
    public void save(ProductDocument document) throws IOException {
13
        client.index(i -> i
14
                .index(INDEX)
15
                .id(String.valueOf(document.getProductId()))
16
                .document(document)
17
        );
18
    }
19
}

1
public void delete(Long productId) throws IOException {
2
    client.delete(d -> d
3
            .index(INDEX)
4
            .id(String.valueOf(productId))
5
    );
6
}

1
public class ProductSearchRequest {
2

3
    private String keyword;
4
    private String brand;
5
    private String category;
6
    private Integer minPrice;
7
    private Integer maxPrice;
8
    private String sortBy;
9
    private Integer pageNo = 1;
10
    private Integer pageSize = 10;
11

12
    // getter setter
13
}

1
sales_desc
2
price_asc
3
price_desc
4
newest

1
public SearchResponse<ProductDocument> search(ProductSearchRequest request) throws IOException {
2
    int from = (request.getPageNo() - 1) * request.getPageSize();
3

4
    return client.search(s -> s
5
                    .index(INDEX)
6
                    .from(from)
7
                    .size(request.getPageSize())
8
                    .query(q -> q.bool(b -> {
9
                        if (hasText(request.getKeyword())) {
10
                            b.must(m -> m.match(mm -> mm
11
                                    .field("title")
12
                                    .query(request.getKeyword())
13
                            ));
14
                        }
15
                        b.filter(f -> f.term(t -> t.field("status").value("ON_SALE")));
16
                        if (hasText(request.getBrand())) {
17
                            b.filter(f -> f.term(t -> t.field("brand").value(request.getBrand())));
18
                        }
19
                        if (hasText(request.getCategory())) {
20
                            b.filter(f -> f.term(t -> t.field("category").value(request.getCategory())));
21
                        }
22
                        if (request.getMinPrice() != null || request.getMaxPrice() != null) {
23
                            b.filter(f -> f.range(r -> {
24
                                r.field("price");
25
                                if (request.getMinPrice() != null) {
26
                                    r.gte(JsonData.of(request.getMinPrice()));
27
                                }
28
                                if (request.getMaxPrice() != null) {
29
                                    r.lte(JsonData.of(request.getMaxPrice()));
30
                                }
31
                                return r;
32
                            }));
33
                        }
34
                        return b;
35
                    }))
36
                    .highlight(h -> h
37
                            .fields("title", hf -> hf
38
                                    .preTags("<em>")
39
                                    .postTags("</em>")
40
                            )
41
                    ),
42
            ProductDocument.class
43
    );
44
}

1
List<ProductSearchVO> list = new ArrayList<>();
2

3
for (Hit<ProductDocument> hit : response.hits().hits()) {
4
    ProductDocument doc = hit.source();
5
    ProductSearchVO vo = ProductSearchVO.from(doc);
6

7
    List<String> titleHighlights = hit.highlight().get("title");
8
    if (titleHighlights != null && !titleHighlights.isEmpty()) {
9
        vo.setTitle(titleHighlights.get(0));
10
    }
11

12
    list.add(vo);
13
}

1
long total = response.hits().total() == null
2
        ? 0
3
        : response.hits().total().value();
4

5
return new PageResult<>(total, request.getPageNo(), request.getPageSize(), list);

1
{
2
  "total": 120,
3
  "pageNo": 1,
4
  "pageSize": 10,
5
  "list": [
6
    {
7
      "productId": 1001,
8
      "title": "无线<em>蓝牙</em>耳机",
9
      "brand": "SoundGo",
10
      "price": 199
11
    }
12
  ]
13
}

1
@RestController
2
@RequestMapping("/api/products/search")
3
public class ProductSearchController {
4

5
    private final ProductSearchService productSearchService;
6

7
    public ProductSearchController(ProductSearchService productSearchService) {
8
        this.productSearchService = productSearchService;
9
    }
10

11
    @GetMapping
12
    public PageResult<ProductSearchVO> search(ProductSearchRequest request) throws IOException {
13
        return productSearchService.searchProducts(request);
14
    }
15
}