From bafe10232e18d07f731a74c8df4e63b4cd06b70b Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Fri, 17 Sep 2021 15:10:20 +0800
Subject: [PATCH] 1.用户新增审查时间 2.派遣更改修改

---
 src/main/java/org/springblade/modules/dispatcher/service/impl/DispatcherUnitServiceImpl.java |  110 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 109 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..6193435 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,15 +18,26 @@
 
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import lombok.AllArgsConstructor;
+import org.springblade.modules.FTP.FtpUtil;
 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;
 import org.springblade.modules.dispatcher.service.IDispatcherUnitService;
 import org.springblade.modules.dispatcher.vo.DispatcherUnitVO;
 import org.springblade.modules.dispatcher.vo.DispatcherVO;
+import org.springblade.modules.system.service.IUserDeptService;
 import org.springframework.stereotype.Service;
+
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
 /**
  *  服务实现类
@@ -35,11 +46,22 @@
  * @since 2021-07-07
  */
 @Service
+@AllArgsConstructor
 public class DispatcherUnitServiceImpl extends ServiceImpl<DispatcherUnitMapper, DispatcherUnit> implements IDispatcherUnitService {
+
+
+	private final IDispatcherService dispatcherService;
+
+	private final IUserDeptService userDeptService;
 
 	@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 +73,90 @@
 	public DispatcherUnitVO selectDispatcherUnitInfo(DispatcherUnit dispatcherUnit) {
 		return baseMapper.selectDispatcherUnitInfo(dispatcherUnit);
 	}
+
+	/**
+	 * 自定义树
+	 * @param dispatcher
+	 * @return
+	 */
+	@Override
+	public List<DispatcherUnitVO> selectDispatcherUnitPageTree(DispatcherUnitVO dispatcher) {
+		return baseMapper.selectDispatcherUnitPageTree(dispatcher);
+	}
+
+	/**
+	 * 派遣导入
+	 * @param data
+	 * @param isCovered
+	 */
+	@Override
+	public void importDispatcherUnit(List<DispatcherUnitExcel> data, Boolean isCovered) {
+		//派遣数据导入
+		data.forEach(dispatcherUnitExcel -> {
+			//派遣单位
+			DispatcherUnit dispatcherUnit = new DispatcherUnit();
+			//数据获取
+			if (dispatcherUnitExcel.getDeptName()!=null && dispatcherUnitExcel.getDeptName()!=""){
+				String deptId = userDeptService.selectIn(dispatcherUnitExcel.getDeptName());
+				dispatcherUnit.setDeptId(Long.parseLong(deptId));
+			}
+
+			if (dispatcherUnitExcel.getName()!=null && dispatcherUnitExcel.getName()!=""){
+				//仅中文
+				String chinese = "^[\\u4E00-\\u9FA5\\uF900-\\uFA2D]+$";
+				Pattern compile = Pattern.compile(chinese);
+				String name = compile.matcher(dispatcherUnitExcel.getName()).replaceAll("");
+				dispatcherUnit.setName(name);
+			}
+
+			if (dispatcherUnitExcel.getLinkman()!=null && dispatcherUnitExcel.getLinkman()!=""){
+				dispatcherUnit.setLinkman(dispatcherUnitExcel.getLinkman());
+			}
+
+			if (dispatcherUnitExcel.getPhone()!=null && dispatcherUnitExcel.getPhone()!=""){
+				dispatcherUnit.setPhone(dispatcherUnitExcel.getPhone());
+			}
+
+			if (dispatcherUnitExcel.getStartTime()!=null){
+				String year = dispatcherUnitExcel.getStartTime().substring(0, 4);
+				String month = dispatcherUnitExcel.getStartTime().substring(4, 6);
+				String day = dispatcherUnitExcel.getStartTime().substring(6, 8);
+				String dateFormat = year + "-" +month+"-" +day;
+				try {
+					Date date = new SimpleDateFormat("yyyy-MM-dd").parse(dateFormat);
+					dispatcherUnit.setStartTime(date);
+				} catch (ParseException e) {
+					e.printStackTrace();
+				}
+			}
+
+			if (dispatcherUnitExcel.getEndTime()!=null){
+				String year = dispatcherUnitExcel.getEndTime().substring(0, 4);
+				String month = dispatcherUnitExcel.getEndTime().substring(4, 6);
+				String day = dispatcherUnitExcel.getEndTime().substring(6, 8);
+				String dateFormat = year + "-" +month+"-" +day;
+				try {
+					Date date = new SimpleDateFormat("yyyy-MM-dd").parse(dateFormat);
+					dispatcherUnit.setEndTime(date);
+				} catch (ParseException e) {
+					e.printStackTrace();
+				}
+			}
+
+			//新增
+			this.save(dispatcherUnit);
+
+			//数据同步
+			String s1 =
+				"insert into sys_dispatcher_unit(id,name,dept_id,linkman,phone,start_time,end_time) " +
+					"values(" + "'" + dispatcherUnit.getId() + "'" + "," +
+					"'" + dispatcherUnit.getName() + "'" + "," +
+					"'" + dispatcherUnit.getDeptId() + "'" + "," +
+					"'" + dispatcherUnit.getLinkman() + "'" + "," +
+					"'" + dispatcherUnit.getPhone() + "'" + "," +
+					"'" + new SimpleDateFormat("yyyy-MM-dd").format(dispatcherUnit.getStartTime()) + "'" + "," +
+					"'" + new SimpleDateFormat("yyyy-MM-dd").format(dispatcherUnit.getEndTime()) + "'" + ")";
+			FtpUtil.sqlFileUpload(s1);
+		});
+	}
 }

--
Gitblit v1.9.3