数据库表字段以表格形式写入Word

在项目的开发中,难免会有编写概要设计、详细设计文档的要求,而在这些文档中,不可避免的就是要把数据库表的字段信息以表格的形式体现出来。例如下面这种格式

表数量少点还可以一点点粘贴,多了的话真的会疯,所以自己编写了一套可以自动读取数据表字段来批量写入文档的代码。

maven项目需要先引入依赖

        <dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>27.0-hbase</version>
        </dependency>

        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>5.0.0</version> <!-- 请检查是否有更新的版本 -->
        </dependency>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.23</version> <!-- 请检查是否有更新的版本 -->
        </dependency>

        <dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>25.1-jre</version>
        </dependency>

代码如下: 



import avro.shaded.com.google.common.collect.Lists;
import org.apache.poi.xwpf.usermodel.*;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.*;

import java.sql.*;
import java.io.FileOutputStream;
import java.io.IOException;
import java.math.BigInteger;
import java.util.List;

public class DatabaseToWordExample{
    public static void main(String[] args) throws SQLException, IOException {
        // 创建Word文档
        XWPFDocument doc = new XWPFDocument();
        // 创建1级标题
        setDocTitle(doc,1,"一、数据库表设计");

        // 连接数据库
        Connection conn = DriverManager.getConnection("jdbc:mysql://{IP}:{port}/{数据库名}", 用户名, 密码);
        List<String> dbList = Lists.newArrayList("user");
        // 获取元数据
        DatabaseMetaData metaData = conn.getMetaData();
        int dbNum = 1;
        for (String database : dbList) {
            // 2级数据库标题名
            String databaseTitle = new StringBuilder(String.format("%d.%d",1,dbNum))
                    .append(" ")
                    .append(database)
                    .append("库")
                    .toString();
            // 创建2级标题
            setDocTitle(doc,2,databaseTitle);

            int tableNum = 1;
            // 查询表列表
            ResultSet tables = metaData.getTables(database, null, null, null);
            while (tables.next()){
                String tableName = tables.getString("TABLE_NAME");
                // 查询表注释
                String sql = "SELECT table_comment FROM information_schema.tables WHERE table_schema = ? AND table_name = ?";
                PreparedStatement ps = conn.prepareStatement(sql);
                ps.setString(1, database);
                ps.setString(2, tableName);
                ResultSet rs = ps.executeQuery();
                String tableComment = "";
                if (rs.next()) {
                    tableComment = rs.getString("table_comment");
                }
                // 3级表标题名
                String tableTitle = new StringBuilder(String.format("%d.%d.%d",1,dbNum,tableNum))
                        .append(" ")
                        .append(tableName)
                        .append(" ")
                        .append(tableComment)
                        .toString();
                // 创建3级标题
                setDocTitle(doc,3,tableTitle);

                // 创建表格
                XWPFTable table = createTable(doc);
                ResultSet columns = metaData.getColumns(database, null, tableName, null);

                // 添加数据行
                while (columns.next()) {
                    XWPFTableRow dataRow = table.createRow();
                    setRowHeight(dataRow, "490");

                    dataRow.getCell(0).setText(columns.getString("COLUMN_NAME"));
                    dataRow.getCell(1).setText(columns.getString("TYPE_NAME"));
                    dataRow.getCell(2).setText(Integer.toString(columns.getInt("COLUMN_SIZE")));
                    dataRow.getCell(3).setText(columns.getString("REMARKS"));

                    // 设置数据行单元格边框
                    for (XWPFTableCell cell : dataRow.getTableCells()) {
                        setCellBorders(cell);
                    }
                }

                tableNum++;
            }
            dbNum++;
        }

        // 保存Word文档
        try (FileOutputStream out = new FileOutputStream("output.docx")) {
            doc.write(out);
        }

        // 关闭数据库连接
        conn.close();
    }

    /**
     * 设置标题
     *
     * @param doc
     * @param titleLevel 标题等级
     * @param titleName 标题名称
     */
    private static void setDocTitle(XWPFDocument doc, int titleLevel, String titleName){
        switch (titleLevel) {
            case 1:
                // 创建一级标题
                XWPFParagraph title1 = doc.createParagraph();
                XWPFRun run1 = title1.createRun();
                run1.setText(titleName);
                run1.setBold(true);
                run1.setFontSize(16);
                break;
            case 2:
                // 创建二级标题
                XWPFParagraph title2 = doc.createParagraph();
                XWPFRun run2 = title2.createRun();
                run2.setText(titleName);
                run2.setBold(true);
                run2.setFontSize(14);
                break;
            case 3:
                // 创建三级标题
                XWPFParagraph title3 = doc.createParagraph();
                XWPFRun run3 = title3.createRun();
                run3.setText(titleName);
                run3.setBold(true);
                run3.setFontSize(12);
                break;
            default:
                break;
        }
    }

    /**
     * 创建表格
     *
     * @param doc
     * @return
     */
    private static XWPFTable createTable(XWPFDocument doc){
        // 创建表格
        XWPFTable table = doc.createTable();

        // 设置表格宽度
        table.setWidth("9990");

        // 创建表头
        XWPFTableRow headerRow = table.getRow(0);
        setRowHeight(headerRow, "490");  // 设置行高
        headerRow.getCell(0).setText("字段名");
        headerRow.addNewTableCell().setText("字段类型");
        headerRow.addNewTableCell().setText("字段长度");
        headerRow.addNewTableCell().setText("字段备注");

        // 计算每个单元格的宽度
        int totalWidth = 8222;  // 总宽度,例如14.5厘米
        int numColumns = 4;  // 列数,例如4列
        int cellWidth = totalWidth / numColumns;  // 计算每个单元格的宽度

        // 设置表格的列宽
        CTTblWidth tblWidth = table.getCTTbl().addNewTblPr().addNewTblW();
        tblWidth.setType(STTblWidth.DXA);
        tblWidth.setW(BigInteger.valueOf(totalWidth));


        // 设置单元格的宽度
        for (XWPFTableRow row : table.getRows()) {
            for (XWPFTableCell cell : row.getTableCells()) {
                CTTcPr tcPr = cell.getCTTc().addNewTcPr();
                CTTblWidth cellWidth1 = tcPr.addNewTcW();
                cellWidth1.setW(BigInteger.valueOf(cellWidth));
                cellWidth1.setType(STTblWidth.DXA);
            }
        }

        // 设置表头单元格边框
        for (XWPFTableCell cell : headerRow.getTableCells()) {
            setCellBorders(cell);
            cell.setColor("D3D3D3");  // 浅灰色的十六进制颜色代码
            // 设置文字竖向居中
            cell.setVerticalAlignment(XWPFTableCell.XWPFVertAlign.CENTER);
        }
        return table;
    }

    private static void setRowHeight(XWPFTableRow row, String height) {
        row.setHeight(Integer.parseInt(height));
    }


    /**
     * 设置单元格边框和竖向居中
     * @param cell
     */
    private static void setCellBorders(XWPFTableCell cell) {

        // 确保单元格的CTTcPr对象已经被初始化
        if (!cell.getCTTc().isSetTcPr()) {
            cell.getCTTc().addNewTcPr();
        }
        // 设置文字竖向居中
        cell.setVerticalAlignment(XWPFTableCell.XWPFVertAlign.CENTER);

        // 设置单元格上边框
        CTTcBorders borders = cell.getCTTc().getTcPr().addNewTcBorders();
        CTBorder topBorder = borders.addNewTop();
        topBorder.setVal(STBorder.SINGLE);
        topBorder.setSz(BigInteger.valueOf(4));
        topBorder.setColor("000000");

        // 设置单元格下边框
        CTBorder bottomBorder = borders.addNewBottom();
        bottomBorder.setVal(STBorder.SINGLE);
        bottomBorder.setSz(BigInteger.valueOf(4));
        bottomBorder.setColor("000000");

        // 设置单元格左边框
        CTBorder leftBorder = borders.addNewLeft();
        leftBorder.setVal(STBorder.SINGLE);
        leftBorder.setSz(BigInteger.valueOf(4));
        leftBorder.setColor("000000");

        // 设置单元格右边框
        CTBorder rightBorder = borders.addNewRight();
        rightBorder.setVal(STBorder.SINGLE);
        rightBorder.setSz(BigInteger.valueOf(4));
        rightBorder.setColor("000000");
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
生成 Java 实体类,需要获取数据库字段名、字段类型、字段注释等信息,然后根据这些信息生成对应的 Java 实体类。下面是一个简单的 Java 实体类生成器的示例代码,可以通过 JDBC 获取数据库信息,然后生成对应的 Java 实体类: ```java import java.sql.*; import java.util.*; public class EntityGenerator { // 数据库连接信息 private static final String DB_URL = "jdbc:mysql://localhost:3306/test"; private static final String DB_USER = "root"; private static final String DB_PASS = "123456"; // Java 包名 private static final String PACKAGE_NAME = "com.example.entity"; // Java 实体类生成路径 private static final String CLASS_PATH = "src/main/java/com/example/entity/"; public static void main(String[] args) { Connection conn = null; try { // 获取数据库连接 conn = DriverManager.getConnection(DB_URL, DB_USER, DB_PASS); // 获取字段信息 List<ColumnInfo> columnInfos = getColumnInfos(conn, "user"); // 生成 Java 实体类 generateEntityClass("User", columnInfos); } catch (SQLException e) { e.printStackTrace(); } finally { // 关闭数据库连接 if (conn != null) { try { conn.close(); } catch (SQLException e) { e.printStackTrace(); } } } } /** * 获取数据库字段信息 * @param conn 数据库连接 * @param tableName 名 * @return 字段信息列 */ private static List<ColumnInfo> getColumnInfos(Connection conn, String tableName) throws SQLException { List<ColumnInfo> columnInfos = new ArrayList<>(); // 获取字段信息的 SQL String sql = String.format("SELECT COLUMN_NAME, DATA_TYPE, COLUMN_COMMENT FROM information_schema.COLUMNS WHERE TABLE_SCHEMA='%s' AND TABLE_NAME='%s'", conn.getCatalog(), tableName); Statement stmt = conn.createStatement(); ResultSet rs = stmt.executeQuery(sql); // 遍历结果集,获取字段信息 while (rs.next()) { String columnName = rs.getString("COLUMN_NAME"); String dataType = rs.getString("DATA_TYPE"); String columnComment = rs.getString("COLUMN_COMMENT"); columnInfos.add(new ColumnInfo(columnName, dataType, columnComment)); } return columnInfos; } /** * 生成 Java 实体类 * @param className 类名 * @param columnInfos 字段信息列 */ private static void generateEntityClass(String className, List<ColumnInfo> columnInfos) { StringBuilder sb = new StringBuilder(); // 生成包名和导入类 sb.append(String.format("package %s;%n%n", PACKAGE_NAME)); sb.append("import java.io.Serializable;\n\n"); // 生成类声明和实现 Serializable 接口 sb.append(String.format("public class %s implements Serializable {\n\n", className)); sb.append(" private static final long serialVersionUID = 1L;\n\n"); // 生成字段声明和 getter/setter 方法 for (ColumnInfo columnInfo : columnInfos) { sb.append(String.format(" /** %s */\n", columnInfo.getColumnComment())); sb.append(String.format(" private %s %s;\n\n", getJavaType(columnInfo.getDataType()), columnInfo.getColumnName())); sb.append(String.format(" public %s get%s() {\n", getJavaType(columnInfo.getDataType()), capitalize(columnInfo.getColumnName()))); sb.append(String.format(" return %s;\n", columnInfo.getColumnName())); sb.append(" }\n\n"); sb.append(String.format(" public void set%s(%s %s) {\n", capitalize(columnInfo.getColumnName()), getJavaType(columnInfo.getDataType()), columnInfo.getColumnName())); sb.append(String.format(" this.%s = %s;\n", columnInfo.getColumnName(), columnInfo.getColumnName())); sb.append(" }\n\n"); } sb.append("}\n"); // 将生成的 Java 代码写入文件 String filePath = CLASS_PATH + className + ".java"; writeToFile(sb.toString(), filePath); } /** * 根据数据库字段类型获取对应的 Java 类型 * @param dataType 数据库字段类型 * @return Java 类型 */ private static String getJavaType(String dataType) { switch (dataType) { case "int": case "tinyint": case "smallint": case "mediumint": case "bigint": return "Long"; case "float": return "Float"; case "double": return "Double"; case "decimal": return "BigDecimal"; case "varchar": case "text": case "longtext": return "String"; case "datetime": case "timestamp": return "Date"; default: return "Object"; } } /** * 将字符串的第一个字符转换为大写字母 * @param str 字符串 * @return 转换后的字符串 */ private static String capitalize(String str) { return str.substring(0, 1).toUpperCase() + str.substring(1); } /** * 将字符串写入文件 * @param content 字符串内容 * @param filePath 文件路径 */ private static void writeToFile(String content, String filePath) { try { Files.write(Paths.get(filePath), content.getBytes()); } catch (IOException e) { e.printStackTrace(); } } /** * 数据库字段信息类 */ private static class ColumnInfo { private String columnName; // 字段名 private String dataType; // 字段类型 private String columnComment; // 字段注释 public ColumnInfo(String columnName, String dataType, String columnComment) { this.columnName = columnName; this.dataType = dataType; this.columnComment = columnComment; } public String getColumnName() { return columnName; } public String getDataType() { return dataType; } public String getColumnComment() { return columnComment; } } } ``` 上面的代码会根据数据库中的名和字段信息,生成对应的 Java 实体类,并将生成的代码写入到指定的文件中。可以根据需要修改 Java 包名、实体类生成路径等参数,以及修改 `getJavaType()` 方法的实现,以适应不同的数据库类型和 Java 类型。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值