/*
 *    Copyright (c) 2018-2025, ztt All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * Author: ztt
 */
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.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 lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * 部门
 *
 * @author luh
 * @date 2020-08-14 15:20:19
 */
@Service
@AllArgsConstructor
public class DivisionServiceImpl extends ServiceImpl<DivisionMapper, Division> implements DivisionService {
       private CompanyMapper companyMapper;
		private FactoryMapper factoryMapper;
	/**
     * Description: 分页
     *
     * @param page
     * @param ew
     * @return IPage<List<Division>>
     * @author: luh
     * @date: 2020/8/18 11:06
     */
	@Override
	public IPage<List<Division>> getDivisionPage(Page page, QueryWrapper<Division> ew) {
		return baseMapper.getDivisionPage(page,ew);
	}

	@Override
	public void importWorkstationExcel(List<DivisionData> list) {
		list.forEach(System.out::println);
		if (CollectionUtil.isEmpty(list)) {
			return;
		}
		List<Company> companyList = companyMapper.selectList(null);
		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 ;
			}
			division.setDivisionNo(data.getDivisionNo());
			division.setDivisionName(data.getDivisionName());
			baseMapper.insert(division);
		}
	}

}