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 距离（KM）
     * @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, 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);
    }

}