【Java】读取EXCEL生成实体类，自带swagger文档配置，复制即可用！

大白不会笑

已于 2023-12-27 15:53:50 修改

阅读量701

点赞数 10

CC 4.0 BY-SA版权

文章标签： java excel 开发语言

于 2023-12-19 16:11:41 首次发布

本文链接：https://blog.youkuaiyun.com/qq_42980844/article/details/135087003

本文介绍了一种方法，利用ApachePOI库读取Excel文件中的数据，自动生成Java实体类，并为Swagger提供注释，简化在内网环境中处理大量字段的工作流程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

背景

数据组在zf内网环境推送了很多表、数据，不能直接连数据库进行传统的代码生成，而业务又没啥逻辑，工作耗时主要在建实体类，动不动100多个字段，所以有了下面的代码，还做了swagger文档配置，复制之后即可使用。

废话不多说，直接上代码


import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.*;
import java.util.HashMap;

/**
 * @author: DaBaiBuHuiXiao
 * @Date: 2023/12/19 10:02
 * 读取excel生成java实体类
 */

public class DomainWork {

    public static void main(String[] args) throws IOException {
        // excel访问路径
        // 所传入文件格式为：第一列字段名，第二列字段中文注释（用于swagger文档）
        String excelFilePath = "C:\\Users\\78596\\Desktop\\beanTest.xlsx";
        // 开流
        FileInputStream fis = new FileInputStream(excelFilePath);
        Workbook workbook = new XSSFWorkbook(fis);
        int headerRowNum = workbook.getSheetAt(0).getLastRowNum();
        HashMap<String, String> hashMap = new HashMap<>();
        for (int i = 0; i <= headerRowNum; i++) {
            // 读取第一个sheet页
            Row row = workbook.getSheetAt(0).getRow(i);
            // 读取第一列为 实体字段名，第二列为swagger注释名
            hashMap.put(row.getCell(0).getStringCellValue(),row.getCell(1).getStringCellValue());
        }
        workbook.close();
        fis.close();
        // 生成java实体类
        final String beanName = "Bean";
        // 代码生成地址
        File file = new File("D:\\"+ beanName +".java");

        try (
            FileWriter fw = new FileWriter(file);
            PrintWriter pw = new PrintWriter(fw)) {
            pw.println("import io.swagger.annotations.ApiModelProperty;");
            pw.println("import lombok.Getter;");
            pw.println("import lombok.Setter;");
            pw.println("import lombok.NoArgsConstructor;");
            pw.println();

            pw.println("@Getter");
            pw.println("@Setter");
            pw.println("@NoArgsConstructor");
            pw.println("public class " + beanName +"{");
            for (String key:hashMap.keySet()
            ) {
                pw.println();
                pw.println("  @ApiModelProperty("+"\""+hashMap.get(key)+"\""+")");
                pw.println("  private String "+key+";");
            }
			pw.println();
            pw.println("}");
        }catch (Exception e){
            e.printStackTrace();
        }
    }

}