cn.creekmoon.excelUtils.core.ExcelImport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of excel-utils-boot-starter Show documentation
一个简易操作Excel的工具类
There is a newer version: 1.3.0
Show newest version
package cn.creekmoon.excelUtils.core;

import cn.creekmoon.excelUtils.config.ExcelUtilsConfig;
import cn.creekmoon.excelUtils.threadPool.CleanTempFilesExecutor;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.text.csv.CsvReader;
import cn.hutool.core.text.csv.CsvRow;
import cn.hutool.core.util.StrUtil;
import cn.hutool.poi.excel.sax.Excel07SaxReader;
import cn.hutool.poi.excel.sax.handler.RowHandler;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
import java.util.concurrent.Semaphore;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;
import java.util.function.Supplier;

/**
 * @author JY
 * @date 2022-01-05
 */
@Slf4j
public class ExcelImport {
    /**
     * 控制导入并发数量
     */
    public static Semaphore importSemaphore;

    public HashMap sheetIndex2SheetReader = new HashMap<>();

    public HashMap> convertObject2rawData = new HashMap<>();
    public HashMap>> sheetIndex2rawData = new LinkedHashMap<>();

    /*当前导入的文件*/
    protected MultipartFile file;

    /*错误次数统计*/
    protected AtomicInteger errorCount = new AtomicInteger(0);
    /**
     * 导入结果
     */
    protected ExcelExport excelExport;


    private ExcelImport() {
    }


    public static void init() {
        if (importSemaphore == null) {
            importSemaphore = new Semaphore(ExcelUtilsConfig.IMPORT_MAX_PARALLEL);
        }
    }


    public static ExcelImport create(MultipartFile file) {
        if (importSemaphore == null) {
            throw new RuntimeException("请使用@EnableExcelUtils进行初始化配置!");
        }
        ExcelImport excelImport = new ExcelImport();
        excelImport.file = file;
        excelImport.csvSupport();
        /*初始化一个导入结果*/
        excelImport.excelExport = ExcelExport.create();

        return excelImport;
    }


    public static  SheetReader create(MultipartFile file, Supplier supplier) {
        ExcelImport excelImport = create(file);
        return excelImport.switchSheet(0, supplier);
    }

    /**
     * 切换读取的sheet页
     *
     * @param sheetIndex 下标,从0开始
     * @param supplier   按行读取时,每行数据的实例化对象构造函数
     * @param 
     * @return
     */
    public  CellReader switchSheetAndUseCellReader(int sheetIndex, Supplier supplier) {
        CellReader reader = new CellReader<>();
        reader.sheetReaderContext = new SheetReaderContext(sheetIndex, supplier);
        ;
        reader.parent = this;
        return reader;
    }

    /**
     * 获取SHEET页的总行数
     *
     * @return
     */
    @SneakyThrows
    public Long getSheetRowCount(int targetSheetIndex) {
        AtomicLong result = new AtomicLong(0);
        Excel07SaxReader excel07SaxReader = new Excel07SaxReader(new RowHandler() {
            @Override
            public void handle(int sheetIndex, long rowIndex, List