前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >poi系列(二):通过poi、poi-ooxml读取写出excel

poi系列(二):通过poi、poi-ooxml读取写出excel

作者头像
全栈程序员站长
发布2022-09-13 10:58:31
3.4K0
发布2022-09-13 10:58:31
举报
文章被收录于专栏:全栈程序员必看

大家好,又见面了,我是你们的朋友全栈君。

(根据网上多篇文档实践整理加工,非原创也非转载)

本类主要是读取后缀为xlsx或xls的excel操作。 需要导入包

<dependency> <groupId>org.apache.poi</groupId> <artifactId>poi</artifactId> <version>3.17</version> </dependency> <dependency> <groupId>org.apache.poi</groupId> <artifactId>poi-ooxml</artifactId> <version>3.17</version> </dependency>

代码语言:javascript
复制
package com.clf.equity.manage.utils;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.apache.commons.beanutils.BeanUtils;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 
/**
 * 里面使用的所有行数都是从0开始的
 * 使用的jar包
 * <dependency>
 * <groupId>org.apache.poi</groupId>
 * <artifactId>poi-ooxml</artifactId>
 * <version>3.17</version>
 * </dependency>
 * <dependency>
 * <groupId>commons-beanutils</groupId>
 * <artifactId>commons-beanutils</artifactId>
 * <version>1.9.3</version>
 * </dependency>
 */
public class ExcelUtil {
 
    /**
     * 将对象数组转换成excel<br/>
     *
     * @param pojoList  对象数组
     * @param out       输出流
     * @param alias     指定对象属性别名,生成列名和列顺序Map<"类属性名","列名">
     * @param utilExcel 表头对象
     * @throws Exception
     */
    public static <T> void pojo2Excel(List<T> pojoList, OutputStream out, LinkedHashMap<String, String> alias, UtilExcel utilExcel) throws Exception {
        //创建一个工作簿
        XSSFWorkbook wb = new XSSFWorkbook();
        if (utilExcel == null) utilExcel = new UtilExcel();
        //创建一个表
        XSSFSheet sheet = wb.createSheet();
        // 需要表头
        if (utilExcel.getFieldRow() > utilExcel.getTableHeadRow()) {
            //创建第一行,作为表名
            XSSFRow row = sheet.createRow(utilExcel.getTableHeadRow());// 这个方法感觉是直接跳到对应行的
            XSSFCell cell = row.createCell(0);
            cell.setCellValue(utilExcel.getTableHeadName());
 
            sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, alias.size() - 1));
        }
 
 
        // 在第一行插入列名
        insertColumnName(utilExcel.getFieldRow(), sheet, alias);
 
        // 从第指定行开始插入数据
        insertColumnDate(utilExcel.getDataStarRow(), pojoList, sheet, alias);
 
        // 输出表格文件
        try {
            wb.write(out);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            wb.close();
        }
    }
 
 
    /**
     * 多个sheet导出excel 复杂表头或者非复杂表头<br/>
     *
     * @param exportList sheet对象的list
     * @param out        输出流
     * @throws Exception
     */
    public static <T> void pojo2ExcelSheetList(List<SheetExport> exportList, OutputStream out) throws Exception {
        //创建一个工作簿
        XSSFWorkbook wb = new XSSFWorkbook();
        // 设置居中样式
        XSSFCellStyle xssStyle = wb.createCellStyle();
        xssStyle.setAlignment(HorizontalAlignment.CENTER);
        xssStyle.setVerticalAlignment(VerticalAlignment.CENTER);
 
        for (SheetExport sheetData : exportList) {
            //创建一个表
            XSSFSheet sheet = wb.createSheet(sheetData.getSheetName());
            // 需要表头
            if (sheetData.getUtilExcel().getFieldRow() > sheetData.getUtilExcel().getTableHeadRow()) {
                XSSFRow row = sheet.createRow(sheetData.getUtilExcel().getTableHeadRow());// 这个方法感觉是直接跳到对应行的
                XSSFCell cell = row.createCell(0);
                cell.setCellValue(sheetData.getUtilExcel().getTableHeadName());
                sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, sheetData.getAlias().size() - 1));
 
                // 设置居中样式
                cell.setCellStyle(xssStyle);
            }
            if (sheetData.getMergeDataList() == null) {
                if (sheetData.getUtilExcel().getFieldRow() < sheetData.getUtilExcel().getDataStarRow()) {
                    // 插入列名
                    insertColumnName(sheetData.getUtilExcel().getFieldRow(), sheet, sheetData.getAlias());
                    // 从第指定行开始插入数据
                    insertColumnDate(sheetData.getUtilExcel().getDataStarRow(), sheetData.getPojoList(), sheet, sheetData.getAlias());
                } else {
                    insertColumnName(sheetData.getUtilExcel().getFieldRow(), sheet, sheetData.getAlias());
                    insertColumnDate(sheetData.getUtilExcel().getFieldRow() + 1, sheetData.getPojoList(), sheet, sheetData.getAlias());
                }
            } else {
                // 插入复杂表头(表的标题和字段名之间)
                XSSFRow rowTable = sheet.createRow(sheetData.getUtilExcel().getTableHeadRow() + 1);
                for (MergeData mergeData : sheetData.getMergeDataList()) {
                    sheet.addMergedRegion(new CellRangeAddress(mergeData.getStartRow(), mergeData.getEndRow(), mergeData.getStartCol(), mergeData.getEndCol()));
                    // 插入复杂表头的数据
                    XSSFCell tableCellValue = rowTable.createCell(mergeData.getStartCol());
                    tableCellValue.setCellValue(mergeData.getName());
                    // 这里可以对单元格做样式处理
                    // 设置居中样式
                    tableCellValue.setCellStyle(xssStyle);
                }
                // 如果插入数据的行小于指定的数据行,就默认在复杂表头的下方
                int maxHeadRow = 0;
                for (MergeData me : sheetData.getMergeDataList()) {
                    if (me.getEndRow() > maxHeadRow) maxHeadRow = me.getEndRow();
                }
                if (maxHeadRow < sheetData.getUtilExcel().getFieldRow() && sheetData.getUtilExcel().getFieldRow() < sheetData.getUtilExcel().getDataStarRow()) {
                    // 插入列名
                    insertColumnName(sheetData.getUtilExcel().getFieldRow(), sheet, sheetData.getAlias());
                    // 从第指定行开始插入数据
                    insertColumnDate(sheetData.getUtilExcel().getDataStarRow(), sheetData.getPojoList(), sheet, sheetData.getAlias());
                } else {
                    insertColumnName(maxHeadRow + 1, sheet, sheetData.getAlias());
                    insertColumnDate(maxHeadRow + 2, sheetData.getPojoList(), sheet, sheetData.getAlias());
                }
            }
        }
        // 输出表格文件
        try {
            wb.write(out);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            wb.close();
        }
    }
    /**
     * 将对象数组转换成excel,并增加复杂表头,除去表名和显示列的名称的那一行
     *
     * @param pojoList      对象数组
     * @param out           输出流
     * @param alias         指定对象属性别名,生成列名和列顺序Map<"类属性名","列名">
     * @param utilExcel     表头对象
     * @param mergeDataList 合并行中的所有数据,包括不和并的
     * @throws Exception
     */
    public static <T> void pojo2Excel(List<T> pojoList, OutputStream out, LinkedHashMap<String, String> alias, UtilExcel utilExcel, List<MergeData> mergeDataList) throws Exception {
        if (utilExcel == null) throw new Exception("UtilExcel 对象为空");
        //创建一个工作簿
        XSSFWorkbook wb = new XSSFWorkbook();
        // 设置居中样式
        // 设置表头文字格式
        // XSSFCellStyle cellStyle = wb.createCellStyle();
        // XSSFFont font = wb.createFont();
        // font.setFontName("宋体");
        // font.setFontHeightInPoints((short) 36);
        // cellStyle.setFont(font);
        // cellStyle.setAlignment(HorizontalAlignment.CENTER);
        XSSFCellStyle xssStyle = wb.createCellStyle();
        xssStyle.setAlignment(HorizontalAlignment.CENTER);
        xssStyle.setVerticalAlignment(VerticalAlignment.CENTER);
 
        //创建一个表
        XSSFSheet sheet = wb.createSheet();
        // 需要表头
        if (utilExcel.getFieldRow() > utilExcel.getTableHeadRow()) {
            //创建第一行,作为表名
            XSSFRow row = sheet.createRow(utilExcel.getTableHeadRow());// 这个方法感觉是直接跳到对应行的 后面不需要再次调用该方法,应该是使用该方法 可以独立设置该行的样式
            XSSFCell cell = row.createCell(0);
            cell.setCellValue(utilExcel.getTableHeadName());
            sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, alias.size() - 1));
 
            // 设置居中样式
            cell.setCellStyle(xssStyle);
        }
 
        // 插入复杂表头
        XSSFRow rowTable = sheet.createRow(utilExcel.getTableHeadRow() + 1);
        for (MergeData mergeData : mergeDataList) {
            sheet.addMergedRegion(new CellRangeAddress(mergeData.getStartRow(), mergeData.getEndRow(), mergeData.getStartCol(), mergeData.getEndCol()));
            // 插入数据
            XSSFCell tableCellValue = rowTable.createCell(mergeData.getStartCol());
            tableCellValue.setCellValue(mergeData.getName());
 
            // 这里可以对单元格做样式处理
            // 设置居中样式
            tableCellValue.setCellStyle(xssStyle);
        }
 
        // 如果插入数据的行小于指定的数据行,就默认在复杂表头的下方
        int maxHeadRow = 0;
        for (MergeData me : mergeDataList) {
            if (me.getEndRow() > maxHeadRow) maxHeadRow = me.getEndRow();
        }
        if (maxHeadRow < utilExcel.getFieldRow() && utilExcel.getFieldRow() < utilExcel.getDataStarRow()) {
            // 插入列名
            insertColumnName(utilExcel.getFieldRow(), sheet, alias);
            // 从第指定行开始插入数据
            insertColumnDate(utilExcel.getDataStarRow(), pojoList, sheet, alias);
        } else {
            insertColumnName(maxHeadRow + 1, sheet, alias);
            insertColumnDate(maxHeadRow + 2, pojoList, sheet, alias);
        }
        // 输出表格文件
        try {
            wb.write(out);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            wb.close();
        }
    }
    /**
     * 将excel表转换成指定类型的对象数组
     *
     * @param claz  类型
     * @param alias 列别名,格式要求:Map<"列名","类属性名">
     * @param param 指定第几行行为字段名(数据在字段的下一行,默认),第一行为0
     * @return
     * @throws IOException
     * @throws IllegalArgumentException
     * @throws IllegalAccessException
     * @throws SecurityException
     * @throws NoSuchFieldException
     * @throws InstantiationException
     * @throws InvocationTargetException
     */
    public static <T> List<T> excel2Pojo(InputStream inputStream, Class<T> claz, LinkedHashMap<String, String> alias, Integer param) throws IOException {
        XSSFWorkbook wb = new XSSFWorkbook(inputStream);
        try {
            XSSFSheet sheet = wb.getSheetAt(0);
 
            //生成属性和列对应关系的map,Map<类属性名,对应一行的第几列>
            Map<String, Integer> propertyMap = generateColumnPropertyMap(sheet, alias, param);
            //根据指定的映射关系进行转换
            List<T> pojoList = generateList(sheet, propertyMap, claz, param);
            return pojoList;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            wb.close();
        }
    }
 
    public static <T> List<T> excel2PojoSheetList(List<SheetImport> list, InputStream inputStream) throws IOException {
        XSSFWorkbook wb = new XSSFWorkbook(inputStream);
        List<T> pojoList = new ArrayList<>();
        for (int i = 0; i < list.size(); i++) {
            try {
                XSSFSheet sheet = wb.getSheetAt(i);
                //生成属性和列对应关系的map,Map<类属性名,对应一行的第几列>
                Map<String, Integer> propertyMap = generateColumnPropertyMap(sheet, list.get(i).getAlias(), list.get(i).getParam());
                //根据指定的映射关系进行转换
                pojoList.add((T) generateList(sheet, propertyMap, list.get(i).getClaz(), list.get(i).getParam()));
 
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                wb.close();
            }
        }
        return pojoList;
    }
 
    /**
     * 将对象数组转换成excel
     *
     * @param pojoList 对象数组
     * @param out      输出流
     * @param alias    指定对象属性别名,生成列名和列顺序
     * @throws Exception
     */
    public static <T> void pojo2Excel(List<T> pojoList, OutputStream out, LinkedHashMap<String, String> alias) throws Exception {
        //获取类名作为标题
        String headLine = "";
        if (pojoList.size() > 0) {
            Object pojo = pojoList.get(0);
            Class<? extends Object> claz = pojo.getClass();
            headLine = claz.getName();
            pojo2Excel(pojoList, out, alias, new UtilExcel(headLine, 1));
        }
    }
 
    /**
     * 将对象数组转换成excel,列名为对象属性名
     *
     * @param pojoList 对象数组
     * @param out      输出流
     * @param headLine 表标题
     * @throws Exception
     */
    public static <T> void pojo2Excel(List<T> pojoList, OutputStream out, String headLine) throws Exception {
        //获取类的属性作为列名
        LinkedHashMap<String, String> alias = new LinkedHashMap<String, String>();
        if (pojoList.size() > 0) {
            Object pojo = pojoList.get(0);
            Field[] fields = pojo.getClass().getDeclaredFields();
            String[] name = new String[fields.length];
            Field.setAccessible(fields, true);
            for (int i = 0; i < name.length; i++) {
                name[i] = fields[i].getName();
                alias.put(isNull(name[i]).toString(), isNull(name[i]).toString());
            }
            pojo2Excel(pojoList, out, alias, new UtilExcel(headLine, 1));
        }
    }
 
    /**
     * 将对象数组转换成excel,列名默认为对象属性名,标题为类名
     *
     * @param pojoList 对象数组
     * @param out      输出流
     * @throws Exception
     */
    public static <T> void pojo2Excel(List<T> pojoList, OutputStream out) throws Exception {
        //获取类的属性作为列名
        LinkedHashMap<String, String> alias = new LinkedHashMap<String, String>();
        //获取类名作为标题
        String headLine = "";
        if (pojoList.size() > 0) {
            Object pojo = pojoList.get(0);
            Class<? extends Object> claz = pojo.getClass();
            headLine = claz.getName();
            Field[] fields = claz.getDeclaredFields();
            String[] name = new String[fields.length];
            Field.setAccessible(fields, true);
            for (int i = 0; i < name.length; i++) {
                name[i] = fields[i].getName();
                alias.put(isNull(name[i]).toString(), isNull(name[i]).toString());
            }
            pojo2Excel(pojoList, out, alias, new UtilExcel(headLine, 1));
        }
    }
 
    /**
     * 此方法作用是创建表头的列名
     *
     * @param alias  要创建的表的列名与实体类的属性名的映射集合
     * @param rowNum 指定行创建列名
     * @return
     */
    private static void insertColumnName(int rowNum, XSSFSheet sheet, Map<String, String> alias) {
        XSSFRow row = sheet.createRow(rowNum);
        //列的数量
        int columnCount = 0;
 
        Set<Entry<String, String>> entrySet = alias.entrySet();
 
        for (Entry<String, String> entry : entrySet) {
            // 创建第一行的第columnCount个格子
            XSSFCell cell = row.createCell(columnCount++);
            // 将此格子的值设置为alias中的键名
            cell.setCellValue(isNull(entry.getValue()).toString());
        }
    }
 
    /**
     * 从指定行开始插入数据
     *
     * @param beginRowNum 开始行
     * @param models      对象数组
     * @param sheet       表
     * @param alias       列别名
     * @throws Exception
     */
    private static <T> void insertColumnDate(int beginRowNum, List<T> models, XSSFSheet sheet, Map<String, String> alias) throws Exception {
        for (T model : models) {
            // 创建新的一行
            XSSFRow rowTemp = sheet.createRow(beginRowNum++);
            // 获取列的迭代
            Set<Entry<String, String>> entrySet = alias.entrySet();
 
            // 从第0个格子开始创建
            int columnNum = 0;
            for (Entry<String, String> entry : entrySet) {
                // 获取属性值
                String property = BeanUtils.getProperty(model, entry.getKey());
                // 创建一个格子
                XSSFCell cell = rowTemp.createCell(columnNum++);
                // 得知string可以转化的类型
                if (isDouble(property)) {
                    cell.setCellValue(Double.valueOf(property));
                } else if (isInt(property)) {
                    cell.setCellValue(Integer.valueOf(property));
                } else if (isDateAndTime(property)) {
                    // 只对日期加time的做转化
                    SimpleDateFormat formatter;
                    if (property.indexOf("-") >= 1) {
                        formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
                    } else if (property.indexOf("/") >= 1) {
                        formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
                    } else if (property.indexOf(".") >= 1) {
                        formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
                    } else {
                        formatter = new SimpleDateFormat("yyyyMMdd HH:mm:ss");
                    }
                    Date date = formatter.parse(property);
                    cell.setCellValue(date);
                } else {
                    cell.setCellValue(property);
                }
            }
        }
    }
 
    // 判断是否为空,若为空设为""
    private static Object isNull(Object object) {
        if (object != null) {
            return object;
        } else {
            return "";
        }
    }
 
    /**
     * 生成一个属性-列的对应关系的map
     *
     * @param sheet 表
     * @param alias 别名
     * @return
     */
    private static Map<String, Integer> generateColumnPropertyMap(XSSFSheet sheet, LinkedHashMap<String, String> alias, Integer param) {
        Map<String, Integer> propertyMap = new HashMap<>();
 
        if (param == null || param < 0) param = 1;
        XSSFRow propertyRow = sheet.getRow(param);
        short firstCellNum = propertyRow.getFirstCellNum();
        short lastCellNum = propertyRow.getLastCellNum();
 
        for (int i = firstCellNum; i < lastCellNum; i++) {
            Cell cell = propertyRow.getCell(i);
            if (cell == null) {
                continue;
            }
            // 列名
            String cellValue = cell.getStringCellValue();
            // 对应属性名
            String propertyName = alias.get(cellValue);
            propertyMap.put(propertyName, i);
        }
        return propertyMap;
    }
 
    /**
     * 根据指定关系将表数据转换成对象数组
     *
     * @param sheet       表
     * @param propertyMap 属性映射关系Map<"属性名",一行第几列>
     * @param claz        类类型
     * @return
     * @throws InstantiationException
     * @throws IllegalAccessException
     * @throws InvocationTargetException
     */
    private static <T> List<T> generateList(XSSFSheet sheet, Map<String, Integer> propertyMap, Class<T> claz, Integer param) throws Exception {
        if (param == null || param < 0) param = 1;
        // 对象数组
        List<T> pojoList = new ArrayList<>();
        int index = 0;
        for (Row row : sheet) {
            // 跳过标题和列名
            if (row.getRowNum() < param + 1) {
                continue;
            }
            T instance = claz.newInstance();
            Set<Entry<String, Integer>> entrySet = propertyMap.entrySet();
            for (Entry<String, Integer> entry : entrySet) {
                /*
                 * CellTypeEnum        类型        值
                 * NUMERIC             数值型      0
                 * STRING              字符串型     1
                 * FORMULA             公式型      2
                 * BLANK               空值        3
                 * BOOLEAN             布尔型      4
                 * ERROR               错误        5
                 *
                 * 4.0以上将会移除 替换为getCellType
                 * */
                // 获取此行指定列的值,即为属性对应的值
                switch (row.getCell(entry.getValue()).getCellTypeEnum()) {
                    case _NONE:
                        System.out.println("****************************不知道的类型*********************************");
                        throw new Exception("第" + index + "行【" + row.getCell(entry.getValue()) + "】导入数据异常");
                    case BLANK:
                        BeanUtils.setProperty(instance, entry.getKey(), null);
                        break;
                    case NUMERIC:
                        int numericType = row.getCell(entry.getValue()).getCellStyle().getDataFormat();
                        if (numericType == 0) {// 数字类型
                            int pInt = (int) row.getCell(entry.getValue()).getNumericCellValue();
                            BeanUtils.setProperty(instance, entry.getKey(), pInt);
                            break;
                        } else {
                            Date date = row.getCell(entry.getValue()).getDateCellValue();
                            BeanUtils.setProperty(instance, entry.getKey(), date);
                            break;
                        }
                    case STRING:
                        String pString = row.getCell(entry.getValue()).getStringCellValue();
                        BeanUtils.setProperty(instance, entry.getKey(), pString);
                        break;
                    case FORMULA:
                        System.out.println("**该类型【FORMULA】未做处理,因为没见过这种类型,于ExcelUtil2.generateList方法中修改!");
                        break;
                    case BOOLEAN:
                        boolean pBoolean = row.getCell(entry.getValue()).getBooleanCellValue();
                        BeanUtils.setProperty(instance, entry.getKey(), pBoolean);
                        break;
                    case ERROR:
                        System.out.println("****************************error*********************************");
                        throw new Exception("第" + index + "行【" + row.getCell(entry.getValue()) + "】导入数据异常");
                }
            }
            pojoList.add(instance);
            index++;
        }
        return pojoList;
    }
 
    /**
     * 将excel表转换成指定类型的对象数组,列名即作为对象属性
     *
     * @param claz 类型
     * @return
     * @throws IOException
     * @throws InstantiationException
     * @throws SecurityException
     * @throws NoSuchFieldException
     * @throws IllegalAccessException
     * @throws IllegalArgumentException
     * @throws InvocationTargetException
     */
    public static <T> List<T> excel2Pojo(InputStream inputStream, Class<T> claz) throws IllegalArgumentException, IllegalAccessException, NoSuchFieldException, SecurityException, InstantiationException, IOException, InvocationTargetException {
        LinkedHashMap<String, String> alias = new LinkedHashMap<String, String>();
        Field[] fields = claz.getDeclaredFields();
        for (Field field : fields) {
            alias.put(field.getName(), field.getName());
        }
        List<T> pojoList = excel2Pojo(inputStream, claz, alias, 1);
        return pojoList;
    }
 
 
    /**
     * String可以转化的类型判断
     *
     * @param str
     */
//    是否为浮点数
    private static boolean isDouble(String str) {
        return str.matches("^[-+]?[1-9][0-9]*\\.?[0-9]+$");
    }
 
    //    是否为整数
    private static boolean isInt(String str) {
        return str.matches("^[-+]?[1-9]\\d*$");
    }
 
    //    必须日期加时间 [2018-02-14 00:00:00] 使用反向引用进行简化,年份0001-9999,格式yyyy-MM-dd或yyyy-M-d,连字符可以没有或是“-”、“/”、“.”之一。
    private static boolean isDateAndTime(String str) {
        return str.matches("^(?:(?!0000)[0-9]{4}([-/.]?)(?:(?:0?[1-9]|1[0-2])\\1(?:0?[1-9]|1[0-9]|2[0-8])|(?:0?[13-9]|1[0-2])\\1(?:29|30)|(?:0?[13578]|1[02])\\1(?:31))|(?:[0-9]{2}(?:0[48]|[2468][048]|[13579][26])|(?:0[48]|[2468][048]|[13579][26])00)([-/.]?)0?2\\2(?:29))\\s+([01][0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9]$");
    }
 
    //    必须日期加时间 [2018-02-14] 使用反向引用进行简化,年份0001-9999,格式yyyy-MM-dd或yyyy-M-d,连字符可以没有或是“-”、“/”、“.”之一。
    private static boolean isDate(String str) {
        return str.matches("^(?:(?!0000)[0-9]{4}([-/.]?)(?:(?:0?[1-9]|1[0-2])([-/.]?)(?:0?[1-9]|1[0-9]|2[0-8])|(?:0?[13-9]|1[0-2])([-/.]?)(?:29|30)|(?:0?[13578]|1[02])([-/.]?)31)|(?:[0-9]{2}(?:0[48]|[2468][048]|[13579][26])|(?:0[48]|[2468][048]|[13579][26])00)([-/.]?)0?2([-/.]?)29)$");
    }
}
 
 
/**
 * @Author: TianYiWu
 * @Description:  表格基本信息类
 * @Date: Created in 2019年3月11日 下午2:18:43
 * @Since: JDK1.8
 * @Version: 1.0
 */
class UtilExcel {
    private String tableHeadName;// 表头名称
    private Integer tableHeadRow;// 表头名称所在的行
    private Integer fieldRow;// 字段所在的行
    private Integer dataStarRow;// 插入数据开始的row
 
    /**
     * 默认sheet的信息<br/>
     * tableHeadName = "export excel"<br/>
     * fieldRow = 0<br/>
     * dataStarRow = 1<br/>
     */
    UtilExcel() {
        this.tableHeadName = "export excel";
        this.tableHeadRow = 0;
        this.fieldRow = 1;
        this.dataStarRow = 2;
    }
 
    /**
     * sheet的基本信息
     *
     * @param tableHeadName 表头名称
     * @param fieldRow      sheet表格对应实体字段所在的行
     */
    public UtilExcel(String tableHeadName, int fieldRow) {
        this.tableHeadName = tableHeadName;
        this.tableHeadRow = 0;// 如果fieldRow = tableHeadRow,则没有表头
        this.fieldRow = fieldRow;
        this.dataStarRow = fieldRow + 1;
    }
 
    /**
     * sheet的基本信息
     *
     * @param tableHeadName 表头名称
     * @param fieldRow      sheet表格字段所在的行
     * @param dataStarRow   插入数据开始的行
     */
    public UtilExcel(String tableHeadName, int fieldRow, int dataStarRow) {
        this.tableHeadName = tableHeadName;
        if (fieldRow > 0) {
            this.tableHeadRow = fieldRow - 1;// 如果fieldRow = tableHeadRow,则没有表头
        } else {
            this.tableHeadRow = 0;
        }
 
        this.fieldRow = fieldRow;
        this.dataStarRow = dataStarRow;
    }
 
    /**
     * sheet的基本信息
     *
     * @param tableHeadName 表头名称
     * @param tableHeadRow  表头名称所在的行
     * @param fieldRow      sheet表格字段所在的行
     * @param dataStarRow   插入数据开始的行
     */
    public UtilExcel(String tableHeadName, int tableHeadRow, int fieldRow, int dataStarRow) {
        this.tableHeadName = tableHeadName;
        this.tableHeadRow = tableHeadRow;
        this.fieldRow = fieldRow;
        this.dataStarRow = dataStarRow;
    }
 
    public String getTableHeadName() {
        return tableHeadName;
    }
 
    public void setTableHeadName(String tableHeadName) {
        this.tableHeadName = tableHeadName;
    }
 
    public Integer getTableHeadRow() {
        return tableHeadRow;
    }
 
    public void setTableHeadRow(Integer tableHeadRow) {
        this.tableHeadRow = tableHeadRow;
    }
 
    public Integer getFieldRow() {
        return fieldRow;
    }
 
    public void setFieldRow(Integer fieldRow) {
        this.fieldRow = fieldRow;
    }
 
    public Integer getDataStarRow() {
        return dataStarRow;
    }
 
    public void setDataStarRow(Integer dataStarRow) {
        this.dataStarRow = dataStarRow;
    }
}
 
/**
 * @Author: TianYiWu
 * @Description:  表格起止控制类
 * @Date: Created in 2019年3月11日 下午2:16:44
 * @Since: JDK1.8
 * @Version: 1.0
 */
class MergeData {
    private String name;
    private int startRow;
    private int endRow;
    private int startCol;
    private int endCol;
 
    public MergeData() {
    }
 
    public MergeData(String name, int startRow, int endRow, int startCol, int endCol) {
        this.name = name;
        this.startRow = startRow;
        this.endRow = endRow;
        this.startCol = startCol;
        this.endCol = endCol;
    }
 
    public String getName() {
        return name;
    }
 
    public void setName(String name) {
        this.name = name;
    }
 
    public int getStartRow() {
        return startRow;
    }
 
    public void setStartRow(int startRow) {
        this.startRow = startRow;
    }
 
    public int getEndRow() {
        return endRow;
    }
 
    public void setEndRow(int endRow) {
        this.endRow = endRow;
    }
 
    public int getStartCol() {
        return startCol;
    }
 
    public void setStartCol(int startCol) {
        this.startCol = startCol;
    }
 
    public int getEndCol() {
        return endCol;
    }
 
    public void setEndCol(int endCol) {
        this.endCol = endCol;
    }
}
/**
 * @Author: TianYiWu
 * @Description:  表格导出实体类
 * @Date: Created in 2019年3月11日 下午2:15:39
 * @Since: JDK1.8
 * @Version: 1.0
 */
class SheetExport {
    private String sheetName;
    private List<?> pojoList;
    private LinkedHashMap<String, String> alias;
    private UtilExcel utilExcel;
    private List<MergeData> mergeDataList;
 
    public String getSheetName() {
        return sheetName;
    }
 
    public void setSheetName(String sheetName) {
        if (!"".equals(sheetName)) this.sheetName = sheetName;
    }
 
    public List<?> getPojoList() {
        return pojoList;
    }
 
    public void setPojoList(List<?> pojoList) {
        this.pojoList = pojoList;
    }
 
    public LinkedHashMap<String, String> getAlias() {
        return alias;
    }
 
    public void setAlias(LinkedHashMap<String, String> alias) {
        this.alias = alias;
    }
 
    public UtilExcel getUtilExcel() {
        return utilExcel;
    }
 
    public void setUtilExcel(UtilExcel utilExcel) {
        this.utilExcel = utilExcel;
    }
 
    public List<MergeData> getMergeDataList() {
        return mergeDataList;
    }
 
    public void setMergeDataList(List<MergeData> mergeDataList) {
        this.mergeDataList = mergeDataList;
    }
}
/**
 * @Author: TianYiWu
 * @Description:  表格导入实体类
 * @Date: Created in 2019年3月11日 下午2:15:39
 * @Since: JDK1.8
 * @Version: 1.0
 */
class SheetImport {
    private Class<?> claz;
    private LinkedHashMap<String, String> alias;
    private Integer param;
 
    public Class<?> getClaz() {
        return claz;
    }
 
    public void setClaz(Class<?> claz) {
        this.claz = claz;
    }
 
    public LinkedHashMap<String, String> getAlias() {
        return alias;
    }
 
    public void setAlias(LinkedHashMap<String, String> alias) {
        this.alias = alias;
    }
 
    public Integer getParam() {
        return param;
    }
 
    public void setParam(Integer param) {
        this.param = param;
    }
}

测试类

代码语言:javascript
复制
package com.clf.equity.manage.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;

import org.apache.commons.beanutils.PropertyUtils;

public class ExcelUtilTest {
	static List<TestUser> pojoList = null;
	static List<TestUser> testUserList = null;

	static List<TestUserSS> testUserSSList = null;

	// excel =》实体类
	// 一个sheet
	public static void test1() throws Exception {
		// 指定输入文件
		FileInputStream fis = new FileInputStream("C:\\data\\test2.xlsx");
		// 指定每列对应的类属性
		LinkedHashMap<String, String> alias = new LinkedHashMap<>();
		alias.put("姓名", "name");
		alias.put("年龄", "age");
		// 转换成指定类型的对象数组
		pojoList = ExcelUtil.excel2Pojo(fis, TestUser.class, alias, 0);

	}

	// excel =》实体类
	// 导入 多个sheet
	public static void test2() throws Exception {
		// 指定输入文件
		FileInputStream fis = new FileInputStream("C:\\data\\test2.xlsx");

		List<SheetImport> listImport = new ArrayList<>();
		// 指定每列对应的类属性
		LinkedHashMap<String, String> alias = new LinkedHashMap<>();
		alias.put("姓名", "name");
		alias.put("年龄", "age");
		SheetImport sheetImport = new SheetImport();
		sheetImport.setAlias(alias);
		sheetImport.setClaz(TestUser.class);
		sheetImport.setParam(0);
		listImport.add(sheetImport);
		LinkedHashMap<String, String> alias2 = new LinkedHashMap<>();
		alias2.put("姓名", "name");
		alias2.put("年龄", "age");
		alias2.put("日期", "dat");
		SheetImport sheetImport2 = new SheetImport();
		sheetImport2.setAlias(alias2);
		sheetImport2.setClaz(TestUserSS.class);
		sheetImport2.setParam(0);
		listImport.add(sheetImport2);
		// 转换成指定类型的对象数组
		List<?> pojoList = ExcelUtil.excel2PojoSheetList(listImport, fis);
		testUserList = new ArrayList<>();
		for (TestUser te : (List<TestUser>) pojoList.get(0)) {
			TestUser testUser = new TestUser();
			PropertyUtils.copyProperties(testUser, te);
			testUserList.add(testUser);
		}

		testUserSSList = new ArrayList<>();
		for (TestUserSS te : (List<TestUserSS>) pojoList.get(1)) {
			TestUserSS testUseSS = new TestUserSS();
			PropertyUtils.copyProperties(testUseSS, te);
			testUserSSList.add(testUseSS);
		}

	}

	// 实体类 =》excel
	// 一个sheet
	public static void test3() throws Exception {
		// 将生成的excel转换成文件,还可以用作文件下载
		File file = new File("C:\\data\\testExport.xlsx");
		FileOutputStream fos = new FileOutputStream(file);

		// 对象集合
		List<TestUser> pojoExport = new ArrayList<>();
		for (int i = 0; i < 5; i++) {
			TestUser user = new TestUser();
			user.setName("老李");
			user.setAge(50);
			pojoExport.add(user);
		}
		// 设置属性别名(列名)
		LinkedHashMap<String, String> aliasE = new LinkedHashMap<>();
		aliasE.put("name", "姓名");
		aliasE.put("age", "年龄");
		// 标题
		String headLine = "用户表";
		List<MergeData> list = new ArrayList<>();
		list.add(new MergeData("start", 1, 2, 0, 2));
		// list.add(new MergeData("end", 1, 2, 2, 3));
		ExcelUtil.pojo2Excel(pojoList, fos, aliasE, new UtilExcel("table", 4), list);
		// ExcelUtil2.pojo2Excel(pojoList, fos, aliasE, new
		// UtilExcel("table",2));

	}

	// 实体类 =》excel
	// 多个sheet
	public static void test4() throws Exception {
		// 将生成的excel转换成文件,还可以用作文件下载
		File file = new File("C:\\data\\testExport2.xlsx");
		FileOutputStream fos = new FileOutputStream(file);

		List<SheetExport> sheetExport = new ArrayList<>();

		SheetExport us = new SheetExport();
		LinkedHashMap<String, String> aliaEp = new LinkedHashMap<>();
		aliaEp.put("name", "姓名");
		aliaEp.put("age", "年龄");
		us.setAlias(aliaEp);
		us.setSheetName("第一");
		us.setPojoList(testUserList);
		us.setUtilExcel(new UtilExcel("table", 1));
		sheetExport.add(us);

		SheetExport usS = new SheetExport();
		LinkedHashMap<String, String> aliasEx = new LinkedHashMap<>();
		aliasEx.put("name", "姓名");
		aliasEx.put("age", "年龄");
		aliasEx.put("dat", "日期");
		usS.setAlias(aliasEx);
		usS.setSheetName("sheet name");
		usS.setPojoList(testUserSSList);
		List<MergeData> lisor = new ArrayList<>();
		lisor.add(new MergeData("start", 1, 2, 0, 2));
		usS.setMergeDataList(lisor);
		usS.setUtilExcel(new UtilExcel("table", 1));
		sheetExport.add(usS);

		ExcelUtil.pojo2ExcelSheetList(sheetExport, fos);

	}

	public static void main(String[] args) throws Exception {
		test1();
		test2();
		test3();
		test4();
	}

}

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/160453.html原文链接:https://javaforall.cn

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档