对复合对象数组以及日期范围的 Elasticsearch 查询

Question

您好，我有一个关于如何为具有日期范围和其他字段参数的嵌套复合对象创建弹性搜索查询的问题

[{
            "name": "A",
            "availability": [
                {
                    "partial": true,
                    "dates": {
                        "gte": "2020-12-01",
                        "lte": "2020-12-02"
                    }
                }
            ]
        },
        {
            "name": "B",
            "availability": [
                {
                    "partial": true,
                    "dates": {
                        "gte": "2020-12-05",
                        "lte": "2020-12-06"
                    }
                },
                {
                    "partial": false,
                    "dates": {
                        "gte": "2020-12-08",
                        "lte": "2020-12-11"
                    }
                }
            ]
        }]

这是我的实体数据

@Document(indexName = "workers")
public class Worker {
    @Id
    private String id;

    @Field(type = FieldType.Text)
    private String name;

    @Field(type = FieldType.Nested)
    private List<Availability> availability;
}

public class Availability {
    @Field(type = FieldType.Boolean)
    private boolean partial;
    
    @Field(type = FieldType.Date_Range, format = DateFormat.custom, pattern = "uuuu-MM-dd")
    private Map<String, LocalDate> dates;
}

这是我目前编写的搜索查询，但结果为空

         final BoolQueryBuilder queryBuilder = QueryBuilders.boolQuery();
queryBuilder.must(QueryBuilders.termQuery("availability.partial", query.isPartial()));
         
         RangeQueryBuilder availability = QueryBuilders.rangeQuery("availability.dates")
                   .gte(query.getStartDate())
                   .lte(query.getEndDate());
queryBuilder.must(availability);
        
        Pageable pageable = PageRequest.of(pageNumber, pageSize);

        // @formatter:off
        return new NativeSearchQueryBuilder()
                .withPageable(pageable)
                .withQuery(queryBuilder)
                .build();

这是我的查询 dto

public class WorkerQuery {
    private boolean partial;
    private LocalDate startDate;
    private LocalDate endDate;
}

// Request data
{
    "partial": true,
    "startDate": "2020-12-01",
    "endDate": "2020-12-02"
}

Answer 1

伟大的开始！因为availability 是nested，所以您只是缺少nested 查询。 Java 查询需要是这样的：

final BoolQueryBuilder queryBuilder = QueryBuilders.boolQuery();
queryBuilder.must(QueryBuilders.termQuery("availability.partial", query.isPartial()));

RangeQueryBuilder availability = QueryBuilders.rangeQuery("availability.dates")
               .gte(query.getStartDate())
               .lte(query.getEndDate())
               .relation("within");
queryBuilder.must(availability);

final NestedQueryBuilder nested = QueryBuilders.nestedQuery("availability", queryBuilder);

Pageable pageable = PageRequest.of(pageNumber, pageSize);

// @formatter:off
return new NativeSearchQueryBuilder()
            .withPageable(pageable)
            .withQuery(nested)
            .build();