From 26fa723daeccb1e731ed24034ac66aa5c56c4269 Mon Sep 17 00:00:00 2001 From: liujie <liujie> Date: 星期五, 26 五月 2023 16:50:40 +0800 Subject: [PATCH] broker 卡车公司端 --- src/main/java/com/stylefeng/guns/modular/system/controller/TCompanyController.java | 16 +++++++++++++--- 1 files changed, 13 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/stylefeng/guns/modular/system/controller/TCompanyController.java b/src/main/java/com/stylefeng/guns/modular/system/controller/TCompanyController.java index 64e0709..8e0133d 100644 --- a/src/main/java/com/stylefeng/guns/modular/system/controller/TCompanyController.java +++ b/src/main/java/com/stylefeng/guns/modular/system/controller/TCompanyController.java @@ -5,9 +5,7 @@ import com.baomidou.mybatisplus.plugins.Page; import com.stylefeng.guns.core.base.controller.BaseController; import com.stylefeng.guns.modular.system.model.*; -import com.stylefeng.guns.modular.system.service.ITCompanyFeeSettingService; -import com.stylefeng.guns.modular.system.service.ITCompanyService; -import com.stylefeng.guns.modular.system.service.ITCompanyServiceService; +import com.stylefeng.guns.modular.system.service.*; import com.stylefeng.guns.modular.system.utils.tips.ErrorTip; import com.stylefeng.guns.modular.system.utils.tips.SuccessTip; import io.swagger.annotations.Api; @@ -44,6 +42,12 @@ @Autowired private ITCompanyServiceService itCompanyServiceService; + + @Autowired + private TPowerUnitsService powerUnitsService; + + @Autowired + private ITDriverService driverService; @@ -225,6 +229,12 @@ BeanUtil.copyProperties(company,tCompanyDto); List<TCompanyService> companyServiceList = itCompanyServiceService.selectList(new EntityWrapper<TCompanyService>().eq("company_id", tCompanyId).eq("type",0)); tCompanyDto.setList(companyServiceList); + int i = powerUnitsService.selectCount(new EntityWrapper<TPowerUnits>().eq("company_id", tCompanyId).eq("type", 1)); + tCompanyDto.setPowerUnits(i); + int e = powerUnitsService.selectCount(new EntityWrapper<TPowerUnits>().eq("company_id", tCompanyId).eq("type", 2)); + tCompanyDto.setChassises(e); + int d = driverService.selectCount(new EntityWrapper<TDriver>().eq("company_id", tCompanyId)); + tCompanyDto.setDrivers(d); return new SuccessTip(tCompanyDto); } -- Gitblit v1.7.1