通过数据库表字段名生产java entity

生成 Java 实体类，需要获取数据库表的字段名、字段类型、字段注释等信息，然后根据这些信息生成对应的 Java 实体类。下面是一个简单的 Java 实体类生成器的示例代码，可以通过 JDBC 获取数据库表信息，然后生成对应的 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 类型。