package com.xiaopeng.operation.service.impl;
|
|
import com.xiaopeng.api.model.operation.XpGeoQueryEntity;
|
import com.xiaopeng.api.vo.automaticPilot.VehiclePosition;
|
import com.xiaopeng.operation.repository.XpGeoQueryEntityRepository;
|
import com.xiaopeng.operation.service.IXpGeoQueryEntityService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.geo.Distance;
|
import org.springframework.data.geo.GeoResults;
|
import org.springframework.data.geo.Metrics;
|
import org.springframework.data.geo.Point;
|
import org.springframework.data.mongodb.core.MongoTemplate;
|
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.data.mongodb.core.query.Update;
|
import org.springframework.stereotype.Service;
|
|
import java.util.Calendar;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* MongoDB 地理查询服务实现类
|
*/
|
@Service
|
public class XpGeoQueryEntityServiceImpl implements IXpGeoQueryEntityService {
|
|
@Autowired
|
private XpGeoQueryEntityRepository repository;
|
|
@Autowired
|
private MongoTemplate mongoTemplate;
|
|
/**
|
* 通过经纬度查询固定范围500米内、一周内、指定时段的数据
|
* @param longitude 经度
|
* @param latitude 纬度
|
* @param time 时段（格式：HH-HH，例如：19-20）
|
* @return 符合条件的实体列表
|
*/
|
@Override
|
public List<XpGeoQueryEntity> findByLocationWithin500MetersAndLastWeekAndStatusGreater(
|
double longitude,
|
double latitude,
|
String time
|
) {
|
// // 计算一周前的时间
|
Calendar calendar = Calendar.getInstance();
|
calendar.setTime(new Date());
|
calendar.add(Calendar.DAY_OF_YEAR, -7);
|
Long startTime = calendar.getTime().getTime();
|
|
// 1. 创建查询点
|
Point referencePoint = new Point(longitude, latitude);
|
|
// 2. 构建 NearQuery - 完全使用 NearQuery 的 API
|
NearQuery nearQuery = NearQuery.near(referencePoint, Metrics.KILOMETERS)
|
.maxDistance(0.5) // 转成公里
|
.spherical(true);
|
|
|
// 3.时间>startTime time = time参数
|
// 3. 创建额外的查询条件
|
Query query = new Query();
|
query.addCriteria(Criteria.where("timestamp").gte(startTime));
|
query.addCriteria(Criteria.where("time").is(time));
|
|
// 将 NearQuery 和额外的查询条件合并
|
nearQuery.query(query);
|
|
|
// 4. 执行查询 - 明确指定集合名
|
GeoResults<XpGeoQueryEntity> geoResults = mongoTemplate.geoNear(
|
nearQuery,
|
XpGeoQueryEntity.class,
|
"xp_geo_query_entity" // 明确指定集合名，避免映射错误
|
);
|
|
// 5. 提取结果
|
return geoResults.getContent().stream()
|
.map(geoResult -> geoResult.getContent())
|
.collect(Collectors.toList());
|
|
}
|
|
/**
|
* 保存地理查询实体
|
* @param entity 地理查询实体
|
* @return 保存后的实体
|
*/
|
@Override
|
public XpGeoQueryEntity save(XpGeoQueryEntity entity) {
|
return repository.save(entity);
|
}
|
|
|
|
@Override
|
public XpGeoQueryEntity updateByOrderId(String orderId) {
|
Query query = new Query(Criteria.where("orderId").is(orderId));
|
|
XpGeoQueryEntity xpGeoQueryEntity = mongoTemplate.findOne(query, XpGeoQueryEntity.class);
|
Long timestamp = xpGeoQueryEntity.getTimestamp();
|
long time = new Date().getTime();
|
// 计算2个时间的秒数差
|
long seconds = (time- timestamp) / 1000;
|
Update update = new Update().set("seconds", seconds).set("status", 1);
|
if(xpGeoQueryEntity.getStatus()==0){
|
mongoTemplate.findAndModify(query, update, XpGeoQueryEntity.class);
|
}
|
return null;
|
|
}
|
|
/**
|
* 根据ID查询实体
|
* @param id 实体ID
|
* @return 实体
|
*/
|
@Override
|
public XpGeoQueryEntity findById(String id) {
|
return repository.findById(id).orElse(null);
|
}
|
|
/**
|
* 删除实体
|
* @param id 实体ID
|
*/
|
@Override
|
public void deleteById(String id) {
|
repository.deleteById(id);
|
}
|
|
/**
|
* 查询所有实体
|
* @return 实体列表
|
*/
|
@Override
|
public List<XpGeoQueryEntity> findAll() {
|
return repository.findAll();
|
}
|
}
|
