ES之GEO_Point 经纬度查询:圆形、矩形、多边形方式

发布时间 2023-06-26 15:10:13作者: 古语云

1、GEO_Point 经纬度查询:圆形、矩形、多边形方式

geoDistanceRangeQuery 圆形查询
geoBoundingBoxQuery 矩形查询
geoPolygonQuery 多边形查询

2、创建mapping

参考:https://blog.csdn.net/wuzhiwei549/article/details/80537753

    @Test
    public void testSettingsMappingsGeo() throws IOException {
        //1:settings
        HashMap<String, Object> settings_map = new HashMap<String, Object>(2);
        settings_map.put("number_of_shards", 3);
        settings_map.put("number_of_replicas", 2);

        //2:mappings(映射、schema)
        XContentBuilder builder = XContentFactory.jsonBuilder()
                .startObject()
                .field("dynamic", "true")
                //设置type中的属性
                .startObject("properties")
                //.startObject("pin")
               // .startObject("properties")
                .startObject("location")
                    .field("type","geo_point")
                .endObject()
               // .endObject()
               // .endObject()
                .endObject()
                .endObject();

        CreateIndexRequestBuilder prepareCreate = client.admin().indices().prepareCreate("carshop");
        //管理索引(user_info)然后关联type(user)
        prepareCreate.setSettings(settings_map).addMapping("shop", builder).get();
    }

3、创建索引,添加数据

@Test
    public void testCreateGeo() throws IOException {
        List<GeoPoint> points = new ArrayList<>();
        points.add(new GeoPoint(40.12, -71.34));
        IndexResponse response = client.prepareIndex("carshop", "shop", "2")
                .setSource(
                        jsonBuilder()
                                .startObject()
                                .field("name", "上海顺丰宝马4S店")
                                .field("location", points)
                                .endObject()
                ).get();
    }

4、基于经纬度搜索

@Test
    public void testQueryGeo() throws IOException {
        //第一个需求:搜索两个坐标点组成的一个区域
        SearchResponse searchResponse = client.prepareSearch("carshop")
                .setTypes("shop")
                .setQuery(QueryBuilders.geoBoundingBoxQuery("location")
                        .setCorners(40.73, -74.1, 40.01, -71.12))
                .get();

        for(SearchHit searchHit : searchResponse.getHits().getHits()) {
            System.out.println(searchHit.getSourceAsString());
        }
        System.out.println("======================================================");

        //第二个需求:指定一个区域,由三个坐标点,组成,比如上海大厦,东方明珠塔,上海火车站
        List<GeoPoint> points = new ArrayList<>();
        points.add(new GeoPoint(40.73, -74.1));
        points.add(new GeoPoint(40.01, -71.12));
        points.add(new GeoPoint(50.56, -90.58));

        searchResponse = client.prepareSearch("carshop")
                .setTypes("shop")
                .setQuery(QueryBuilders.geoPolygonQuery("location", points))
                .get();

        for(SearchHit searchHit : searchResponse.getHits().getHits()) {
            System.out.println(searchHit.getSourceAsString());
        }

        System.out.println("====================================================");

        //第三个需求:搜索距离当前位置在200公里内的4s店
        searchResponse = client.prepareSearch("carshop")
                .setTypes("shop")
                .setQuery(QueryBuilders.geoDistanceQuery("location")
                        .point(40, -70)
                        .distance(200, DistanceUnit.KILOMETERS))
                .get();

        for(SearchHit searchHit : searchResponse.getHits().getHits()) {
            System.out.println(searchHit.getSourceAsString());
        }

        client.close();
    }