From bcfbba254d346d1d2a97703069bdaa3a4fb3a6df Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期五, 13 六月 2025 10:19:56 +0800
Subject: [PATCH] 保洁巡检本周代码

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLocationTypeServiceImpl.java |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLocationTypeServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLocationTypeServiceImpl.java
index 894b540..0dfd83e 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLocationTypeServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLocationTypeServiceImpl.java
@@ -1,13 +1,18 @@
 package com.ruoyi.system.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.common.basic.PageInfo;
+import com.ruoyi.system.mapper.TLeaveMapper;
+import com.ruoyi.system.mapper.TLocationMapper;
 import com.ruoyi.system.mapper.TLocationTypeMapper;
+import com.ruoyi.system.model.TLocation;
 import com.ruoyi.system.model.TLocationType;
 import com.ruoyi.system.query.LocationTypeListQuery;
 import com.ruoyi.system.service.TLocationTypeService;
 import com.ruoyi.system.vo.system.LocationTypeListVO;
 import com.ruoyi.system.vo.system.ProjectDeptListVO;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
@@ -23,10 +28,16 @@
 @Service
 public class TLocationTypeServiceImpl extends ServiceImpl<TLocationTypeMapper, TLocationType> implements TLocationTypeService {
 
+    @Autowired
+    private TLocationMapper locationMapper;
     @Override
     public PageInfo<LocationTypeListVO> pageList(LocationTypeListQuery query) {
         PageInfo<LocationTypeListVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
         List<LocationTypeListVO> list = this.baseMapper.pageList(query,pageInfo);
+        for (LocationTypeListVO locationTypeListVO : list) {
+            locationTypeListVO.setLocationCount(locationMapper.selectCount(new LambdaQueryWrapper<TLocation>()
+                    .eq(TLocation::getLocationType,locationTypeListVO.getId()))+"");
+        }
         pageInfo.setRecords(list);
         return pageInfo;
     }

--
Gitblit v1.7.1