com.zhuang.excel.easyexcel.EasyExcelUtils Maven / Gradle / Ivy

Go to download
package com.zhuang.excel.easyexcel;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.alibaba.excel.write.metadata.fill.FillWrapper;

import java.io.*;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

public class EasyExcelUtils {

    public static void export(InputStream inputStream, OutputStream outputStream, List fillItemList) {
        ExcelWriter excelWriter = EasyExcel.write(outputStream).withTemplate(inputStream).build();
        WriteSheet writeSheet = EasyExcel.writerSheet().build();
        for (FillItem fillItem : fillItemList) {
            if (fillItem.getData() instanceof Collection) {
                excelWriter.fill(new FillWrapper(fillItem.getName(), (Collection) fillItem.getData()), fillItem.getFillConfig(), writeSheet);
            } else {
                excelWriter.fill(fillItem.getData(), fillItem.getFillConfig(), writeSheet);
            }
        }
        excelWriter.finish();
    }

    public static void export(String inputFilePath, String outputFilePath, List fillItemList) {
        try {
            InputStream inputStream = new FileInputStream(inputFilePath);
            OutputStream outputStream = new FileOutputStream(outputFilePath);
            export(inputStream, outputStream, fillItemList);
            outputStream.close();
            inputStream.close();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public static  void export(OutputStream outputStream, List data, Class head) {
        EasyExcel.write(outputStream, head).sheet().doWrite(data);
    }

    public static  void export(String outputFilePath, List data, Class head) {
        OutputStream outputStream = null;
        try {
            outputStream = new FileOutputStream(outputFilePath);
            EasyExcel.write(outputStream, head).sheet().doWrite(data);
            outputStream.close();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public static  List readToList(String inputFilePath, Class head) {
        try {
            InputStream inputStream = new FileInputStream(inputFilePath);
            List result = readToList(inputStream, head);
            inputStream.close();
            return result;
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public static  List readToList(InputStream inputStream, Class head) {
        List result = new ArrayList<>();
        EasyExcel.read(inputStream, head, new AnalysisEventListener() {
            @Override
            public void invoke(T data, AnalysisContext context) {
                result.add(data);
            }

            @Override
            public void doAfterAllAnalysed(AnalysisContext context) {
            }
        }).sheet().doRead();
        return result;
    }
}