From f435d50064d048c7bbc2fb46ffcd518cb19b9764 Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Mon, 23 Aug 2021 09:47:04 +0800
Subject: [PATCH] 文件上传接口新增

---
 src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java |   30 +++++++++++++++++++++++++++++-
 1 files changed, 29 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 be057b3..bf5f30a 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
@@ -18,8 +18,10 @@
 
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import lombok.AllArgsConstructor;
 import org.springblade.modules.dispatcher.entity.Dispatcher;
 import org.springblade.modules.dispatcher.entity.DispatcherUnit;
+import org.springblade.modules.dispatcher.excel.DispatcherUnitExcel;
 import org.springblade.modules.dispatcher.mapper.DispatcherMapper;
 import org.springblade.modules.dispatcher.mapper.DispatcherUnitMapper;
 import org.springblade.modules.dispatcher.service.IDispatcherService;
@@ -28,6 +30,8 @@
 import org.springblade.modules.dispatcher.vo.DispatcherVO;
 import org.springframework.stereotype.Service;
 
+import java.util.List;
+
 /**
  *  服务实现类
  *
@@ -35,11 +39,20 @@
  * @since 2021-07-07
  */
 @Service
+@AllArgsConstructor
 public class DispatcherUnitServiceImpl extends ServiceImpl<DispatcherUnitMapper, DispatcherUnit> implements IDispatcherUnitService {
+
+
+	private final 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.getDispatcherCount(dispatcherUnitVO1));
+		});
+		return page.setRecords(dispatcherUnitVOS);
 	}
 
 	/**
@@ -51,4 +64,19 @@
 	public DispatcherUnitVO selectDispatcherUnitInfo(DispatcherUnit dispatcherUnit) {
 		return baseMapper.selectDispatcherUnitInfo(dispatcherUnit);
 	}
+
+	/**
+	 * 自定义树
+	 * @param dispatcher
+	 * @return
+	 */
+	@Override
+	public List<DispatcherUnitVO> selectDispatcherUnitPageTree(DispatcherUnitVO dispatcher) {
+		return baseMapper.selectDispatcherUnitPageTree(dispatcher);
+	}
+
+	@Override
+	public void importDispatcherUnit(List<DispatcherUnitExcel> data, Boolean isCovered) {
+
+	}
 }

--
Gitblit v1.9.3