From 8fc672da0775664f3f48f1cbb19e09361100138c Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Thu, 11 Nov 2021 08:50:55 +0800
Subject: [PATCH] 派遣查询接口修改

---
 src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java |   13 ++++++++++++-
 1 files changed, 12 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 c16b1dd..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,8 +20,10 @@
 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;
@@ -35,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);
 	}
 
 	/**

--
Gitblit v1.9.3