From b5960d1968e007b91d4d33dd7cbb74f1b566f2c1 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Fri, 24 May 2024 10:20:01 +0800
Subject: [PATCH] 到期时间修改

---
 src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java |   25 ++++++++++++++++++++++++-
 1 files changed, 24 insertions(+), 1 deletions(-)

diff --git a/src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java b/src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java
index adb3b09..7c604aa 100644
--- a/src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java
+++ b/src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java
@@ -20,9 +20,13 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.modules.dispatcher.entity.DispatcherUnit;
 import org.springblade.modules.dispatcher.mapper.DispatcherUnitMapper;
+import org.springblade.modules.dispatcher.service.IDispatcherService;
 import org.springblade.modules.dispatcher.service.IDispatcherUnitService;
 import org.springblade.modules.dispatcher.vo.DispatcherUnitVO;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.util.List;
 
 /**
  *  服务实现类
@@ -33,9 +37,18 @@
 @Service
 public class DispatcherUnitServiceImpl extends ServiceImpl<DispatcherUnitMapper, DispatcherUnit> implements IDispatcherUnitService {
 
+	@Autowired
+	private IDispatcherService dispatcherService;
+
+
 	@Override
 	public IPage<DispatcherUnitVO> selectDispatcherUnitPage(IPage<DispatcherUnitVO> page, DispatcherUnitVO dispatcherUnitVO) {
-		return page.setRecords(baseMapper.selectDispatcherUnitPage(page, dispatcherUnitVO));
+		List<DispatcherUnitVO> dispatcherUnitVOS = baseMapper.selectDispatcherUnitPage(page, dispatcherUnitVO);
+		dispatcherUnitVOS.forEach(dispatcherUnitVO1 -> {
+			//查询派遣到每家单位的保安员数量
+			dispatcherUnitVO1.setNum(dispatcherService.getDispatcherNum(dispatcherUnitVO1));
+		});
+		return page.setRecords(dispatcherUnitVOS);
 	}
 
 	/**
@@ -47,4 +60,14 @@
 	public DispatcherUnitVO selectDispatcherUnitInfo(DispatcherUnit dispatcherUnit) {
 		return baseMapper.selectDispatcherUnitInfo(dispatcherUnit);
 	}
+
+	/**
+	 * 自定义树
+	 * @param dispatcher
+	 * @return
+	 */
+	@Override
+	public List<DispatcherUnitVO> selectDispatcherUnitPageTree(DispatcherUnitVO dispatcher) {
+		return baseMapper.selectDispatcherUnitPageTree(dispatcher);
+	}
 }

--
Gitblit v1.9.3