地图距离排序二(Java篇)

我们在 mongodb 篇中介绍了如何使用 mongodb 进行地图的距离排序和筛选，今天我们来介绍一下，用 Java 和 SpringData 来实现的时候，需要注意的一些知识点。

首先，我们需要为实体设置地图索引。在 SpringData 中，我们可以通过使用 @GeoSpatialIndexed 注解来设置索引。但是这个索引注解默认使用的是2d索引，我们希望使用 2dsphere 索引，因此我们需要将重新设置这个注解的 type(type= GeoSpatialIndexType.GEO_2DSPHERE)。姿势如下：

@Document(collection = "location")
public class Location {

    @Field(value = "lng")
    private double lng;

    @Field(value = "lat")
    private double lat;

    @Field(value = "location")
    @GeoSpatialIndexed(type = GeoSpatialIndexType.GEO_2DSPHERE)
    private double[] location;
}

在设置好索引之后，我们就可以愉快的开始距离的排序了，第一种方式默认按照由近到远排序。姿势如下：

public List<Location> getNearLocation(double lng, double lat) {
    Query query = new Query();
    Point point = new Point(lng, lat);
    Criteria locCri = Criteria.where("location").nearSphere(point);
    query.addCriteria(locCri);        
    return mongoAccess.find(query, Location.class);
}

除了上面这种方式，按照离我最近，还可以使用 Aggregate 来实现。姿势如下：

public List<Location> getNearLocation(double lng, double lat) {
    Point point = new Point(lng, lat);
    Sort disSort = new Sort(Sort.Direction.ASC, "distance");
    List<AggregationOperation> aggregationOperations = new ArrayList<>();
    aggregationOperations.add(Aggregation.geoNear(NearQuery.near(point).inKilometers().spherical(true), "distance"));
    aggregationOperations.add(Aggregation.sort(disSort));
    Aggregation aggregation = Aggregation.newAggregation(aggregationOperations);
    AggregationResults<Location> results = this.mongoAccess.aggregate(aggregation, Location.class, Location.class);
    return results.getMappedResults();
}

如果我们希望查询以某个点为中心的圆几公里以内的位置数据，那你的姿势可以如下：

public List<Location> getCircleLocation(double lng, double lat, double radius) {
    //获取半径内的所有位置
    Point point = new Point(lng, lat);
    Distance distance = new Distance(radius, Metrics.KILOMETERS);
    Circle circle = new Circle(point, distance);
    Criteria locCri = Criteria.where("location").withinSphere(circle);
    Query query = new Query();
    query.addCriteria(locCri);
    return mongoAccess.find(query, Location.class);
}

注：代码中的 mongoAccess 为 mongodb 的基本操作封装，大家替换为自己的实现即可。