Spring Data MongDB空间索引(判断一个点Point是否在一个区域Polygon内)

这里要连接MongoDB数据库，在配置文件里：spring.data.mongodb.uri = mongodb://root:root@localhost:27017/happy

两个root分别是用户名和密码，happy是数据库名。

 

首先创建一个实体类SiteExt.java

import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; import org.springframework.data.annotation.Id; import org.springframework.data.mongodb.core.geo.GeoJsonPoint; import org.springframework.data.mongodb.core.index.CompoundIndex; import org.springframework.data.mongodb.core.index.CompoundIndexes; import org.springframework.data.mongodb.core.mapping.Document; @Data @AllArgsConstructor @NoArgsConstructor @Document(collection = "SiteExt") @CompoundIndexes({ @CompoundIndex(name = "location_index", def = "{'location': '2dsphere'}"), }) public class SiteExt { @Id private String plateNo; private GeoJsonPoint location; }

注意这里使用了2dsphere索引

{'location': '2dsphere'}


测试用的就不再规范地写接口了，只写个实现类了。

SiteExtDao.java

import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.mongodb.core.BulkOperations; import org.springframework.data.mongodb.core.MongoTemplate; import org.springframework.data.mongodb.core.geo.GeoJsonPolygon; import org.springframework.data.mongodb.core.query.Criteria; import org.springframework.data.mongodb.core.query.Query; import org.springframework.stereotype.Repository; import xy.study.mongdb.model.SiteExt; import java.util.List; @Repository public class SiteExtDao { @Autowired private MongoTemplate mongoTemplate; public List<SiteExt> findPointInPolygon(GeoJsonPolygon geoJsonPolygon, String collectionName) { Query query = new Query(Criteria.where("location").within(geoJsonPolygon)); List<SiteExt> list = mongoTemplate.find(query, SiteExt.class,collectionName); return list; } public void insert(SiteExt siteExt, String collectionName) { mongoTemplate.insert(siteExt,collectionName); } public void insertBatch(List<SiteExt> list, String collectionName) { // BulkMode.UNORDERED:表示并行处理，遇到错误时能继续执行不影响其他操作；BulkMode.ORDERED：表示顺序执行，遇到错误时会停止所有执行
        BulkOperations ops = mongoTemplate.bulkOps(BulkOperations.BulkMode.UNORDERED, collectionName); ops.insert(list); // 执行操作
 ops.execute(); } public List<SiteExt> findAll(String collectionName) { Query query=new Query(); List<SiteExt> list = mongoTemplate.find(query, SiteExt.class,collectionName); return list; } }

 

注意这里创建查询条件这里：Query query = new Query(Criteria.where("location").within(geoJsonPolygon));
location是实体类SiteExt中建立了2dsphere索引的字段名，再调用within方法，传入区域对象GeoJsonPloygon。
GeoJsonPlaygon代表的是一个区域对象，这个类在org.springframework.data.mongodb.core.geo包下面。

先插入两条数据做测试：

@Autowired
private SiteExtDao siteExtDao;

        GeoJsonPoint geoJsonPoint1 = new GeoJsonPoint(new Point(113.330908,23.155678)); SiteExt siteExt1 = new SiteExt("A",geoJsonPoint1); GeoJsonPoint geoJsonPoint2 = new GeoJsonPoint(new Point(113.33831,23.137335)); SiteExt siteExt2 = new SiteExt("B",geoJsonPoint2);
 siteExtDao.insert(siteExt1,"SiteExt"); siteExtDao.insert(siteExt2,"SiteExt");

 

然后查询哪些点在指定区域的操作来了：

        Point p1 = new Point(113.314882,23.163055); Point p2 = new Point(113.355845,23.167042); Point p3 = new Point(113.370289,23.149564); Point p4 = new Point(113.356779,23.129758); Point p5 = new Point(113.338238,23.13913); Point p6 = new Point(113.330979,23.124706); Point p7 = new Point(113.313588,23.140858); Point p8 = new Point(113.323865,23.158204); Point p9 = new Point(113.314882,23.163055); List<Point> list = new ArrayList<>(); list.add(p1); list.add(p2); list.add(p3); list.add(p4); list.add(p5); list.add(p6); list.add(p7); list.add(p8); list.add(p9); 　　　　 //用9个点围成一个区域，首尾两个点p1和p9要相同，才能构成一个区域 GeoJsonPolygon geoJsonPolygon = new GeoJsonPolygon(list); 
　　　　 //传入区域和数据库表名 List<SiteExt> pointInPolygon = siteExtDao.findPointInPolygon(geoJsonPolygon,"SiteExt");

　　　　　　

        pointInPolygon.forEach(
             p -> {
                System.out.println(p.getLocation());
             }
        );

 

上面9个点围成了一个多边形，在"SiteExt"表中查询哪些记录在这个多边形区域中。
结果如下：
Point [x=113.330908, y=23.155678]

总结：就是建立2dsphere索引，利用查询语句Query query = new Query(Criteria.where("location").within(geoJsonPolygon))查询符合该区域的所有点;

MongoDB原生查询数据表中哪些点记录在指定的区域：https://www.jb51.net/article/105796.htm