Java之elasticsearch geo 距离排序检索

/**
     * 根据经纬度定位用户所在地
     *
     * @param token
     * @return
     */
    @PostMapping("...")
    public AppResDto searchAreaByLatAndLon(@RequestParam("token") String token,
                                           @RequestParam("lat") Double lat,
                                           @RequestParam("lon") Double lon) throws Exception {
        
        //设定搜索半径

        GeoDistanceQueryBuilder queryBuilder = QueryBuilders.geoDistanceQuery("location")
                                                            .point(lat, lon)
                                                            .distance(1500, DistanceUnit.KILOMETERS)
                                                            .geoDistance(GeoDistance.PLANE);
        
        //按距离排序
        GeoDistanceSortBuilder sort = SortBuilders.geoDistanceSort("location", lat, lon);
        sort.order(SortOrder.ASC);
        sort.point(lat, lon);
        //构建检索
        SearchSourceBuilder searchSourceBuilder = SearchSourceBuilder.searchSource()
                                                                     .from(0)
                                                                     .size(20)
                                                                     .query(queryBuilder )
                                                                     .sort(sort);
        SearchHits searchHits = elasearchService.searchDocument(ElsIndexEnums.rz_area.getIndex(), searchSourceBuilder);

        ArrayList<Map> arrayList = new ArrayList<>();

        {//控制台测试显示！生产删除！
            searchHits.forEach(hit->{
                Map map = new HashMap();
                map.put("complete_path", hit.getSourceAsMap().get("complete_path"));
                //获取距离值，并保留两位小数点
                BigDecimal geoDis = new BigDecimal((Double) hit.getSortValues()[0]);
                //坐标
                Object location2 = hit.getSourceAsMap().get("location");
                Map<String, Object> hitMap = hit.getSourceAsMap();
                //计算距离
                hitMap.put("geoDistance", geoDis.setScale(0, BigDecimal.ROUND_HALF_DOWN));
                map.put("range", hit.getSourceAsMap().get("geoDistance"));
                arrayList.add(map);
                System.out.println(hit.getSourceAsMap().get("complete_path") + "的坐标：" +
                        location2 + "与我的距离" +
                        hit.getSourceAsMap().get("geoDistance") +
                        DistanceUnit.METERS.toString());
            });
        }

        Map<String, Object> map = new HashMap<>();
        if (searchHits.getHits().length > 0) {
            map = searchHits.getHits()[0].getSourceAsMap();
        }
        //return AppResDto.success(arrayList);
        return AppResDto.success(map);
    }

今天的文章Java之elasticsearch geo 距离排序检索分享到此就结束了，感谢您的阅读。