java 反射实现excel导出通用工具类

废话不多说,直接上代码吧,使用的方法在最下面

此导出工具类借鉴了一些大神的写法以及结合自己业务上的需求



import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.*;

import java.awt.*;
import java.lang.reflect.Method;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.Map;

public class ExcelUtil {


    /**
     * @param title:sheet名称
     * @param propertyHeaderMap:<property, header>(<T中的property名称、有getter就行, 对应显示在Excel sheet中的列标题>)
     *                                     用LinkedHashMap保证读取的顺序和put的顺序一样
     * @param dataSet:实体类集合
     * @description: 适用于单表导出
     * @author: xch
     * @date: 2018/5/23 9:38
     * 根据输入的数据生成一个XSSFWorkbook
     * @return:XSSFWorkbook
     */
    public static <T> XSSFWorkbook generateXlsxWorkbook(String title, LinkedHashMap<String, String> propertyHeaderMap, Collection<T> dataSet) {
        // 声明一个工作薄
        XSSFWorkbook workbook = new XSSFWorkbook();
        // 生成一个表格
        XSSFSheet sheet = workbook.createSheet(title);

        // 设置表格默认列宽度为20个字节
        sheet.setDefaultColumnWidth((int) 20);

        XSSFCellStyle headerStyle = getHeaderStyle(workbook);
        XSSFCellStyle contentStyle = getContentStyle(workbook);

        // 生成表格标题行
        XSSFRow row = sheet.createRow(0);
        int i = 0;
        //传过来的map设置表头
        for (String key: propertyHeaderMap.keySet()) {
            XSSFCell cell = row.createCell(i);
            cell.setCellStyle(headerStyle);
            XSSFRichTextString text = new XSSFRichTextString(propertyHeaderMap.get(key));
            cell.setCellValue(text);
            i++;
        }

        //循环dataSet,每一条对应一行
        int index = 0;
        for (T data: dataSet) {
            index++;
            row = sheet.createRow(index);

            int j = 0;
            for (String property: propertyHeaderMap.keySet()) {
                XSSFCell cell = row.createCell(j);
                cell.setCellStyle(contentStyle);

                //拼装getter方法名              //toUpperCase()英文字符转换为大写字母,例如P+kId;完成拼出:getPkId
                String getMethodName = "get" + property.substring(0, 1).toUpperCase() + property.substring(1);

                try {
                    //利用反射机制获取dataSet中的属性值,填进cell中
                    Class<? extends Object> tCls = data.getClass();
                    Method getMethod = tCls.getMethod(getMethodName, new Class[]{});
                    Object value = getMethod.invoke(data, new Object[]{}); //调用getter从data中获取数据

                    // 判断值的类型后进行类型转换
                    String textValue = null;

                    if (value != null) {
                        if (value instanceof Double) {
                            //double类型如果小数点后为零显示整数,否则保留
                            if ((Double) value % 1.0 == 0) {
                                textValue = String.valueOf(((Double) value).longValue());
                            } else {
                                textValue = value.toString();
                            }
                        } else if (value instanceof Date) {
                            Date date = (Date) value;
                            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
                            textValue = sdf.format(date);
                        } else {
                            // 其它数据类型都当作字符串简单处理
                            textValue = value.toString();
                        }
                    } else {
                        textValue = "";
                    }
                    XSSFRichTextString richString = new XSSFRichTextString(textValue);
                    cell.setCellValue(richString);

                } catch (Exception e) {
                    e.printStackTrace();
                }
                j++;
            }
        }

        return workbook;

    }


   
    public static <T> XSSFWorkbook generateXlsxWorkbook(String title, LinkedHashMap<String, String> propertyHeaderMap, Collection<T> dataSet, Map<String, Object[]> map) {
        // 声明一个工作薄
        XSSFWorkbook workbook = new XSSFWorkbook();
        // 生成一个表格
        XSSFSheet sheet = workbook.createSheet(title);

        // 设置表格默认列宽度为20个字节
        sheet.setDefaultColumnWidth((int) 20);

        XSSFCellStyle headerStyle = getHeaderStyle(workbook);
        XSSFCellStyle contentStyle = getContentStyle(workbook);

        // 生成表格标题行
        XSSFRow row = sheet.createRow(0);
        int i = 0;
        //传过来的map设置表头
        for (String key: propertyHeaderMap.keySet()) {
            XSSFCell cell = row.createCell(i);
            cell.setCellStyle(headerStyle);
            XSSFRichTextString text = new XSSFRichTextString(propertyHeaderMap.get(key));
            cell.setCellValue(text);
            i++;
        }

        //循环dataSet,每一条对应一行
        int index = 0;
        for (T data: dataSet) {
            index++;
            row = sheet.createRow(index);

            int j = 0;
            for (String property: propertyHeaderMap.keySet()) {
                XSSFCell cell = row.createCell(j);
                cell.setCellStyle(contentStyle);

                //拼装getter方法名              //toUpperCase()英文字符转换为大写字母,例如P+kId
                String getMethodName = "get" + property.substring(0, 1).toUpperCase() + property.substring(1);

                try {
                    //利用反射机制获取dataSet中的属性值,填进cell中
                    Class<? extends Object> tCls = data.getClass();
                    Method getMethod = tCls.getMethod(getMethodName, new Class[]{});
                    Object value = getMethod.invoke(data, new Object[]{}); //调用getter从data中获取数据


                    int a = 0;
                    int b = 0;
                    if (map != null) {
                        if (map.get(property) != null) {
                            a = Integer.parseInt(value.toString());
                            b = map.get(property).length;
                            for (int n = 0; n < b; n++) {
                                value = map.get(property)[a];
                            }
                        }
                    }
                    // 判断值的类型后进行类型转换
                    String textValue = null;
                    if (value != null) {
                        if (value instanceof Double) {
                            //double类型如果小数点后为零显示整数,否则保留
                            if ((Double) value % 1.0 == 0) {
                                textValue = String.valueOf(((Double) value).longValue());
                            } else {
                                textValue = value.toString();
                            }
                        } else if (value instanceof Date) {
                            Date date = (Date) value;
                            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
                            textValue = sdf.format(date);
                        } else {
                            // 其它数据类型都当作字符串简单处理
                            textValue = value.toString();
                        }
                    } else {
                        textValue = "";
                    }
                    XSSFRichTextString richString = new XSSFRichTextString(textValue);
                    cell.setCellValue(richString);

                } catch (Exception e) {
                    e.printStackTrace();
                }
                j++;
            }
        }

        return workbook;

    }


    /**
     * @param workbook
     * @return style
     * @description: 生成一个标题style
     * @author: xch
     * @date: 2018/5/23 9:39
     */
    public static XSSFCellStyle getHeaderStyle(XSSFWorkbook workbook) {
        return getHeaderStyle(workbook, Color.WHITE, IndexedColors.BLACK.getIndex());
    }

    /**
     * @param workbook
     * @param foregroundColor
     * @param fontColor
     * @description: 生成一个指定颜色的标题style
     * @author: xch
     * @date: 2018/5/23 9:39
     */
    public static XSSFCellStyle getHeaderStyle(Workbook workbook, Color foregroundColor, short fontColor) {

        // 生成一个样式(用于标题)
        XSSFCellStyle style = (XSSFCellStyle) workbook.createCellStyle();
        // 设置这些样式
        style.setFillForegroundColor(new XSSFColor(foregroundColor));
        style.setFillPattern(XSSFCellStyle.SOLID_FOREGROUND);
        style.setBorderBottom(XSSFCellStyle.BORDER_THIN);
        style.setBorderLeft(XSSFCellStyle.BORDER_THIN);
        style.setBorderRight(XSSFCellStyle.BORDER_THIN);
        style.setBorderTop(XSSFCellStyle.BORDER_THIN);
        style.setAlignment(XSSFCellStyle.ALIGN_CENTER);
        // 生成一个字体
        XSSFFont font = (XSSFFont) workbook.createFont();
        font.setColor(fontColor);
        font.setFontHeightInPoints((short) 12);
        font.setBoldweight(XSSFFont.BOLDWEIGHT_BOLD);
        // 把字体应用到当前的样式
        style.setFont(font);

        return style;
    }


    /**
     * @param workbook
     * @return
     * @description: 生成一个用于内容的style
     * @author: xch
     * @date: 2018/5/23 9:39
     */
    public static XSSFCellStyle getContentStyle(XSSFWorkbook workbook) {
        // 生成并设置另一个样式(用于内容)
        XSSFCellStyle style = (XSSFCellStyle) workbook.createCellStyle();
        //style.setFillForegroundColor(new XSSFColor(Color.YELLOW));
        //style.setFillPattern(XSSFCellStyle.SOLID_FOREGROUND);
        style.setBorderBottom(XSSFCellStyle.BORDER_THIN);
        style.setBorderLeft(XSSFCellStyle.BORDER_THIN);
        style.setBorderRight(XSSFCellStyle.BORDER_THIN);
        style.setBorderTop(XSSFCellStyle.BORDER_THIN);
        style.setAlignment(XSSFCellStyle.ALIGN_CENTER);
        style.setVerticalAlignment(XSSFCellStyle.VERTICAL_CENTER);
        // 生成另一个字体
        XSSFFont font = (XSSFFont) workbook.createFont();
        font.setBoldweight(XSSFFont.BOLDWEIGHT_NORMAL);
        // 把字体应用到当前的样式
        style.setFont(font);

        return style;
    }
}

如何使用

方法1:generateXlsxWorkbook(String title, LinkedHashMap<String, String> propertyHeaderMap, Collection dataSet)

1、第一个参数为excel的title名字
2、第二个参数选择LinkedHashMap的原因在于内部维持了一个双向链表,可以保持顺序;map的键 必须和实体类的属性名称一致
值为导出excel的表头名称
3、第三个参数为导出数据的集合

 Map<String, Object> temp = new HashMap();
        LinkedHashMap<String, String> map = new LinkedHashMap<>();
        map.put("folderName", "订单号");
        map.put("customerName", "客户名");
        map.put("housingAddress", "客户地址");
        map.put("fileNumber", "文件数");
        map.put("createName", "创建人");
        map.put("createTime", "创建时间");

        XSSFWorkbook ex = ExcelUtil.generateXlsxWorkbook("文档中心", map, fileFolderList);
方法2:generateXlsxWorkbook(String title, LinkedHashMap<String, String> propertyHeaderMap, Collection dataSet, Map<String, Object[]> map)

一些特别的业务有很多种状态,数据库查询返回的都是int型的数字,每一个数字代表不同的状态,这时候就需要第四个参数了:将返回状态的int型数字作为下标来将状态转为需要导出的内容

        Map<String, Object> temp = new HashMap();
        LinkedHashMap<String, String> map = new LinkedHashMap<>();
        map.put("po.orderNumber", "订单编号");
        map.put("name", "销售");
        map.put("area", "门店");
        map.put("customerName", "客户");
        map.put("customerPhone", "客户电话");
        map.put("price", "已收金额");
        map.put("po.pactPrices", "订单金额");
        map.put("po.billType", "状态");
        map.put("po.startTime", "进场时间");
        map.put("po.endTime", "出场时间");
        map.put("po.createTime", "录入时间");
        map.put("po.updateTime", "更新时间");

        String[] billType = {"已取消", "正常", "草稿", "正在等待删除审核", "正在生产审核的"};
        Map<String, Object[]> types = new LinkedHashMap<>();
        types.put("po.billType", billType);
      
            XSSFWorkbook ex = ExcelUtilDTO.generateXlsxWorkbook("取消订单列表", map, podList, types);
            String path = ServletActionContext.getServletContext().getRealPath("/") + "excel";
            String fileName = "取消订单报表" + UUID.randomUUID().toString() + ".xlsx";
            File file = new File(path);
            if (!file.exists()) {
                file.mkdir();
            }
            OutputStream out = new FileOutputStream(file.getPath() + File.separator + fileName);
            BufferedOutputStream bos = new BufferedOutputStream(out);
            ex.write(bos);
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值