liujie
2023-05-26 26fa723daeccb1e731ed24034ac66aa5c56c4269
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);
    }