package com.xiaopeng.operation.utils.mango;
|
|
import com.xiaopeng.api.vo.automaticPilot.VehiclePosition;
|
import org.springframework.data.domain.Sort;
|
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.stereotype.Service;
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@Service
|
public class VehiclePositionService {
|
|
@Resource
|
private MongoTemplate mongoTemplate;
|
|
/**
|
* 查询附近指定距离内且状态正常的车辆
|
* @param longitude 经度
|
* @param latitude 纬度
|
* @param distanceInMeters 距离（米）
|
* @return 符合条件的车辆列表
|
*/
|
public List<VehiclePosition> findAvailableVehiclesNearby(double longitude,
|
double latitude,
|
double distanceInMeters,int num) {
|
try {
|
// // 1. 创建查询点
|
// Point referencePoint = new Point(longitude, latitude);
|
//
|
// // 2. 构建 NearQuery - 完全使用 NearQuery 的 API
|
// NearQuery nearQuery = NearQuery.near(referencePoint, Metrics.KILOMETERS)
|
// .maxDistance(distanceInMeters / 1000) // 转成公里
|
// .spherical(true);
|
//
|
//
|
// // 4. 执行查询 - 明确指定集合名
|
// GeoResults<VehiclePosition> geoResults = mongoTemplate.geoNear(
|
// nearQuery,
|
// VehiclePosition.class,
|
// "vehicle_position" // 明确指定集合名，避免映射错误
|
// );
|
//
|
// // 5. 提取结果
|
// return geoResults.getContent().stream()
|
// .map(geoResult -> geoResult.getContent())
|
// .collect(Collectors.toList());
|
|
//找到中心点
|
GeoJsonPoint geoJsonPoint = new GeoJsonPoint(longitude, latitude);
|
//构造半径
|
Distance distanceR = new Distance(distanceInMeters / 1000, Metrics.KILOMETERS);
|
//画圆
|
Circle circle = new Circle(geoJsonPoint, distanceR);
|
// 构造query对象
|
Query query = Query.query(Criteria.where("point").withinSphere(circle));
|
List<VehiclePosition> vehiclePositions = mongoTemplate.find(query, VehiclePosition.class);
|
return vehiclePositions;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return new ArrayList<>();
|
}
|
}
|
|
|
/**
|
* 根据车辆ID查询车辆信息
|
* @param vehicleId 车辆ID
|
* @return 车辆信息，不存在则返回null
|
*/
|
public VehiclePosition findByVehicleId(Long vehicleId) {
|
Query query = new Query();
|
query.addCriteria(Criteria.where("vehicle_id").is(vehicleId));
|
return mongoTemplate.findOne(query, VehiclePosition.class);
|
}
|
|
|
/**
|
* 所有的车辆
|
*/
|
public List<VehiclePosition> findAll() {
|
|
// 构建查询条件
|
Query query = new Query();
|
query.addCriteria(Criteria.where("status").is(1)); // 运营中
|
query.addCriteria(Criteria.where("state").is(1)); // 接单中
|
|
// 可以添加排序，按距离排序
|
query.with(Sort.by(Sort.Direction.ASC, "point"));
|
|
return mongoTemplate.find(query, VehiclePosition.class);
|
}
|
|
}
|
