Examples with GeoBoundingBoxQueryBuilder org.opensearch.index.query.GeoBoundingBoxQueryBuilder used on opensource projects

Example 1 with GeoBoundingBoxQueryBuilder

use of org.opensearch.index.query.GeoBoundingBoxQueryBuilder in project OpenSearch by opensearch-project.

the class GeoHashGridIT method testFiltered.

public void testFiltered() throws Exception {
    GeoBoundingBoxQueryBuilder bbox = new GeoBoundingBoxQueryBuilder("location");
    bbox.setCorners(smallestGeoHash).queryName("bbox");
    for (int precision = 1; precision <= PRECISION; precision++) {
        SearchResponse response = client().prepareSearch("idx").addAggregation(AggregationBuilders.filter("filtered", bbox).subAggregation(geohashGrid("geohashgrid").field("location").precision(precision))).get();
        assertSearchResponse(response);
        Filter filter = response.getAggregations().get("filtered");
        GeoGrid geoGrid = filter.getAggregations().get("geohashgrid");
        for (GeoGrid.Bucket cell : geoGrid.getBuckets()) {
            String geohash = cell.getKeyAsString();
            long bucketCount = cell.getDocCount();
            int expectedBucketCount = expectedDocCountsForGeoHash.get(geohash);
            assertNotSame(bucketCount, 0);
            assertTrue("Buckets must be filtered", geohash.startsWith(smallestGeoHash));
            assertEquals("Geohash " + geohash + " has wrong doc count ", expectedBucketCount, bucketCount);
        }
    }
}