JAVA导出EXCEL

JAVA导出EXCEL

1.MAVEN依赖

   <!--导出导入excel-->
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>4.1.2</version> <!-- 根据你的需求选择版本 -->
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>4.1.2</version> <!-- 根据你的需求选择版本 -->
        </dependency>

2.代码实现

  @RequestMapping
    public Object export(QdUserBean qdUserBean) {
        Assert.notNull(qdUserBean.getCodes(), "用户codes不能为空");

        List<QdUserBean> qdUserBeans = qdUserService.findAll(QdUserBean.builder().codes(qdUserBean.getCodes()).build());


        // 创建一个新的工作簿
        Workbook workbook = new XSSFWorkbook();

        //导出用户信息
        if (CollectionUtils.isNotEmpty(qdUserBeans)) {
            excelUser(qdUserBeans, workbook);
        }

        String excelUrl = null;

        try {
            // 生成随机文件名
            String name = UUID.randomUUID().toString().replace("-", "") + ".xlsx";
            //String filePath = "nas_server_data/EXCEL/" + name;
            String filePath = name;
            logger.info("文件名：" + name);

            // 保存 Excel 文件
            FileOutputStream outputStream = new FileOutputStream(filePath);
            workbook.write(outputStream);
            outputStream.close();

            // 创建输入流，读取文件内容并计算字节数
            File inputFile = new File(filePath);
            FileInputStream inputStream = new FileInputStream(inputFile);
            byte[] fileBytes = IOUtils.toByteArray(inputStream);
            long fileSize = inputFile.length();
            inputStream.close();

            // 创建 NasFileBean
            NasFileBean nasFileBean = new NasFileBean();
            nasFileBean.setData(fileBytes);
            nasFileBean.setSize(fileSize);
            nasFileBean.setType("xlsx");
            nasFileBean.setName(name);
            nasFileBean.setCategoryFlag("COMMON");

            // 上传文件并获取返回的 NasFileBean
            nasFileBean = this.fileRecordService.uploadFile(nasFileBean);

            // 关闭工作簿
            workbook.close();

            if (nasFileBean != null) {
                excelUrl = fileHost + nasFileBean.getCode();
            } else {
                return ResponseBean.serverError("操作失败！");
            }
        } catch (IOException e) {
            e.printStackTrace();
            return ResponseBean.serverError("文件处理失败！");
        } catch (Exception e) {
            e.printStackTrace();
            return ResponseBean.serverError(e.getMessage());
        }

        return ResponseBean.success("操作成功！").addData("url", excelUrl);
    }

    /**
     * 导出项目信息
     *
     * @return
     */
    public static void excelUser(List<QdUserBean> qdUserBeans, Workbook workbook) {
        // 创建一个工作表
        Sheet sheet = workbook.createSheet("用户信息");
        // 创建表头行
        Row headerRow = sheet.createRow(0);
        // 使用配置数据生成表头
        int columnIndex = 0;

        //遍历表头，一列一列渲染数据
        //项目参数表头
        List<String> head = new ArrayList<>();
        head.add("ID");
        head.add("昵称");
        head.add("头像");
        head.add("手机号码");
        head.add("最近登录时间");
        head.add("登录IP");
        head.add("所在城市");
        head.add("注册时间");
        head.add("用户状态");
        for (String name : head) {
            int columnWidth = 5000;

            // 创建单元格
            Cell cell = headerRow.createCell(columnIndex);
            cell.setCellValue(name);
            // 设置列宽
            sheet.setColumnWidth(columnIndex, columnWidth);

            columnIndex++;
        }

        int rowIndex = 1;
        for (QdUserBean qdUserBean : qdUserBeans) {
            // 保存项目数据到excel
            Row dataRow = sheet.createRow(rowIndex);

            // 遍历表头行的每个单元格，获取表头信息
            int totalColumns = headerRow.getLastCellNum(); // 获取总列数
            for (int colIndex = 0; colIndex < totalColumns; colIndex++) {
                Cell cell = headerRow.getCell(colIndex);
                String columnName = cell.getStringCellValue();

                // 将属性值写入单元格
                Cell dataCell = dataRow.createCell(colIndex);

                try {
                    Object value = null;
                    Field field = null;
                    if (columnName.equals("ID")) {
                        field = qdUserBean.getClass().getDeclaredField("userId");
                    }
                    if (columnName.equals("昵称")) {
                        field = qdUserBean.getClass().getDeclaredField("nickname");
                    }
                    if (columnName.equals("头像")) {
                        field = qdUserBean.getClass().getDeclaredField("portrait");
                    }
                    if (columnName.equals("手机号码")) {
                        field = qdUserBean.getClass().getDeclaredField("phone");
                    }
                    if (columnName.equals("最近登录时间")) {
                        field = qdUserBean.getClass().getDeclaredField("lastLoginTime");
                    }
                    if (columnName.equals("登录IP")) {
                        field = qdUserBean.getClass().getDeclaredField("ip");
                    }
                    if (columnName.equals("所在城市")) {
                        field = qdUserBean.getClass().getDeclaredField("address");
                    }
                    if (columnName.equals("注册时间")) {
                        field = qdUserBean.getClass().getDeclaredField("createTime");
                    }
                    if (columnName.equals("用户状态")) {
                        field = qdUserBean.getClass().getDeclaredField("status");
                    }

                    field.setAccessible(true); // 设置属性为可访问
                    value = field.get(qdUserBean);

                    if (value != null) {
                        if (value instanceof String) {
                            if (value.equals("normal")) {
                                value = "正常";
                            }
                            if (value.equals("freeze")) {
                                value = "冻结";
                            }
                            dataCell.setCellValue((String) value);
                        } else if (value instanceof Number) {
                            dataCell.setCellValue(((Number) value).doubleValue());
                        } else if (value instanceof Date) {
                            dataCell.setCellValue((Date) value);
                        }
                    }
                } catch (NoSuchFieldException | IllegalAccessException e) {
                    e.printStackTrace();
                }
            }

            rowIndex++;
        }
    }