From 4fe1a741d807b55f7fc14aed85e232b9d7972425 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期一, 23 九月 2024 13:28:44 +0800 Subject: [PATCH] # --- zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/BusController.java | 10 +++++++++- 1 files changed, 9 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 50856dd..84e9973 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 @@ -11,9 +11,11 @@ import com.zy.acs.manager.manager.controller.param.BusCreateParam; 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.service.BusService; import com.zy.acs.manager.manager.service.LocService; import com.zy.acs.manager.manager.service.StaService; +import com.zy.acs.manager.manager.service.TaskService; import com.zy.acs.manager.system.controller.BaseController; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.security.access.prepost.PreAuthorize; @@ -34,13 +36,19 @@ private StaService staService; @Autowired private LocService locService; + @Autowired + private TaskService taskService; @PreAuthorize("hasAuthority('manager:bus:list')") @PostMapping("/bus/page") public R page(@RequestBody Map<String, Object> map) { BaseParam baseParam = buildParam(map, BaseParam.class); PageParam<Bus, BaseParam> pageParam = new PageParam<>(baseParam, Bus.class); - return R.ok().add(busService.page(pageParam, pageParam.buildWrapper(true))); + 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()))); + } + return R.ok().add(page); } @PreAuthorize("hasAuthority('manager:bus:list')") -- Gitblit v1.9.1