From 93ffa1e9e9ec29418078b4e26c51e51e66003813 Mon Sep 17 00:00:00 2001 From: liujie <1793218484@qq.com> Date: 星期二, 19 八月 2025 20:48:17 +0800 Subject: [PATCH] 是否包车 --- UserQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java | 50 ++++++++++++++++++++++++++++++-------------------- 1 files changed, 30 insertions(+), 20 deletions(-) diff --git a/UserQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java b/UserQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java index 9c730f0..70e2b5d 100644 --- a/UserQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java +++ b/UserQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java @@ -1,6 +1,8 @@ package com.stylefeng.guns.modular.api; +import com.alibaba.fastjson.JSON; import com.baomidou.mybatisplus.mapper.EntityWrapper; +import com.stylefeng.guns.core.util.ToolUtil; import com.stylefeng.guns.modular.system.model.Driver; import com.stylefeng.guns.modular.system.service.IDriverService; import com.stylefeng.guns.modular.system.service.IOrderEvaluateService; @@ -13,19 +15,19 @@ import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; +import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; -import org.springframework.data.geo.GeoResult; -import org.springframework.data.geo.GeoResults; 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.redis.core.RedisTemplate; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.bind.annotation.RestController; +import javax.annotation.Resource; import java.util.*; import java.util.stream.Collectors; @@ -35,6 +37,7 @@ @Api @RestController @RequestMapping("") +@Slf4j public class DriverController { @Autowired @@ -45,6 +48,9 @@ @Autowired private MongoTemplate mongoTemplate; + + @Resource + private RedisTemplate<String, Object> redisTemplate; @@ -66,6 +72,9 @@ }) public ResultUtil<BaseWarpper> queryIdleDriver(Integer type, Double lon, Double lat){ try { + if(type==2){ + type=null; + } List<Driver> list = driverService.queryIdleDriver(type, lon, lat, 5D, null); BaseWarpper baseWarpper = new BaseWarpper(); baseWarpper.setNumber(list.size()); @@ -140,26 +149,27 @@ }) public ResultUtil<List<Map<String, Object>>> queryNearbyCar(String lon, String lat){ //mongodb查询附近5公里范围内的坐标 - NearQuery query = NearQuery.near(new Point(Double.parseDouble(lon), Double.parseDouble(lat))); - query.query(Query.query(Criteria.where("point"))); - query.maxDistance(5000); - GeoResults<DriverPosition> results = mongoTemplate.geoNear(query, DriverPosition.class); + Query query = Query.query(Criteria.where("point").nearSphere(new Point(Double.parseDouble(lon), Double.parseDouble(lat))).maxDistance(5000)); + List<DriverPosition> driverPositions = mongoTemplate.find(query, DriverPosition.class); + log.info("地理空间检索出的数据:{}", JSON.toJSONString(driverPositions)); List<Map<String, Object>> collect = new ArrayList<>(); - if(collect.size() > 0){ - List<GeoResult<DriverPosition>> content1 = results.getContent(); - List<DriverPosition> positionList = content1.stream().map(GeoResult::getContent).collect(Collectors.toList()); - List<Integer> driverIds = positionList.stream().map(DriverPosition::getDriverId).collect(Collectors.toList()); - List<Driver> drivers = driverService.selectList(new EntityWrapper<Driver>().in("driverId", driverIds) + if(driverPositions.size() > 0){ + List<Integer> driverIds = driverPositions.stream().map(DriverPosition::getDriverId).collect(Collectors.toList()); + List<Driver> drivers = driverService.selectList(new EntityWrapper<Driver>().in("id", driverIds) .eq("authState", 2).eq("state", 2)); collect = drivers.stream().map(driver -> { - Optional<DriverPosition> driverPosition = positionList.stream().filter(position -> position.getDriverId().equals(driver.getId())).findFirst(); - Map<String, Object> map = new HashMap<>(); - map.put("driverId", driver.getId()); - map.put("name", driver.getName()); - map.put("lot", driverPosition.isPresent() ? driverPosition.get().getPoint().getX() : ""); - map.put("lat", driverPosition.isPresent() ? driverPosition.get().getPoint().getY() : ""); - return map; - }).collect(Collectors.toList()); + String value = (String) redisTemplate.opsForValue().get("dache:DRIVER" + String.valueOf(driver.getId())); + if(ToolUtil.isNotEmpty(value)){ + Optional<DriverPosition> driverPosition = driverPositions.stream().filter(position -> position.getDriverId().equals(driver.getId())).findFirst(); + Map<String, Object> map = new HashMap<>(); + map.put("driverId", driver.getId()); + map.put("name", driver.getName()); + map.put("lot", driverPosition.isPresent() ? driverPosition.get().getPoint().getX() : ""); + map.put("lat", driverPosition.isPresent() ? driverPosition.get().getPoint().getY() : ""); + return map; + } + return null; + }).filter(Objects::nonNull).collect(Collectors.toList()); } return ResultUtil.success(collect); } -- Gitblit v1.7.1