开发模型3业务逻辑

Pu Zhibing

2025-02-28 173428adf09179860a3cb1a88f46378fd98473b2

 guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/PatrolTaskServiceImpl.java

@@ -8,6 +8,7 @@
import com.stylefeng.guns.modular.system.model.TaskDetailVehicles;
import com.stylefeng.guns.modular.system.model.TaskDetailVehiclesChannel;
import com.stylefeng.guns.modular.system.model.enums.ImageModelEnum;
import com.stylefeng.guns.modular.system.model.enums.VideoChannelEnum;
import com.stylefeng.guns.modular.system.model.vo.PatrolTaskVo;
import com.stylefeng.guns.modular.system.model.vo.VehicleVo;
import com.stylefeng.guns.modular.system.service.ITaskDetailService;
@@ -25,7 +26,9 @@
import org.quartz.SchedulerException;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Lazy;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import com.stylefeng.guns.modular.system.dao.PatrolTaskMapper;
import com.stylefeng.guns.modular.system.service.IPatrolTaskService;
@@ -62,6 +65,12 @@
   
   @Autowired
   private QuartzUtil quartzUtil;
	
   @Value("${maxThreadCount}")
   private Integer maxThreadCount;
	
   @Resource
   private RedisTemplate redisTemplate;
   
   
   
@@ -190,14 +199,34 @@
      
      //获取当前任务中的车船信息
      List<TaskDetailVehicles> taskDetailVehicles = taskDetailVehiclesService.selectList(new EntityWrapper<TaskDetailVehicles>().eq("task_detail_id", id));
      //根据企业id分组用于定义多线上处理
      Set<String> collect = taskDetailVehicles.stream().map(TaskDetailVehicles::getCompanyId).collect(Collectors.toSet());
      //根据车辆数量，分组线程数
      int num;
      int threadCount;
      if(taskDetailVehicles.size() <= maxThreadCount){
         threadCount = taskDetailVehicles.size();
         num = 1;
      }else{
         num = taskDetailVehicles.size() / maxThreadCount;
         if(taskDetailVehicles.size() % maxThreadCount > 0){
            num++;
         }
         threadCount = taskDetailVehicles.size() / num;
         if(taskDetailVehicles.size() % num > 0){
            threadCount++;
         }
      }
		
      //定义线程池
      ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(collect.size(), collect.size(), 0L, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<Runnable>());
      ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(threadCount, threadCount, 0L, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<Runnable>());
      //定义多线程阻塞
      CountDownLatch countDownLatch = new CountDownLatch(collect.size());
      for (String companyId : collect) {
         List<TaskDetailVehicles> vehicles = taskDetailVehicles.stream().filter(s -> s.getCompanyId().equals(companyId)).collect(Collectors.toList());
      CountDownLatch countDownLatch = new CountDownLatch(threadCount);
      for (int i = 0; i < threadCount; i++) {
         int start = i * num;
         int end = (i + 1) * num;
         if((i + 1) == threadCount){
            end = taskDetailVehicles.size();
         }
         List<TaskDetailVehicles> vehicles = taskDetailVehicles.subList(start, end);
         threadPoolExecutor.execute(new Runnable() {
            @Override
            public void run() {
@@ -236,65 +265,70 @@
                        taskDetailVehiclesService.updateById(taskDetailVehicle);
                        continue;
                     }
                     taskDetailVehicle.setVehicleGpsProtocol(vehicleOnline1.getVehicleGpsProtocol());
                     taskDetailVehicle.setTerminalId(vehicleOnline1.getTerminalId());
                     taskDetailVehicle.setStatus(2);
                     taskDetailVehiclesService.updateById(taskDetailVehicle);
                     //成功
                     List<TaskDetailVehiclesChannel> taskDetailVehiclesChannels = taskDetailVehiclesChannelService.selectList(new EntityWrapper<TaskDetailVehiclesChannel>().eq("task_detail_vehicles_id", taskDetailVehicleId));
                     //开始发送拍照指令，获取照片地址
                     String vehicleGpsProtocol = vehicleOnline1.getVehicleGpsProtocol();
                     String terminalId = vehicleOnline1.getTerminalId();
                     for (TaskDetailVehiclesChannel taskDetailVehiclesChannel : taskDetailVehiclesChannels) {
                        Integer videoChannel = taskDetailVehiclesChannel.getVideoChannel();
                        try {
                           if("1".equals(vehicleGpsProtocol)){
                              //2018 协议
                              String fileId = VideoGateway.cameraShot2018(terminalId, videoChannel, 0);
                              if(ToolUtil.isNotEmpty(fileId)){
                                 //等待30秒
                                 Thread.sleep(30000);
                                 String fileUrl = VideoGateway.getCameraShotByFileId2018(terminalId, fileId, 0);
                                 if(ToolUtil.isNotEmpty(fileUrl)){
                                    taskDetailVehiclesChannel.setImageUrl(fileUrl);
                                    taskDetailVehiclesChannel.setSysCreateTime(LocalDateTime.now());
                                    taskDetailVehiclesChannel.setArtificialStatus(1);
                                    //开始调模型接口得出结果
                                    Map<String, Object> map = ImageModelUtil.modelCheck(fileUrl, ImageModelEnum.getImageModelEnum(patrolTask.getImageModel()));
                                    Boolean b = (Boolean) map.get("b");
                                    String r = map.get("r").toString();
                                    taskDetailVehiclesChannel.setSysStatus(b ? 2 : 3);
                                    taskDetailVehiclesChannel.setResult(r);
                                    taskDetailVehiclesChannelService.updateById(taskDetailVehiclesChannel);
                     //非倾倒模型，建筑垃圾倾倒模型由网关接口触发判断执行结果
                     if(!patrolTask.getImageModel().equals(ImageModelEnum.CONSTRUCTION_WASTE_LOAD.getCode())){
                        //成功
                        List<TaskDetailVehiclesChannel> taskDetailVehiclesChannels = taskDetailVehiclesChannelService.selectList(new EntityWrapper<TaskDetailVehiclesChannel>().eq("task_detail_vehicles_id", taskDetailVehicleId));
                        //开始发送拍照指令，获取照片地址
                        String vehicleGpsProtocol = vehicleOnline1.getVehicleGpsProtocol();
                        String terminalId = vehicleOnline1.getTerminalId();
                        for (TaskDetailVehiclesChannel taskDetailVehiclesChannel : taskDetailVehiclesChannels) {
                           Integer videoChannel = taskDetailVehiclesChannel.getVideoChannel();
                           try {
                              if("1".equals(vehicleGpsProtocol)){
                                 //2018 协议
                                 String fileId = VideoGateway.cameraShot2018(terminalId, videoChannel, 0);
                                 if(ToolUtil.isNotEmpty(fileId)){
                                    //等待30秒
                                    Thread.sleep(30000);
                                    String fileUrl = VideoGateway.getCameraShotByFileId2018(terminalId, fileId, 0);
                                    if(ToolUtil.isNotEmpty(fileUrl)){
                                       taskDetailVehiclesChannel.setImageUrl(fileUrl);
                                       taskDetailVehiclesChannel.setSysCreateTime(LocalDateTime.now());
                                       taskDetailVehiclesChannel.setArtificialStatus(1);
                                       //开始调模型接口得出结果
                                       Map<String, Object> map = ImageModelUtil.modelCheck(fileUrl, ImageModelEnum.getImageModelEnum(patrolTask.getImageModel()));
                                       Boolean b = (Boolean) map.get("b");
                                       String r = map.get("r").toString();
                                       taskDetailVehiclesChannel.setSysStatus(b ? 2 : 3);
                                       taskDetailVehiclesChannel.setResult(r);
                                       taskDetailVehiclesChannelService.updateById(taskDetailVehiclesChannel);
                                    }
                                 }
                              }else{
                                 //2023 协议
                                 String fileId = VideoGateway.cameraShot2023(vehicleId, vehicleType, terminalId, videoChannel, 0);
                                 if(ToolUtil.isNotEmpty(fileId)){
                                    //等待30秒
                                    Thread.sleep(30000);
                                    String fileUrl = VideoGateway.getCameraShotByFileId2023(fileId, 0);
                                    if(ToolUtil.isNotEmpty(fileUrl)){
                                       taskDetailVehiclesChannel.setImageUrl(fileUrl);
                                       taskDetailVehiclesChannel.setSysCreateTime(LocalDateTime.now());
                                       taskDetailVehiclesChannel.setArtificialStatus(1);
                                       //开始调模型接口得出结果
                                       Map<String, Object> map = ImageModelUtil.modelCheck(fileUrl, ImageModelEnum.getImageModelEnum(patrolTask.getImageModel()));
                                       Boolean b = (Boolean) map.get("b");
                                       String r = map.get("r").toString();
                                       taskDetailVehiclesChannel.setSysStatus(b ? 2 : 3);
                                       taskDetailVehiclesChannel.setResult(r);
                                       taskDetailVehiclesChannelService.updateById(taskDetailVehiclesChannel);
                                    }
                                 }
                              }
                           }else{
                              //2023 协议
                              String fileId = VideoGateway.cameraShot2023(vehicleId, vehicleType, terminalId, videoChannel, 0);
                              if(ToolUtil.isNotEmpty(fileId)){
                                 //等待30秒
                                 Thread.sleep(30000);
                                 String fileUrl = VideoGateway.getCameraShotByFileId2023(fileId, 0);
                                 if(ToolUtil.isNotEmpty(fileUrl)){
                                    taskDetailVehiclesChannel.setImageUrl(fileUrl);
                                    taskDetailVehiclesChannel.setSysCreateTime(LocalDateTime.now());
                                    taskDetailVehiclesChannel.setArtificialStatus(1);
                                    //开始调模型接口得出结果
                                    Map<String, Object> map = ImageModelUtil.modelCheck(fileUrl, ImageModelEnum.getImageModelEnum(patrolTask.getImageModel()));
                                    Boolean b = (Boolean) map.get("b");
                                    String r = map.get("r").toString();
                                    taskDetailVehiclesChannel.setSysStatus(b ? 2 : 3);
                                    taskDetailVehiclesChannel.setResult(r);
                                    taskDetailVehiclesChannelService.updateById(taskDetailVehiclesChannel);
                                 }
                              }
                           }catch (Exception e){
                              taskDetailVehiclesChannel.setResult(e.getMessage());
                              taskDetailVehiclesChannelService.updateById(taskDetailVehiclesChannel);
                           }
                        }catch (Exception e){
                           taskDetailVehiclesChannel.setResult(e.getMessage());
                           taskDetailVehiclesChannelService.updateById(taskDetailVehiclesChannel);
                        }
                        taskDetailVehicle.setEndExecutionTime(LocalDateTime.now());
                        taskDetailVehicle.setStatus(3);
                        taskDetailVehiclesService.updateById(taskDetailVehicle);
                     }
                     taskDetailVehicle.setEndExecutionTime(LocalDateTime.now());
                     taskDetailVehicle.setStatus(3);
                     taskDetailVehiclesService.updateById(taskDetailVehicle);
                  }catch (Exception e){
                     e.printStackTrace();
                     //失败
@@ -319,12 +353,184 @@
   }
   
   
	
	
   public static void main(String[] args) {
      LocalDateTime now = LocalDateTime.now();
      now = now.plusDays(5);
      int value = now.getDayOfWeek().getValue();
      System.err.println(value);
   /**
    * 变更车辆状态
    * @param vehicleNum
    * @param vehicleStatus         状态（1=行驶中，2=已停止）
    * @param communicationNum
    * @param vehicleGpsProtocol
    * @param latitude
    * @param longitude
    */
   @Override
   public void vehicleSpeed(String vehicleNum, Integer vehicleStatus, Integer vehicleType, String communicationNum, String vehicleGpsProtocol, Double latitude, Double longitude) {
      Boolean lock = redisTemplate.hasKey("model:3:lock:" + vehicleNum + vehicleType);
      if(lock){
         return;
      }
      //添加redis 锁
      redisTemplate.opsForValue().set("model:3:lock:" + vehicleNum + vehicleType, System.currentTimeMillis(), 1, TimeUnit.MINUTES);
      List<TaskDetail> taskDetails = taskDetailService.selectList(new EntityWrapper<TaskDetail>().eq("status", 2).eq("del_flag", 0));
      if(taskDetails.size() > 0){
         Set<Integer> collect = taskDetails.stream().map(TaskDetail::getPatrolTaskId).collect(Collectors.toSet());
         List<PatrolTask> patrolTasks = this.selectList(new EntityWrapper<PatrolTask>().in("id", collect).eq("image_model", ImageModelEnum.CONSTRUCTION_WASTE_LOAD.getCode()));
         if(patrolTasks.size() > 0){
            List<Integer> collect1 = patrolTasks.stream().map(PatrolTask::getId).collect(Collectors.toList());
            List<TaskDetailVehicles> taskDetailVehicles = taskDetailVehiclesService.selectList(new EntityWrapper<TaskDetailVehicles>().eq("vehicleNum", vehicleNum)
                  .eq("vehicleType", vehicleType).eq("status", 2).in("patrol_task_id", collect1));
            if(taskDetailVehicles.size() > 0){
               TaskDetailVehicles taskDetailVehicles1 = taskDetailVehicles.get(0);
               String terminalId = taskDetailVehicles1.getTerminalId();
               Integer vehicleId = taskDetailVehicles1.getVehicleId();
               String vehicleIdUnique = taskDetailVehicles1.getVehicleIdUnique();
               //开始调用顶盖模型3次判断结果
               int loaded_cargo_num = 0;
               int empty_container_num = 0;
               String loaded_cargo_fileUrl = "";
               String empty_container_fileUrl = "";
               String r = null;
               for (int i = 0; i < 3; i++) {
                  //开始发送拍照指令，获取照片地址
                  try {
                     if("1".equals(vehicleGpsProtocol)){
                        //2018 协议
                        String fileId = VideoGateway.cameraShot2018(terminalId, VideoChannelEnum.CONTAINER.getId(), 0);
                        if(ToolUtil.isNotEmpty(fileId)){
                           //等待30秒
                           Thread.sleep(30000);
                           String fileUrl = VideoGateway.getCameraShotByFileId2018(terminalId, fileId, 0);
                           if(ToolUtil.isNotEmpty(fileUrl)){
                              //开始调模型接口得出结果
                              Map<String, Object> map = ImageModelUtil.modelCheck(fileUrl, ImageModelEnum.TOP_SEAL);
                              r = map.get("r").toString();
                              List<String> result = JSON.parseArray(r, String.class);
                              //装载状态
                              Optional<String> loaded_cargo = result.stream().filter(s -> s.contains("Loaded Cargo")).findFirst();
                              if(loaded_cargo.isPresent()){
                                 loaded_cargo_num++;
                                 loaded_cargo_fileUrl = fileUrl;
                              }
                              //非装载状态
                              Optional<String> empty_container = result.stream().filter(s -> s.contains("Empty Container")).findFirst();
                              if(empty_container.isPresent()){
                                 empty_container_num++;
                                 empty_container_fileUrl = fileUrl;
                              }
                           }
                        }
                     }else{
                        //2023 协议
                        String fileId = VideoGateway.cameraShot2023(vehicleId, vehicleType, terminalId, VideoChannelEnum.CONTAINER.getId(), 0);
                        if(ToolUtil.isNotEmpty(fileId)){
                           //等待30秒
                           Thread.sleep(30000);
                           String fileUrl = VideoGateway.getCameraShotByFileId2023(fileId, 0);
                           if(ToolUtil.isNotEmpty(fileUrl)){
                              //开始调模型接口得出结果
                              Map<String, Object> map = ImageModelUtil.modelCheck(fileUrl, ImageModelEnum.TOP_SEAL);
                              r = map.get("r").toString();
                              List<String> result = JSON.parseArray(r, String.class);
                              //装载状态
                              Optional<String> loaded_cargo = result.stream().filter(s -> s.contains("Loaded Cargo")).findFirst();
                              if(loaded_cargo.isPresent()){
                                 loaded_cargo_num++;
                                 loaded_cargo_fileUrl = fileUrl;
                              }
                              //非装载状态
                              Optional<String> empty_container = result.stream().filter(s -> s.contains("Empty Container")).findFirst();
                              if(empty_container.isPresent()){
                                 empty_container_num++;
                                 empty_container_fileUrl = fileUrl;
                              }
                           }
                        }
                     }
                  }catch (Exception e){
                     e.printStackTrace();
                     //执行失败
                     for (TaskDetailVehicles taskDetailVehicle : taskDetailVehicles) {
                        taskDetailVehicle.setStatus(5);
                        taskDetailVehicle.setEndExecutionTime(LocalDateTime.now());
                     }
                     taskDetailVehiclesService.updateBatchById(taskDetailVehicles);
                     List<Integer> collect2 = taskDetailVehicles.stream().map(TaskDetailVehicles::getId).collect(Collectors.toList());
                     //修改通道结果
                     List<TaskDetailVehiclesChannel> list = taskDetailVehiclesChannelService.selectList(new EntityWrapper<TaskDetailVehiclesChannel>().eq("task_detail_vehicles_id", collect2));
                     for (TaskDetailVehiclesChannel taskDetailVehiclesChannel : list) {
                        taskDetailVehiclesChannel.setResult(e.getMessage());
                     }
                     taskDetailVehiclesChannelService.updateBatchById(list);
							
                     //开始修改任务状态
                     Set<Integer> collect3 = taskDetailVehicles.stream().map(TaskDetailVehicles::getTaskDetailId).collect(Collectors.toSet());
                     for (Integer integer : collect3) {
                        int count = taskDetailVehiclesService.selectCount(new EntityWrapper<TaskDetailVehicles>().eq("task_detail_id", integer).in("status", Arrays.asList(1, 2)));
                        if(0 == count){
                           TaskDetail taskDetail = new TaskDetail();
                           taskDetail.setId(integer);
                           taskDetail.setStatus(4);
                           taskDetailService.updateById(taskDetail);
                        }
                     }
							
                     //删除redis锁key
                     redisTemplate.delete("model:3:lock:" + vehicleNum + vehicleType);
                     return;
                  }
               }
					
               //装载状态
               if(loaded_cargo_num > empty_container_num && 2 == vehicleStatus){
                  //装载状态下，判断车辆如果车速为0，则需要判断是否在围栏内，如果在则添加标识
                  //判断车辆是否在围栏内
                  boolean fenceByPoint = VideoGateway.getFenceByPoint(longitude.toString(), latitude.toString(), 0);
                  if(fenceByPoint){
                     redisTemplate.opsForValue().set("model:3:" + vehicleIdUnique, System.currentTimeMillis());
                  }
						
               }
					
               //空载状态
               if(empty_container_num > loaded_cargo_num){
                  //车辆空载状态，如果车辆车速大于0，且车辆有在围栏内的标识，则判定为正常倾倒
                  Boolean aBoolean = redisTemplate.hasKey("model:3:" + vehicleIdUnique);
                  if(1 == vehicleStatus){
                     for (TaskDetailVehicles taskDetailVehicle : taskDetailVehicles) {
                        taskDetailVehicle.setStatus(3);
                        taskDetailVehicle.setEndExecutionTime(LocalDateTime.now());
                     }
                     taskDetailVehiclesService.updateBatchById(taskDetailVehicles);
                     List<Integer> collect2 = taskDetailVehicles.stream().map(TaskDetailVehicles::getId).collect(Collectors.toList());
                     //修改通道结果
                     List<TaskDetailVehiclesChannel> list = taskDetailVehiclesChannelService.selectList(new EntityWrapper<TaskDetailVehiclesChannel>().eq("task_detail_vehicles_id", collect2));
                     for (TaskDetailVehiclesChannel taskDetailVehiclesChannel : list) {
                        taskDetailVehiclesChannel.setImageUrl(empty_container_fileUrl);
                        taskDetailVehiclesChannel.setSysCreateTime(LocalDateTime.now());
                        taskDetailVehiclesChannel.setArtificialStatus(1);
                        taskDetailVehiclesChannel.setSysStatus(aBoolean ? 2 : 3);
                        taskDetailVehiclesChannel.setResult(r);
                     }
                     taskDetailVehiclesChannelService.updateBatchById(list);
							
                     //开始修改任务状态
                     Set<Integer> collect3 = taskDetailVehicles.stream().map(TaskDetailVehicles::getTaskDetailId).collect(Collectors.toSet());
                     for (Integer integer : collect3) {
                        int count = taskDetailVehiclesService.selectCount(new EntityWrapper<TaskDetailVehicles>().eq("task_detail_id", integer).in("status", Arrays.asList(1, 2)));
                        if(0 == count){
                           TaskDetail taskDetail = new TaskDetail();
                           taskDetail.setId(integer);
                           taskDetail.setStatus(3);
                           taskDetailService.updateById(taskDetail);
                        }
                     }
                     //删除标识
                     redisTemplate.delete("model:3:" + vehicleIdUnique);
                  }
               }
            }
         }
      }
      //删除redis锁key
      redisTemplate.delete("model:3:lock:" + vehicleNum + vehicleType);
   }
}