From 6ee9d58d62dbbccf95ce809f358ec9f8d088b705 Mon Sep 17 00:00:00 2001
From: Fixiaobai <fixiaobai@163.com>
Date: 星期五, 17 十一月 2023 09:16:16 +0800
Subject: [PATCH] Changes15 websocket
---
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