EasyExcel实现多列导出

至死孤独

已于 2023-06-21 14:30:36 修改

阅读量519

点赞数 1

文章标签： excel java

于 2023-06-19 10:34:08 首次发布

本文链接：https://blog.csdn.net/u013013708/article/details/131282503

版权

能够实现动态列导出

public static byte[] exportExcelFile(LinkedHashMap<String, String> headColumnMap, List<Map<String, Object>> dataList, ){
    //获取列名称
    List<List<String>> excelHead = new ArrayList<>();
    if(MapUtils.isNotEmpty(headColumnMap)){
        //key为匹配符，value为列名，如果多级列名用逗号隔开
        headColumnMap.entrySet().forEach(entry -> {
            excelHead.add(Lists.newArrayList(entry.getValue().split(",")));
        });
    }
    List<List<Object>> excelRows = new ArrayList<>();
    if(MapUtils.isNotEmpty(headColumnMap) && CollectionUtils.isNotEmpty(dataList)){
        for (Map<String, Object> dataMap : dataList) {
            List<Object> rows = new ArrayList<>();
            headColumnMap.entrySet().forEach(headColumnEntry -> {
                if(dataMap.containsKey(headColumnEntry.getKey())){
                    Object data = dataMap.get(headColumnEntry.getKey());
                    rows.add(data);
                }
            });
            excelRows.add(rows);
        }
    }
    byte[] stream = createExcelFile(excelHead, excelRows, ExcelTypeEnum.XLS);
    return stream;
}

    private static byte[] createExcelFile(List<List<String>> excelHead, List<List<Object>> excelRows, ExcelTypeEnum type){
        try {
            if(CollectionUtils.isNotEmpty(excelHead)){
                ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
                EasyExcel.write(outputStream).excelType(type).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
                        .head(excelHead)
                        .sheet(DEFAULT_SHEET_NAME)
                        .doWrite(excelRows);
                return outputStream.toByteArray();
            }
        } catch (Exception e) {
            System.err.print("动态生成excel文件失败，headColumns：" + JSONArray.toJSONString(excelHead) + "，excelRows：" + JSONArray.toJSONString(excelRows));
        }
        return null;
    }