组织导入数据

zss

2023-11-03 99d9153a8c33ce32b898c7d9098038caa3447206

 mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/CrewServiceImpl.java

@@ -3,24 +3,27 @@
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.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.basic.dto.CrewDTO;
import com.chinaztt.mes.basic.entity.Crew;
import com.chinaztt.mes.basic.entity.JoinCrewStaff;
import com.chinaztt.mes.basic.entity.Part;
import com.chinaztt.mes.basic.entity.Staff;
import com.chinaztt.mes.basic.entity.*;
import com.chinaztt.mes.basic.excel.CrewData;
import com.chinaztt.mes.basic.excel.PostData;
import com.chinaztt.mes.basic.mapper.CrewMapper;
import com.chinaztt.mes.basic.mapper.JoinCrewStaffMapper;
import com.chinaztt.mes.basic.mapper.StaffMapper;
import com.chinaztt.mes.basic.service.CrewService;
import lombok.AllArgsConstructor;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * 人员班组维护表
@@ -82,6 +85,37 @@
      return staffMapper.selectStaffById(crewDTO.getId());
   }

   //导入excel数据
   @Override
   public void importWorkstationExcel(List<CrewData> list) {
      if (CollectionUtil.isEmpty(list)) {
         return;
      }
      List<Staff> staff = staffMapper.selectList(null);
      Crew crew = new Crew();
      for (CrewData data : list) {
         //新增班组表
         if (ObjectUtils.isNotEmpty(data.getCrewName())) {
            crew.setCrewName(data.getCrewName());
            crew.setWorkShop(data.getWorkShop());
            baseMapper.insert(crew);
         }
         JoinCrewStaff joinCrewStaff = new JoinCrewStaff();
         joinCrewStaff.setCrewId(crew.getId());
         for (int i = 0; i < staff.size(); i++) {
            if (staff.get(i).getStaffNo().equals(data.getStaffNo())) {
               joinCrewStaff.setStaffId(staff.get(i).getId());
            }
         }
         if (joinCrewStaff.getStaffId() == null) {
            log.error("人员:" + data.getStaffNo() + "不存在");
            continue;
         }
         //新增人员班组维护表
         joinCrewStaffMapper.insert(joinCrewStaff);
      }
   }

   public void saveJoin(CrewDTO crewDTO) {
      if (CollectionUtil.isNotEmpty(crewDTO.getStaffList())) {
         List<Staff> staffList = joinCrewStaffMapper.selectStaffByCrewId(crewDTO.getId());