一个简单易用的Excel导出工具

import org.apache.commons.collections.CollectionUtils;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.usermodel.CellStyle;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.List;

/**
 * @author bilibli
 * @Title: excel
 * @Package com.ht.ams.car.util
 * @Description: excel
 * @date 2020年5月29日 09点25分
 */
public class ExcelUtil {
    /**
     * @param - [lists 数据集, fieldMap (字段,中文), response, fileName 文件名]
     * @return com.ht.ussp.core.Result
     * @Description: 导出excel
     * @author 向舜捷
     * @date 2020年5月29日 09点25分
     */
    public static Result downExcelForList(List<?> lists, LinkedHashMap<String, String> fieldMap, HttpServletResponse response, String fileName) {
        OutputStream os = null;
        try {
            setHeader(response, fileName);
            os = response.getOutputStream();
            listToExcel(lists, fieldMap, fileName).write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
            return Result.fail(502, "导出失败");
        } finally {
            if (null != os) {
                try {
                    os.flush();
                    os.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }

            }
        }
        return Result.success();
    }

    private static void setHeader(HttpServletResponse response, String fileName) throws Exception {
        response.setContentType("application/vnd.ms-excel");
        String codedFileName = java.net.URLEncoder.encode(fileName , "UTF-8");
        response.setCharacterEncoding("utf-8");
        response.setHeader("content-disposition", "attachment;filename=" + codedFileName+new SimpleDateFormat("yyyyMMddHHmmss ").format(new Date()) + ".xls");
    }

    private static void setCellHead(HSSFSheet sheet, LinkedHashMap<String, String> fieldMap, HSSFCellStyle headStyle) {

        HSSFRow row1 = sheet.createRow(0);
        //设置行高
        row1.setHeight((short) 500);
        int i = 0;
        for (String colHead : fieldMap.values()) {
            HSSFCell cell = row1.createCell(i);
            cell.setCellValue(colHead);
            cell.setCellStyle(headStyle);
            i++;
        }
    }

    private static void setCell(HSSFWorkbook wb
            , String fileName
            , LinkedHashMap<String, String> fieldMap
            , List<?> lists
            , HSSFCellStyle headStyle
            , HSSFCellStyle bodyStyle
    ) throws Exception {
        HSSFSheet sheet = null;
        int j = 0;
        if (lists == null || lists.size() == 0) {
            sheet = wb.createSheet(fileName);
            setCellHead(sheet, fieldMap, headStyle);
            for (int k = 0; k < fieldMap.keySet().size(); k++) {
                sheet.setColumnWidth(k, 20 * 256);
            }
        }else {
            for (int i = 0; i < lists.size(); i++) {
                if (i == 0) {
                    sheet = wb.createSheet(fileName);
                    setCellHead(sheet, fieldMap, headStyle);
                } else if (i % 30000 == 0) {
                    j++;
                    sheet = wb.createSheet(fileName + j);
                    setCellHead(sheet, fieldMap, headStyle);
                }
                for (int k = 0; k < fieldMap.keySet().size(); k++) {
                    sheet.setColumnWidth(k, 20 * 256);
                }
                Object item = lists.get(i);
                HSSFRow row = sheet.createRow(i + 1);
                //设置行高
                row.setHeight((short) 500);
                int x = 0;
                Class itemClazz = lists.get(0).getClass();
                for (String colBody : fieldMap.keySet()) {
                    Field field = getFieldByName(colBody, itemClazz);
                    field.setAccessible(true);
                    HSSFCell c = row.createCell(x);
                    if(field.get(item) instanceof  Date){
                        c.setCellValue(field.get(item)!=null?new SimpleDateFormat("yyyy-MM-dd HH:mm").format( field.get(item)):"");
                    }else if(field.get(item) == null){
                        c.setCellValue(" ");
                    }
                    else{
                        c.setCellValue(String.valueOf(field.get(item)));
                    }
                    c.setCellStyle(bodyStyle);
                    x++;
                }
            }
        }
    }

    private static Field getFieldByName(String fieldName, Class<?> clazz) {
        Field[] selfFields = clazz.getDeclaredFields();
        for (Field field : selfFields) {
            if (field.getName().equals(fieldName)) {
                return field;
            }
        }

        Class<?> superClazz = clazz.getSuperclass();
        if (superClazz != null && superClazz != Object.class) {
            return getFieldByName(fieldName, superClazz);
        }

        //如果本类和父类都没有,则返回空
        return null;
    }

    private static HSSFWorkbook listToExcel(List<?> lists,
                                            LinkedHashMap<String, String> fieldMap,
                                            String fileName) throws Exception {
        // 创建HSSFWorkbook对象(excel的文档对象)
        HSSFWorkbook hssfWorkbook = new HSSFWorkbook();
        HSSFCellStyle headStyle = getHeadStyle(hssfWorkbook);
        HSSFCellStyle bodyStyle = getBodyStyle(hssfWorkbook);
        setCell(hssfWorkbook, fileName, fieldMap, lists, headStyle, bodyStyle);
        return hssfWorkbook;
    }

    private static HSSFCellStyle getHeadStyle(HSSFWorkbook hssfWorkbook) {
        HSSFCellStyle headStyle = hssfWorkbook.createCellStyle();
        headStyle.setFillForegroundColor(HSSFColor.PALE_BLUE.index);
        headStyle.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
        headStyle.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        headStyle.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        headStyle.setBorderRight(HSSFCellStyle.BORDER_THIN);
        headStyle.setBorderTop(HSSFCellStyle.BORDER_THIN);
        headStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        // 纵向居中
        headStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);

        HSSFFont font = hssfWorkbook.createFont();
        font.setFontName("黑体");
        //设置字体大小
        font.setFontHeightInPoints((short) 12);
        headStyle.setFont(font);
        return headStyle;
    }

    private static HSSFCellStyle getBodyStyle(HSSFWorkbook hssfWorkbook) {
        HSSFCellStyle bodyStyle = hssfWorkbook.createCellStyle();

        bodyStyle.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        bodyStyle.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        bodyStyle.setBorderRight(HSSFCellStyle.BORDER_THIN);
        bodyStyle.setBorderTop(HSSFCellStyle.BORDER_THIN);
        bodyStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        // 纵向居中
        bodyStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);

        HSSFFont font = hssfWorkbook.createFont();
        font.setFontName("黑体");
        bodyStyle.setFont(font);
        return bodyStyle;
    }


    /**
     * 传入一个导出的数据源类型,该方法会帮助你拿到  @ApiModelProperty 里面的Value 和 绑定一起的 attributeName
     * 并以Map的形式返回
     * @param apiModelClass         要拿到 @ApiModelProperty value 的类型
     * @param <T>                   改类 的 T
     * @return                      返回一个 Map
     */
    public static <T> LinkedHashMap<String, String> getFieldsMap(Class<T> apiModelClass){
        List<ApiModelPropertyFixedTo> fixedVoList = AnnotationTool.getFixedVoList(apiModelClass);
        LinkedHashMap<String, String> linkedHashMap = new LinkedHashMap<>();
        if(CollectionUtils.isEmpty(fixedVoList)) {
            return linkedHashMap;
        }
        fixedVoList.forEach((item)-> {
            // 隐藏的部分不加载
//            if (!item.isHidden()){
//                linkedHashMap.put(item.getAttributeName(),item.getValue());
//            }
            linkedHashMap.put(item.getAttributeName(),item.getValue());
        });
        return linkedHashMap;
    }


}

使用:

 List<CustomerExtendPo> list = ucCustomerService.getCustomerExtendPoByWrapperList(this.getWrapper(customerReqVo));
        list.forEach(x -> {
            x.setSex("未知");
            if ("M".equals(x.getSex())) {
                x.setSex("男");
            } else {
                x.setSex("女");
            }
        });
        LinkedHashMap<String, String> fieldsMap = new LinkedHashMap<>(9);
        fieldsMap.put("nickname", "用户名");
        fieldsMap.put("sex", "性别");
        fieldsMap.put("createTime", "注册时间");
        fieldsMap.put("tell", "手机号");
        fieldsMap.put("lastLoginTime", "最后登录时间");
        fieldsMap.put("lastOrderTime", "最后下单时间");
        fieldsMap.put("firstOrderTime", "首次下单时间");
        ExcelUtil.downExcelForList(list, fieldsMap, response, "客户信息");
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

科学熊

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值