Elasticsearch：对聚合的数据选项进行条件过滤

落日映苍穹つ 2024-03-24 10:51 63阅读 0赞

import org.apache.http.HttpHost;
    import org.elasticsearch.client.RestClient;
    import org.elasticsearch.client.RestHighLevelClient;
     
    @Configuration
    public class ElasticsearchConfig {
     
     
        @Bean
        public RestHighLevelClient restHighLevelClient(){
            return new RestHighLevelClient(RestClient.builder(
                    HttpHost.create("http://192.168.150.101:9200")
            ));
        }
    }

@Data
    public class Demo {
     
        private String brand;
        private String city;
        private String starName;
    }

import com.alibaba.fastjson.JSON;
    import lombok.extern.slf4j.Slf4j;
    import org.apache.commons.lang3.StringUtils;
    import org.elasticsearch.action.search.SearchRequest;
    import org.elasticsearch.action.search.SearchResponse;
    import org.elasticsearch.client.RestHighLevelClient;
    import org.elasticsearch.index.query.QueryBuilders;
    import org.elasticsearch.search.SearchHit;
    import org.elasticsearch.search.SearchHits;
     
    @Slf4j
    @Service
    public class DemoService extends ServiceImpl<DemoMapper, Demo> implements IDemoService {
     
        @Autowired
        private RestHighLevelClient restHighLevelClient;
     
        @Override
        public PageResult search(RequestParams params) {
            try {
                // 1.准备Request
                SearchRequest request = new SearchRequest("demo");
                // 2.准备请求参数
                // 2.1.query
                String key = params.getKey();
                if (StringUtils.isNotBlank(key)) {
                    // 不为空，根据关键字查询
                    request.source().query(QueryBuilders.matchQuery("all", key));
                } else {
                    // 为空，查询所有
                    request.source().query(QueryBuilders.matchAllQuery());
                }
                // 2.2.分页
                int page = params.getPage();
                int size = params.getSize();
                request.source().from((page - 1) * size).size(size);
    
                // 品牌，城市，星级都是聚合数据
                // 3.1.品牌
                String brand = params.getBrand();
                if (StringUtils.isNotBlank(brand)) {
                    boolQuery.filter(QueryBuilders.termQuery("brand", brand));
                }
                // 3.3.城市
                String city = params.getCity();
                if (StringUtils.isNotBlank(city)) {
                    boolQuery.filter(QueryBuilders.termQuery("city", city));
                }
                // 3.4.星级
                String starName = params.getStarName();
                if (StringUtils.isNotBlank(starName)) {
                    boolQuery.filter(QueryBuilders.termQuery("starName", starName));
                }
                // 3.5.价格范围
                Integer minPrice = params.getMinPrice();
                Integer maxPrice = params.getMaxPrice();
                if (minPrice != null && maxPrice != null) {
                    maxPrice = maxPrice == 0 ? Integer.MAX_VALUE : maxPrice;
                    boolQuery.filter(QueryBuilders.rangeQuery("price").gte(minPrice).lte(maxPrice));
                }
     
                // 4.发送请求
                SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT);
                // 5.解析响应
                return handleResponse(response);
            } catch (IOException e) {
                throw new RuntimeException("搜索数据失败", e);
            }
        }
     
     
        private PageResult handleResponse(SearchResponse response) {
            SearchHits searchHits = response.getHits();
            // 4.1.总条数
            long total = searchHits.getTotalHits().value;
            // 4.2.获取文档数组
            SearchHit[] hits = searchHits.getHits();
            // 4.3.遍历
            List<Demo> demoList = new ArrayList<>(hits.length);
            for (SearchHit hit : hits) {
                // 4.4.获取source
                String json = hit.getSourceAsString();
                // 4.5.反序列化，非高亮的
                Demo demo= JSON.parseObject(json, Demo.class);
                // 4.9.放入集合
                demoList.add(demo);
            }
            return new PageResult(total, hotels);
        }
    }