From 43f2fe7fa6ec90b957d3182730e5c726b7158ced Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期一, 23 九月 2024 14:49:49 +0800
Subject: [PATCH] #
---
zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/BusController.java | 7 ++++++-
1 files changed, 6 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..c7e937f 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,11 @@
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()));
+ record.setTasksNum(tasks.size());
+ }
}
return R.ok().add(page);
}
--
Gitblit v1.9.1