4dkankan-parent/4dkankan-center-manage/src/main/java/com/fdkankan/manage/service/impl/ExcelServiceImpl.java

package com.fdkankan.manage.service.impl;

import cn.hutool.core.date.DateUtil;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fdkankan.common.exception.BusinessException;
import com.fdkankan.common.util.DateExtUtil;
import com.fdkankan.manage.constant.CameraOutTypeEnum;
import com.fdkankan.manage.entity.Company;
import com.fdkankan.manage.service.ICameraService;
import com.fdkankan.manage.service.ICompanyService;
import com.fdkankan.manage.service.IExcelService;
import com.fdkankan.manage.service.IInvoiceService;
import com.fdkankan.manage.util.ExcelUtil;
import com.fdkankan.manage.vo.InvoicePageParamVO;
import com.fdkankan.manage.vo.InvoiceVO;
import com.fdkankan.manage.vo.request.CameraInOutParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.HashMap;
import java.util.List;

/**
 * <p>
 * TODO
 * </p>
 *
 * @author dengsixing
 * @since 2022/6/6
 **/
@Service
public class ExcelServiceImpl implements IExcelService {

    @Autowired
    private IInvoiceService invoiceService;
    @Autowired
    ICompanyService companyService;
    @Autowired
    ICameraService cameraService;

    @Override
    public void exportInvoice(HttpServletRequest request, HttpServletResponse response, InvoicePageParamVO param)
        throws IOException {
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        String fileName = "开票申请数据" + DateUtil.format(Calendar.getInstance().getTime(), DateExtUtil.dateStyle6) + ".xlsx";
        fileName = URLEncoder.encode(fileName, "UTF-8");
        response.setHeader("Content-disposition", "attachment;filename=" + fileName);
        try (ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream(), InvoiceVO.class).build()) {
            // 这里注意 如果同一个sheet只要创建一次
            WriteSheet writeSheet = EasyExcel.writerSheet("开票申请数据").build();
            // 去调用写入,这里我调用了五次，实际使用时根据数据库分页的总的页数来
            int pageNum = 1;
            param.setPageSize(5000);
            Page<InvoiceVO> invoiceVOPage;
            do {
                param.setPageNum(pageNum);
                invoiceVOPage = invoiceService.pageInvoice(param);
                excelWriter.write(invoiceVOPage.getRecords(), writeSheet);
                ++pageNum;

            }while (invoiceVOPage.hasNext());
        }
    }

    @Override
    public Integer uploadExcel(MultipartFile file, Integer type) throws IOException {
        String originalFilename = file.getOriginalFilename();
        assert originalFilename != null;
        String fileType=originalFilename.substring(originalFilename.lastIndexOf(".")+1);
        if (!fileType.equalsIgnoreCase("xls") && !fileType.equalsIgnoreCase("xlsx")) {
            throw new BusinessException(-1,"文件格式校验，需为excel文件;");
        }
        List<HashMap<Integer, String>> excelRowList = ExcelUtil.getExcelRowList(file);
        List<String> wifiNameList = new ArrayList<>();
        List<CameraInOutParam> params = new ArrayList<>();
        List<CameraInOutParam> companyParams = new ArrayList<>();
        for (HashMap<Integer, String> map : excelRowList) {
            if(type == 0){      //入库
                String wifiName = map.get(0);
                wifiNameList.add(wifiName);
            }else if(type == 1){    //出库
                String snCode = map.get(0);
                String outTypeString = map.get(1);
                String companyName = map.get(2);
                String orderSn = map.get(3);

                CameraOutTypeEnum outTypeEnum = CameraOutTypeEnum.getByMsg(outTypeString);
                if(outTypeEnum == null){
                    throw new BusinessException(-1,"出库类型错误："+outTypeString);
                }
                int outType = outTypeEnum.getCode();
                Company company = companyService.getCompanyByName(companyName);
                CameraInOutParam param = new CameraInOutParam();
                param.setCompanyId(company.getId());
                param.setOutType(outType);
                param.setOrderSn(orderSn);
                param.setSnCode(snCode);
                params.add(param);
            }else if(type == 2){   //关联客户
                CameraInOutParam param = new CameraInOutParam();
                param.setCompanyName(map.get(0));
                param.setSnCode(map.get(1));
                companyParams.add(param);
            }
        }
        if(wifiNameList.size() >0){
            return cameraService.ins(wifiNameList);
        }
        if(params.size() >0){
            return cameraService.outs(params);
        }
        if(companyParams.size() >0){
            return cameraService.updateCompany(companyParams);
        }
        return 0;
    }

}