From 22c67fd681856d2b6513d02329b63d76017e1120 Mon Sep 17 00:00:00 2001
From: zss <zss@example.com>
Date: 星期五, 03 十一月 2023 18:05:05 +0800
Subject: [PATCH] 库位导入数据
---
mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/PostServiceImpl.java | 56 +++++++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 55 insertions(+), 1 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..74adc64 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
@@ -5,11 +5,17 @@
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.Post;
+import com.chinaztt.mes.basic.entity.*;
+import com.chinaztt.mes.basic.excel.PostData;
+import com.chinaztt.mes.basic.excel.StaffData;
+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;
@@ -26,6 +32,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 +61,48 @@
}
}
}
+
+ //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) {
+ Post post = new Post();
+ BeanUtils.copyProperties(data, post);
+ post.setPostName(data.getWorkShops()+"-"+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