From 591fb57c675a1849e6e0ab1d968e1d58f6826f01 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期一, 23 九月 2024 14:10:40 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/BusController.java |    6 +++++-
 1 files changed, 5 insertions(+), 1 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/BusController.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/BusController.java
index 84e9973..a714c47 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/BusController.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/BusController.java
@@ -23,6 +23,7 @@
 
 import javax.servlet.http.HttpServletResponse;
 import java.util.*;
+import java.util.stream.Collectors;
 
 @RestController
 @RequestMapping("/api")
@@ -46,7 +47,10 @@
         PageParam<Bus, BaseParam> pageParam = new PageParam<>(baseParam, Bus.class);
         PageParam<Bus, BaseParam> page = busService.page(pageParam, pageParam.buildWrapper(true));
         for (Bus record : page.getRecords()) {
-            record.setTasksNum(taskService.count(new LambdaQueryWrapper<Task>().eq(Task::getBusId, record.getId())));
+            List<Task> tasks = taskService.list(new LambdaQueryWrapper<Task>().eq(Task::getBusId, record.getId()));
+            if (!Cools.isEmpty(tasks)) {
+                record.setTaskIds(tasks.stream().map(Task::getId).collect(Collectors.toList()));
+            }
         }
         return R.ok().add(page);
     }

--
Gitblit v1.9.1