From 2b0dc519ddc4fefcabc7f60dcda4968964155e11 Mon Sep 17 00:00:00 2001 From: zouyu <2723363702@qq.com> Date: 星期五, 17 十一月 2023 16:45:43 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/StaffServiceImpl.java | 104 ++++++++++++++++++++++++++++++++++++++++++++-------- 1 files changed, 88 insertions(+), 16 deletions(-) diff --git a/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/StaffServiceImpl.java b/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/StaffServiceImpl.java index fc11d47..e27a26a 100644 --- a/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/StaffServiceImpl.java +++ b/mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/StaffServiceImpl.java @@ -16,18 +16,23 @@ */ 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.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; -import com.chinaztt.mes.basic.entity.Staff; -import com.chinaztt.mes.basic.mapper.StaffMapper; +import com.chinaztt.mes.basic.entity.*; +import com.chinaztt.mes.basic.excel.StaffData; +import com.chinaztt.mes.basic.mapper.*; import com.chinaztt.mes.basic.service.StaffService; -import com.chinaztt.ztt.common.data.datascope.DataScope; import lombok.AllArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.BeanUtils; import org.springframework.stereotype.Service; import java.util.List; +import java.util.Map; +import java.util.Objects; /** * 鍛樺伐 @@ -37,23 +42,27 @@ */ @Service @AllArgsConstructor +@Slf4j public class StaffServiceImpl extends ServiceImpl<StaffMapper, Staff> implements StaffService { + private DivisionMapper divisionMapper; + private PostMapper postMapper; + private CompanyMapper companyMapper; + private FactoryMapper factoryMapper; - - /** - * Description: 鍒嗛〉 - * - * @param page - * @param ew - * @return IPage<List<Staff>> - * @author: luh - * @date: 2020/8/18 11:06 - */ + /** + * Description: 鍒嗛〉 + * + * @param page + * @param ew + * @return IPage<List < Staff>> + * @author: luh + * @date: 2020/8/18 11:06 + */ @Override public IPage<List<Staff>> getStaffPage(Page page, QueryWrapper<Staff> ew) { - return baseMapper.getStaffPage(page,ew); + return baseMapper.getStaffPage(page, ew); } /** @@ -61,14 +70,77 @@ * * @param page * @param staff - * @return IPage<List<Staff> + * @return IPage<List < Staff> * @author: fenglang * @date: 2020/8/28 11:06 */ @Override public IPage<List<Staff>> toResource(Page page, Staff staff) { - return baseMapper.toResource(page,staff); + return baseMapper.toResource(page, staff); } + //瀵煎叆excel鏁版嵁 + @Override + public void importWorkstationExcel(List<StaffData> list) { + if (CollectionUtil.isEmpty(list)) { + return; + } + List<Division> divisions = divisionMapper.selectList(null); + List<Post> posts = postMapper.selectList(null); + List<Company> companies = companyMapper.selectList(null); + List<Factory> factories = factoryMapper.selectList(null); + for (StaffData staffData : list) { + QueryWrapper<Staff> QueryWrapper = new QueryWrapper<>(); + QueryWrapper.lambda().eq(Staff::getStaffNo,staffData.getStaffNo()); + Staff staff1 = baseMapper.selectOne(QueryWrapper); + if(!Objects.isNull(staff1)){ + log.error("鍛樺伐缂栧彿閲嶅======銆�"+staffData.getStaffNo()); + continue; + } + Staff staff = new Staff(); + BeanUtils.copyProperties(staffData, staff); + for (int i = 0; i < divisions.size(); i++) { + if (divisions.get(i).getDivisionName().equals(staffData.getDivisionName())) { + staff.setDivisionId(divisions.get(i).getId()); + } + } + if (staff.getDivisionId() == null) { + log.error("閮ㄩ棬:" + staffData.getDivisionName() + "涓嶅瓨鍦�"); + continue; + } + for (int i = 0; i < posts.size(); i++) { + if (posts.get(i).getPostName().equals(staffData.getPost())) { + staff.setPostId(posts.get(i).getId()); + } + } + if (staff.getPostId() == null) { + log.error("宀椾綅:" + staffData.getPost() + "涓嶅瓨鍦�"); + continue; + } + for (int i = 0; i <companies.size() ; i++) { + if (companies.get(i).getCompanyName().equals(staffData.getCompanyName())){ + staff.setCompanyId(companies.get(i).getId()); + } + } + if (staff.getCompanyId()==null){ + log.error("鍏徃:" + staffData.getCompanyName() + "涓嶅瓨鍦�"); + continue ; + } + for (int i = 0; i <factories.size() ; i++) { + if (factories.get(i).getFactoryName().equals(staffData.getFactoryName())){ + staff.setFactoryId(factories.get(i).getId()); + } + } + if (staff.getFactoryId()==null){ + log.error("宸ュ巶:" + staffData.getFactoryName() + "涓嶅瓨鍦�"); + continue ; + } + baseMapper.insert(staff); + } + } + @Override + public List<Map<String, Object>> chooseStaff() { + return baseMapper.chooseStaff(); + } } -- Gitblit v1.9.3