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