package com.ydtech.utils;

import com.alibaba.excel.EasyExcel;
import com.ydtech.exception.SystemException;

import java.util.List;

/**
 * excel操作类  使用 easyexcel类库
 */
public class ExcelUtils<T> {
    private final Class<T> clazz;

    public ExcelUtils(Class<T> clazz) {
        this.clazz = clazz;
    }


    /**
     * 写入到Xls文件中
     * @param templatePath  模板路径
     * @param saveFileName  保存的文件名称
     * @param sheetName     sheet名称
     * @param data          数据
     */
    public boolean writeExcel(String templatePath,String saveFileName, String sheetName, List<T> data){
        try {
            EasyExcel.write(saveFileName, clazz).withTemplate(templatePath).sheet(sheetName).doWrite(data);
            return true;
        }catch(Exception e){
            return false;
        }
    }

    /**
     * 写入到Xls文件中
     * @param saveFileName  保存的文件名称
     * @param sheetName     sheet名称
     * @param data          数据
     * @return
     */
    public boolean writeExcel(String saveFileName, String sheetName,List<T> data) {
        try {
            EasyExcel.write(saveFileName, clazz).sheet(sheetName).registerWriteHandler(new CustomCellWriteHandler(0)).doWrite(data);
            return true;
        }catch(Exception e){
            return false;
        }
    }

    /**
     * 写入到Xls文件中
     * @param saveFileName  保存的文件名称
     * @param sheetName     sheet名称
     * @param data          数据
     * @return
     */
    public void writeExcelNew(String saveFileName, String sheetName,List<T> data) {
        try {
            EasyExcel.write(saveFileName, clazz).sheet(sheetName).registerWriteHandler(new CustomCellWriteHandler(0)).doWrite(data);
        }catch(Exception e){
            throw new SystemException(e.getMessage());
        }
    }
}