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