[开发测试]JSON数组动态转EXCEL下载

JSON数组动态转EXCEL下载

把json数组动态的转换成excel下载下来

    public static void jsonToXSSF(String str, HttpServletResponse response) throws IOException {
        response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("mongo.xlsx", "UTF-8"));
        try (XSSFWorkbook wb = new XSSFWorkbook();
             OutputStream output = response.getOutputStream()) {
            // 创建HSSFSheet对象
            XSSFSheet sheet = wb.createSheet("mongo");
            // 记录标题
            Map<String, Integer> titles = new HashMap<>();
            // 记录行号 0 是标题 数据从 1 开始
            int rowNo = 1;
            // 记录最大的列号 用于添加筛选器时的参数 也可以通过 titles 的长度获取
            int cellMax = 0;
            // 记录起始列号
            int celltitleNo = 0;
            // 把json数组字符串转成json数组对象
            JSONArray jsonArray = JSONArray.parseArray(str);
            
            // 创建标题行
            XSSFRow rowtitle = sheet.createRow(0);
            
            // 循环处理数据
            for (Object item : jsonArray) {
                // 获取每行数据
                JSONObject jsonObject = (JSONObject) item;
                
                // 获取标题
                Set<String> keys = jsonObject.keySet();
                
                // 创建数据行
                XSSFRow row = sheet.createRow(rowNo++);
                
                // 获取最大列号
                cellMax = Math.max((keys.size() - 1), cellMax);
                
                // 循环处理列数据
                for (String key : keys) {
                    // 如果标题不存在，就添加一列标题
                    if (!titles.containsKey(key)) {
                        //  添加标题 和列号
                        titles.put(key, celltitleNo++);
                        
                        // 创建标题样试
                        XSSFCellStyle cellStyle = wb.createCellStyle();
                        cellStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
                        cellStyle.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex());
                        
                        // 添加标题
                        XSSFCell celltitle = rowtitle.createCell(titles.get(key));
                        celltitle.setCellStyle(cellStyle);
                        celltitle.setCellValue(key);
                    }
                    
                    // 添加列数据
                    XSSFCell cell = row.createCell(titles.get(key));
                    
                    // 转换类型
                    if (jsonObject.get(key) instanceof Double) {
                        cell.setCellValue(jsonObject.getDouble(key));
                    } else if (jsonObject.get(key) instanceof Integer) {
                        cell.setCellValue(jsonObject.getInteger(key));
                    } else if (jsonObject.get(key) instanceof Date) {
                        cell.setCellValue(jsonObject.getDate(key));
                    } else if (jsonObject.get(key) instanceof Boolean) {
                        cell.setCellValue(jsonObject.getBoolean(key));
                    } else {
                        cell.setCellValue(jsonObject.getString(key));
                    }
                }
            }
            // 输出Excel文件
            // 添加筛选器  参数： 起始行  结束行  起始列  结束列
            sheet.setAutoFilter(new CellRangeAddress(0, rowNo, 0, cellMax));
            wb.write(output);
        }
    }