package com.xiaopeng.map.mangodb.service.impl;
|
|
import com.mongodb.client.FindIterable;
|
import com.mongodb.client.MongoCollection;
|
import com.mongodb.client.model.Filters;
|
import com.xiaopeng.api.model.map.XpGeoFencingPolygon;
|
import com.xiaopeng.map.mangodb.model.GeoPoint;
|
import com.xiaopeng.map.mangodb.service.XpGeoFencingPolygonService;
|
import com.xiaopeng.map.utils.mangoUtils.MongoGeoFenceUtil;
|
import org.bson.Document;
|
import org.bson.conversions.Bson;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.mongodb.core.MongoTemplate;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.Collections;
|
import java.util.List;
|
|
@Service
|
public class XpGeoFencingPolygonServiceImpl implements XpGeoFencingPolygonService {
|
|
@Autowired
|
private MongoTemplate mongoTemplate;
|
|
@Override
|
public int create(XpGeoFencingPolygon xpGeoFencingPolygon) {
|
mongoTemplate.insert(xpGeoFencingPolygon);
|
return 1;
|
}
|
|
@Override
|
public XpGeoFencingPolygon findById(String id) {
|
return mongoTemplate.findById(id, XpGeoFencingPolygon.class);
|
}
|
|
@Override
|
public List<XpGeoFencingPolygon> findAll() {
|
return mongoTemplate.findAll(XpGeoFencingPolygon.class);
|
}
|
|
|
|
/**
|
* 方法3：查找包含指定点的所有围栏（支持自定义地理字段名称）
|
*/
|
public List<String> findFencesContainingPoint(GeoPoint point, String collectionName, String geoFieldName) {
|
MongoCollection<Document> collection = mongoTemplate.getCollection(collectionName);
|
List<String> fenceIds = new ArrayList<>();
|
// 查询所有包含该点的围栏
|
Bson filter = Filters.geoIntersects(geoFieldName, MongoGeoFenceUtil.toGeoJsonPoint(point.getLongitude(), point.getLatitude()));
|
FindIterable<Document> documents = collection.find(filter);
|
for (Document document : documents) {
|
fenceIds.add(document.getString("fenceId"));
|
}
|
|
return fenceIds;
|
}
|
|
|
/**
|
* 查找指定父围栏ID的禁行区域点位
|
* @param parentId 父围栏ID
|
* @param parentIdField 父围栏ID字段名
|
* @param collectionName 集合名称
|
* @return 禁行区域点位列表
|
*/
|
public List<Document> findPointsByParentId(String parentId, String parentIdField, String collectionName) {
|
MongoCollection<Document> collection = mongoTemplate.getCollection(collectionName);
|
List<Document> points = new ArrayList<>();
|
|
// 查询父围栏ID等于指定值的所有文档
|
Bson filter = Filters.eq(parentIdField, parentId);
|
|
collection.find(filter).forEach(doc -> {
|
points.add(doc);
|
});
|
|
return points;
|
}
|
|
|
|
/**
|
* 根据经纬度点位判断是否在指定的电子围栏中
|
* @param longitude 经度
|
* @param latitude 纬度
|
* @param fenceId 围栏ID
|
* @param collectionName 集合名称
|
* @return 是否在围栏中 true在 false不在
|
*/
|
public boolean isInFence(double longitude, double latitude, String fenceId, String collectionName) {
|
if (fenceId == null || fenceId.trim().isEmpty()) {
|
return false;
|
}
|
MongoCollection<Document> collection = mongoTemplate.getCollection(collectionName);
|
Bson fenceIdFilter = Filters.or(Filters.eq("fenceId", fenceId), Filters.eq("parentId", fenceId));
|
Bson geoFilter = Filters.geoIntersects("points", MongoGeoFenceUtil.toGeoJsonPoint(longitude, latitude));
|
Bson filter = Filters.and(fenceIdFilter, geoFilter);
|
return collection.find(filter).limit(1).first() != null;
|
}
|
|
|
}
|
