SpringBoot项目使用MongoDB距离查询

• 数据库
  

• 实体类

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.bson.types.ObjectId;
import org.springframework.data.mongodb.core.geo.GeoJsonPoint;
import org.springframework.data.mongodb.core.index.CompoundIndex;
import org.springframework.data.mongodb.core.mapping.Document;


@Data
@AllArgsConstructor
@NoArgsConstructor
@Document("places")
@CompoundIndex(name = "location_index",def = "{'location'}:{'2dsphere'}")
public class Places {
    private ObjectId id;
    private String title;
    private String address;
    private GeoJsonPoint location;
}

• 测试

import com.czxy.domain.Places;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.geo.*;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.geo.GeoJsonPoint;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.NearQuery;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.List;


@RunWith(SpringRunner.class)
@SpringBootTest(classes = MongoDemoApplication.class)
public class Test {
    @Autowired
    private MongoTemplate mongoTemplate;
    
    @org.junit.Test
    public void  test01(){
        //构造坐标点
        GeoJsonPoint point = new GeoJsonPoint(116.40, 39.91); //天安门坐标
        //构造半径
        Distance distanceObj = new Distance(1, Metrics.KILOMETERS);
        // 画了一个圆圈
        Circle circle = new Circle(point, distanceObj);
        // 构造query对象
        Query query = Query.query(Criteria.where("location").withinSphere(circle));
        // 省略其他内容
        List list = mongoTemplate.find(query, Places.class);
        list.forEach(System.out::println);
    }
    
    @org.junit.Test
    public void testGeoNear() {
        //构建中心点
        GeoJsonPoint point = new GeoJsonPoint(116.404, 39.915);
        //创建NearQuery对象
        NearQuery nearQuery = NearQuery.near(point,Metrics.KILOMETERS).maxDistance(1,Metrics.KILOMETERS); //maxDistance 最大距离
        //发送查询
        GeoResults results = mongoTemplate.geoNear(nearQuery, Places.class);
        for (GeoResult result : results) {
            System.out.println(result.getContent());
            System.out.println("距离"+result.getDistance().getValue()+"km");
        }
    }
}