package com.xiaopeng.operation.utils.mango;

import com.xiaopeng.api.vo.automaticPilot.VehiclePosition;
import org.springframework.data.domain.Sort;
import org.springframework.data.geo.Point;
import org.springframework.data.geo.Distance;
import org.springframework.data.geo.Metrics;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.List;

@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) {
        // 创建查询点
        Point referencePoint = new Point(longitude, latitude);
        
        // 设置最大距离（MongoDB使用公里或米，这里转换为公里）
        distanceInMeters = distanceInMeters / 1000;

        // 构建查询条件
        Query query = new Query();
        query.addCriteria(Criteria.where("point")
                .near(referencePoint)
                .maxDistance(distanceInMeters));
        query.addCriteria(Criteria.where("status").is(1)); // 运营中
        query.addCriteria(Criteria.where("state").is(0));  // 空闲中
        
        // 可以添加排序，按距离排序
         query.with(Sort.by(Sort.Direction.ASC, "point"));
        
//        // 可以限制返回数量
//         query.limit(num);
        
        return mongoTemplate.find(query, VehiclePosition.class);
    }


    /**
     * 根据车辆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);
    }

}