4dkankan-center-manage/src/main/java/com/fdkankan/manage/factory/CustomerRelaHandler.java

package com.fdkankan.manage.factory;

import cn.hutool.core.collection.CollUtil;
import com.fdkankan.manage.common.ResultCode;
import com.fdkankan.manage.constant.CameraOutTypeEnum;
import com.fdkankan.manage.entity.AgentNew;
import com.fdkankan.manage.exception.BusinessException;
import com.fdkankan.manage.service.ICameraService;
import com.fdkankan.manage.service.IExcelService;
import com.fdkankan.manage.vo.request.CameraInOutParam;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

@Component("customerRela")
public class CustomerRelaHandler implements ImportExcelHandler{

    @Autowired
    private IExcelService excelService;
    @Autowired
    private ICameraService cameraService;

    @Override
    public int importExcel(List<HashMap<Integer, String>> excelRowList) {

        if(CollUtil.isEmpty(excelRowList)){
            return 0;
        }

        List<Integer> errorIndex = new ArrayList<>();
        List<CameraInOutParam> companyParams = new ArrayList<>();

        Integer index = 0;
        for (HashMap<Integer, String> map : excelRowList) {
            index ++;
            if(index == 0 && !map.get(0).equals("客户关联模板")){
                throw new BusinessException(ResultCode.TEMPLATE_TYPE_ERROR);
            }
            if(index <4){
                continue;
            }
            CameraInOutParam param = new CameraInOutParam();
            param.setCompanyName(map.get(0));
            param.setSnCode(map.get(1));
            if(StringUtils.isBlank(param.getSnCode()) || StringUtils.isBlank(param.getCompanyName())){
                errorIndex.add(index -3 );
            }
            companyParams.add(param);
        }
        excelService.toExcelError(errorIndex);

        if(companyParams.size() <=0){
            throw new BusinessException(ResultCode.COMPANY_TEMPLATE_EMPTY);
        }

        return cameraService.updateCompany(companyParams);
    }
}