From 3059430a0db3f75426e435c06090358646fc3792 Mon Sep 17 00:00:00 2001
From: Fixiaobai <fixiaobai@163.com>
Date: 星期四, 09 十一月 2023 09:11:17 +0800
Subject: [PATCH] Changes11

---
 mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/DivisionServiceImpl.java |   14 ++++++++++++--
 1 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/DivisionServiceImpl.java b/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/DivisionServiceImpl.java
index ba3b912..be123fa 100644
--- a/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/DivisionServiceImpl.java
+++ b/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/DivisionServiceImpl.java
@@ -23,15 +23,18 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 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.excel.DivisionData;
 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.service.DivisionService;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
 import java.util.Objects;
+import java.util.stream.Collectors;
 
 /**
  * 閮ㄩ棬
@@ -43,7 +46,7 @@
 @AllArgsConstructor
 public class DivisionServiceImpl extends ServiceImpl<DivisionMapper, Division> implements DivisionService {
        private CompanyMapper companyMapper;
-
+		private FactoryMapper factoryMapper;
 	/**
      * Description: 鍒嗛〉
      *
@@ -60,11 +63,13 @@
 
 	@Override
 	public void importWorkstationExcel(List<DivisionData> list) {
+		list.forEach(System.out::println);
 		if (CollectionUtil.isEmpty(list)) {
 			return;
 		}
 		List<Company> companyList = companyMapper.selectList(null);
-		 for (DivisionData data :list){
+		List<Factory> factories = factoryMapper.selectList(null);
+		for (DivisionData data :list){
 			 QueryWrapper<Division>QueryWrapper = new QueryWrapper<>();
 			 QueryWrapper.lambda().eq(Division::getDivisionNo,data.getDivisionNo());
 			 Division division1 = baseMapper.selectOne(QueryWrapper);
@@ -78,6 +83,11 @@
 				 division.setCompanyId(companyList.get(i).getId());
 				}
 			}
+			List<Factory> collect = factories.stream().filter(f -> f.getFactoryName().equals(data.getFactoryName())).collect(Collectors.toList());
+			if(collect.size() > 0){
+				division.setFactoryId(collect.get(0).getId());
+				division.setFactoryName(collect.get(0).getFactoryName());
+			}
 			if (division.getCompanyId()==null){
 				log.error("鍏徃:" + data.getCompanyName() + "涓嶅瓨鍦�");
 				continue ;

--
Gitblit v1.9.3