Changes

Fixiaobai

2023-11-04 958e45e28ca4f591ab13f2c6f4fe27d57c4880d5

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

@@ -3,14 +3,15 @@
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.excel.CrewData;
import com.chinaztt.mes.basic.mapper.CrewMapper;
import com.chinaztt.mes.basic.mapper.JoinCrewStaffMapper;
import com.chinaztt.mes.basic.mapper.StaffMapper;
@@ -19,8 +20,8 @@
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

/**
 * 人员班组维护表
@@ -82,6 +83,44 @@
      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) {
         QueryWrapper<Crew>QueryWrapper = new QueryWrapper<>();
         QueryWrapper.lambda().eq(Crew::getCrewName,data.getCrewName());
         Crew crew1 = baseMapper.selectOne(QueryWrapper);
         if(!Objects.isNull(crew1)){
            log.error("班组名重复====》"+crew1.getCrewName());
            continue;
         }
         //新增班组表
         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());