组织导入数据

zss

2023-11-03 99d9153a8c33ce32b898c7d9098038caa3447206

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

@@ -5,11 +5,17 @@
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
import com.chinaztt.mes.basic.entity.Post;
import com.chinaztt.mes.basic.entity.*;
import com.chinaztt.mes.basic.excel.PostData;
import com.chinaztt.mes.basic.excel.StaffData;
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.mapper.PostMapper;
import com.chinaztt.mes.basic.service.PostService;
import com.chinaztt.mes.common.numgen.NumberGenerator;
import lombok.AllArgsConstructor;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

@@ -26,6 +32,10 @@
@Transactional(rollbackFor = Exception.class)
public class PostServiceImpl extends ServiceImpl<PostMapper, Post> implements PostService {
   private NumberGenerator<Post> postNumberGenerator;
   private DivisionMapper divisionMapper;
   private PostMapper postMapper;
   private CompanyMapper companyMapper;
   private FactoryMapper factoryMapper;

   @Override
   public boolean save(Post post) {
@@ -51,4 +61,48 @@
         }
      }
   }

   //excel导入数据
   @Override
   public void importWorkstationExcel(List<PostData> list) {
      if (CollectionUtil.isEmpty(list)) {
         return;
      }
      List<Company> companies = companyMapper.selectList(null);
      List<Factory> factories = factoryMapper.selectList(null);
      List<Division> divisions = divisionMapper.selectList(null);
      for (PostData data : list) {
         Post post = new Post();
         BeanUtils.copyProperties(data, post);
         post.setPostName(data.getWorkShops()+"-"+data.getPostName());
         for (int i = 0; i <companies.size() ; i++) {
            if (companies.get(i).getCompanyName().equals(data.getCompanyName())){
               post.setCompanyId(companies.get(i).getId());
            }
         }
         if (post.getCompanyId()==null){
            log.error("公司:" + data.getCompanyName() + "不存在");
            continue ;
         }
         for (int i = 0; i <factories.size() ; i++) {
            if (factories.get(i).getFactoryName().equals(data.getFactoryName())){
               post.setFactoryId(factories.get(i).getId());
            }
         }
         if (post.getFactoryId()==null){
            log.error("工厂:" + data.getFactoryName() + "不存在");
            continue ;
         }
         for (int i = 0; i < divisions.size(); i++) {
            if (divisions.get(i).getDivisionName().equals(data.getDivisionName())) {
               post.setDivisionId(divisions.get(i).getId());
            }
         }
         if (post.getDivisionId() == null) {
            log.error("部门:" + data.getDivisionName() + "不存在");
            continue;
         }
         baseMapper.insert(post);
      }
   }
}