From af389156a78401b1e51807efe90b7310fc4c20f6 Mon Sep 17 00:00:00 2001
From: Fixiaobai <fixiaobai@163.com>
Date: 星期五, 10 十一月 2023 19:50:28 +0800
Subject: [PATCH] Changes11

---
 mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/PostServiceImpl.java |   66 +++++++++++++++++++++++++++++++++
 1 files changed, 66 insertions(+), 0 deletions(-)

diff --git a/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/PostServiceImpl.java b/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/PostServiceImpl.java
index ca0b849..c39ae5c 100644
--- a/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/PostServiceImpl.java
+++ b/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/PostServiceImpl.java
@@ -1,19 +1,29 @@
 package com.chinaztt.mes.basic.service.impl;
 
 import cn.hutool.core.collection.CollectionUtil;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.StringUtils;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
+import com.chinaztt.mes.basic.entity.Company;
+import com.chinaztt.mes.basic.entity.Division;
+import com.chinaztt.mes.basic.entity.Factory;
 import com.chinaztt.mes.basic.entity.Post;
+import com.chinaztt.mes.basic.excel.PostData;
+import com.chinaztt.mes.basic.mapper.CompanyMapper;
+import com.chinaztt.mes.basic.mapper.DivisionMapper;
+import com.chinaztt.mes.basic.mapper.FactoryMapper;
 import com.chinaztt.mes.basic.mapper.PostMapper;
 import com.chinaztt.mes.basic.service.PostService;
 import com.chinaztt.mes.common.numgen.NumberGenerator;
 import lombok.AllArgsConstructor;
+import org.springframework.beans.BeanUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
 import java.util.List;
+import java.util.Objects;
 
 /**
  * 宀椾綅琛�
@@ -26,6 +36,10 @@
 @Transactional(rollbackFor = Exception.class)
 public class PostServiceImpl extends ServiceImpl<PostMapper, Post> implements PostService {
 	private NumberGenerator<Post> postNumberGenerator;
+	private DivisionMapper divisionMapper;
+	private PostMapper postMapper;
+	private CompanyMapper companyMapper;
+	private FactoryMapper factoryMapper;
 
 	@Override
 	public boolean save(Post post) {
@@ -51,4 +65,56 @@
 			}
 		}
 	}
+
+	//excel瀵煎叆鏁版嵁
+	@Override
+	public void importWorkstationExcel(List<PostData> list) {
+		if (CollectionUtil.isEmpty(list)) {
+			return;
+		}
+		List<Company> companies = companyMapper.selectList(null);
+		List<Factory> factories = factoryMapper.selectList(null);
+		List<Division> divisions = divisionMapper.selectList(null);
+		for (PostData data : list) {
+			QueryWrapper<Post>QueryWrapper = new QueryWrapper<>();
+			QueryWrapper.lambda().eq(Post::getPostNo,data.getPostNo());
+			Post post1 = baseMapper.selectOne(QueryWrapper);
+			if(!Objects.isNull(post1)){
+				log.error("宀椾綅缂栧彿閲嶅=====銆�"+data.getPostNo());
+				continue;
+			}
+			Post post = new Post();
+			BeanUtils.copyProperties(data, post);
+			//TODO: data.getDivisionName()+"-"+   宀椾綅缁勫悎 閮ㄩ棬+宀椾綅
+			post.setPostName(data.getPostName());
+			for (int i = 0; i <companies.size() ; i++) {
+				if (companies.get(i).getCompanyName().equals(data.getCompanyName())){
+					post.setCompanyId(companies.get(i).getId());
+				}
+			}
+			if (post.getCompanyId()==null){
+				log.error("鍏徃:" + data.getCompanyName() + "涓嶅瓨鍦�");
+				continue ;
+			}
+			for (int i = 0; i <factories.size() ; i++) {
+				if (factories.get(i).getFactoryName().equals(data.getFactoryName())){
+					post.setFactoryId(factories.get(i).getId());
+				}
+			}
+			if (post.getFactoryId()==null){
+				log.error("宸ュ巶:" + data.getFactoryName() + "涓嶅瓨鍦�");
+				continue ;
+			}
+			for (int i = 0; i < divisions.size(); i++) {
+				if (divisions.get(i).getDivisionName().equals(data.getDivisionName())) {
+					post.setDivisionId(divisions.get(i).getId());
+				}
+			}
+			if (post.getDivisionId() == null) {
+				log.error("閮ㄩ棬:" + data.getDivisionName() + "涓嶅瓨鍦�");
+				continue;
+			}
+			baseMapper.insert(post);
+		}
+	}
 }

--
Gitblit v1.9.3