From 1060750d8d49c9d555f15b260ec22de7bd6524a9 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期五, 06 十二月 2024 15:37:15 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/BusServiceImpl.java |   38 +++++++++++++++++++++++++++++++++-----
 1 files changed, 33 insertions(+), 5 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/BusServiceImpl.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/BusServiceImpl.java
index c667477..21e8ad2 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/BusServiceImpl.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/BusServiceImpl.java
@@ -3,22 +3,22 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.zy.acs.framework.common.BaseRes;
+import com.zy.acs.framework.common.Cools;
+import com.zy.acs.framework.exception.CoolException;
 import com.zy.acs.manager.common.domain.TaskDto;
 import com.zy.acs.manager.manager.controller.param.OpenBusSubmitParam;
 import com.zy.acs.manager.manager.entity.Bus;
+import com.zy.acs.manager.manager.entity.Task;
 import com.zy.acs.manager.manager.enums.BusStsType;
+import com.zy.acs.manager.manager.enums.TaskStsType;
 import com.zy.acs.manager.manager.mapper.BusMapper;
 import com.zy.acs.manager.manager.service.BusService;
-import com.zy.acs.framework.common.Cools;
 import com.zy.acs.manager.manager.service.TaskService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
-import java.util.ArrayList;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
 
 @Slf4j
 @Service("busService")
@@ -120,4 +120,32 @@
         return this.list(wrapper);
     }
 
+    @Override
+    public void checkoutComplete(Long busId) {
+        if (null == busId) {
+            return;
+        }
+        int taskTotalCount = taskService.count(new LambdaQueryWrapper<Task>().eq(Task::getBusId, busId));
+        List<Task> actualDoneTasks = taskService.list(new LambdaQueryWrapper<Task>()
+                .eq(Task::getBusId, busId)
+                .in(Task::getTaskSts
+                        , TaskStsType.COMPLETE.val()
+                        , TaskStsType.CANCEL.val()
+                )
+        );
+        if (taskTotalCount == actualDoneTasks.size()) {
+            Bus bus = this.getById(busId);
+            long cancelTasksCount = actualDoneTasks.stream().filter(task -> TaskStsType.CANCEL.val() == task.getTaskSts()).count();
+            if (cancelTasksCount == taskTotalCount) {
+                bus.setBusSts(BusStsType.CANCEL.val());
+            } else {
+                bus.setBusSts(BusStsType.FINISH.val());
+            }
+            bus.setUpdateTime(new Date());
+            if (!this.updateById(bus)) {
+                throw new CoolException("Bus[{" + busId + "}] failed to update");
+            }
+        }
+    }
+
 }

--
Gitblit v1.9.1