Changes11

Fixiaobai

2023-11-09 3059430a0db3f75426e435c06090358646fc3792

 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 com.chinaztt.ztt.common.core.util.R;
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,17 +63,31 @@

   @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){
         Division division = new Division();
      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);
          if(!Objects.isNull(division1)){
             log.error("部门编号重复========》"+data.getDivisionNo());
             continue;
          }
          Division division = new Division();
         for (int i = 0; i <companyList.size() ; i++) {
            if (companyList.get(i).getCompanyName().equals(data.getCompanyName())){
             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 ;