mybatis_generator

news/2025/11/16 15:39:56/文章来源:https://www.cnblogs.com/SetariaViridis/p/19228301
点击查看 
 
import lombok.Data;import java.sql.*;
import java.util.*;public class CodeGenerator {// 数据库连接配置private static final String DB_URL = "jdbc:mysql://localhost:3306/test";private static final String DB_USERNAME = "DB_USERNAME";private static final String DB_PASSWORD = "DB_PASSWORD";// 代码生成配置private static final String BASE_PACKAGE = "com.example";private static final String AUTHOR = "CodeGenerator";private static final String OUTPUT_DIR = "./src/main/java/";private static final String XML_OUTPUT_DIR = "./src/main/resources/mapper/";public static void main(String[] args) {try {// 加载数据库驱动Class.forName("com.mysql.cj.jdbc.Driver");// 获取数据库连接try (Connection connection = DriverManager.getConnection(DB_URL, DB_USERNAME, DB_PASSWORD)) {System.out.println("数据库连接成功！");// 获取所有表名（这里以user表为例，您可以修改为需要生成的表名）//List<String> tableNames = getTableNames(connection);//或者指定特定表：List<String> tableNames = Arrays.asList("demo");for (String tableName : tableNames) {System.out.println("开始生成表 " + tableName + " 的代码...");// 获取表结构信息TableInfo tableInfo = getTableInfo(connection, tableName);// 生成实体类generateEntity(tableInfo);// 生成Mapper接口generateMapperInterface(tableInfo);// 生成Mapper XMLgenerateMapperXml(tableInfo);System.out.println("表 " + tableName + " 的代码生成完成！");}}} catch (Exception e) {e.printStackTrace();}}/*** 获取数据库中的所有表名*/private static List<String> getTableNames(Connection connection) throws SQLException {List<String> tableNames = new ArrayList<>();DatabaseMetaData metaData = connection.getMetaData();try (ResultSet tables = metaData.getTables(connection.getCatalog(), null, null, new String[]{"TABLE"})) {while (tables.next()) {String tableName = tables.getString("TABLE_NAME");// 过滤系统表if (!tableName.toLowerCase().contains("schema") && !tableName.toLowerCase().contains("temp") &&!tableName.toLowerCase().contains("backup")) {tableNames.add(tableName);}}}return tableNames;}/*** 获取表结构信息*/private static TableInfo getTableInfo(Connection connection, String tableName) throws SQLException {TableInfo tableInfo = new TableInfo();tableInfo.setTableName(tableName);tableInfo.setClassName(convertToCamelCase(tableName, true));tableInfo.setObjectName(convertToCamelCase(tableName, false));DatabaseMetaData metaData = connection.getMetaData();// 获取列信息try (ResultSet columns = metaData.getColumns(connection.getCatalog(), null, tableName, null)) {List<ColumnInfo> columnList = new ArrayList<>();boolean hasBigDecimal = false;boolean hasLocalDateTime = false;while (columns.next()) {ColumnInfo column = new ColumnInfo();column.setColumnName(columns.getString("COLUMN_NAME"));column.setPropertyName(convertToCamelCase(column.getColumnName(), false));column.setJdbcType(convertJdbcType(columns.getString("TYPE_NAME")));column.setJavaType(convertJavaType(columns.getString("TYPE_NAME"), columns.getInt("COLUMN_SIZE")));column.setRemarks(columns.getString("REMARKS"));column.setNullable(columns.getInt("NULLABLE") == DatabaseMetaData.columnNullable);if ("BigDecimal".equals(column.getJavaType())) {hasBigDecimal = true;}if ("LocalDateTime".equals(column.getJavaType()) || "LocalDate".equals(column.getJavaType()) || "LocalTime".equals(column.getJavaType())) {hasLocalDateTime = true;}columnList.add(column);}tableInfo.setColumns(columnList);tableInfo.setHasBigDecimal(hasBigDecimal);tableInfo.setHasLocalDateTime(hasLocalDateTime);}// 获取主键信息try (ResultSet primaryKeys = metaData.getPrimaryKeys(connection.getCatalog(), null, tableName)) {List<String> primaryKeyList = new ArrayList<>();while (primaryKeys.next()) {primaryKeyList.add(primaryKeys.getString("COLUMN_NAME"));}tableInfo.setPrimaryKeys(primaryKeyList);}return tableInfo;}/*** 生成实体类（使用Lombok注解）*/private static void generateEntity(TableInfo tableInfo) {StringBuilder sb = new StringBuilder();// 包名sb.append("package ").append(BASE_PACKAGE).append(".entity;\n\n");// 导入语句generateEntityImports(sb, tableInfo);// 类注释sb.append("/**\n");sb.append(" * ").append(tableInfo.getTableName()).append(" 实体类\n");sb.append(" * \n");sb.append(" * @author ").append(AUTHOR).append("\n");sb.append(" * @date ").append(new java.util.Date()).append("\n");sb.append(" */\n");// Lombok注解sb.append("@Data\n");sb.append("@Builder\n");sb.append("@NoArgsConstructor\n");sb.append("@AllArgsConstructor\n");// 类定义sb.append("public class ").append(tableInfo.getClassName()).append(" implements Serializable {\n");sb.append("    private static final long serialVersionUID = 1L;\n\n");// 字段定义for (ColumnInfo column : tableInfo.getColumns()) {sb.append("    /**\n");sb.append("     * ").append(column.getRemarks().isEmpty() ? column.getColumnName() : column.getRemarks()).append("\n");sb.append("     */\n");sb.append("    private ").append(column.getJavaType()).append(" ").append(column.getPropertyName()).append(";\n\n");}sb.append("}\n");// 写入文件writeToFile(OUTPUT_DIR + BASE_PACKAGE.replace('.', '/') + "/entity/",tableInfo.getClassName() + ".java", sb.toString());}/*** 生成实体类的导入语句*/private static void generateEntityImports(StringBuilder sb, TableInfo tableInfo) {sb.append("import java.io.Serializable;\n");if (tableInfo.isHasBigDecimal()) {sb.append("import java.math.BigDecimal;\n");}if (tableInfo.isHasLocalDateTime()) {sb.append("import java.time.LocalDateTime;\n");sb.append("import java.time.LocalDate;\n");sb.append("import java.time.LocalTime;\n");}sb.append("\n");sb.append("import lombok.Data;\n");sb.append("import lombok.Builder;\n");sb.append("import lombok.NoArgsConstructor;\n");sb.append("import lombok.AllArgsConstructor;\n\n");}/*** 生成Mapper接口*/private static void generateMapperInterface(TableInfo tableInfo) {StringBuilder sb = new StringBuilder();// 包名和导入sb.append("package ").append(BASE_PACKAGE).append(".mapper;\n\n");sb.append("import ").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append(";\n");sb.append("import org.apache.ibatis.annotations.Mapper;\n");sb.append("import org.apache.ibatis.annotations.Param;\n");sb.append("import java.util.List;\n\n");// 接口注释sb.append("/**\n");sb.append(" * ").append(tableInfo.getTableName()).append(" Mapper接口\n");sb.append(" * \n");sb.append(" * @author ").append(AUTHOR).append("\n");sb.append(" * @date ").append(new java.util.Date()).append("\n");sb.append(" */\n");// 接口定义sb.append("@Mapper\n");sb.append("public interface ").append(tableInfo.getClassName()).append("Mapper {\n\n");// 基本CRUD方法sb.append("    int deleteByPrimaryKey(").append(getPrimaryKeyParams(tableInfo)).append(");\n\n");sb.append("    int insert(").append(tableInfo.getClassName()).append(" record);\n\n");sb.append("    int insertSelective(").append(tableInfo.getClassName()).append(" record);\n\n");sb.append("    ").append(tableInfo.getClassName()).append(" selectByPrimaryKey(").append(getPrimaryKeyParams(tableInfo)).append(");\n\n");sb.append("    int updateByPrimaryKeySelective(").append(tableInfo.getClassName()).append(" record);\n\n");sb.append("    int updateByPrimaryKey(").append(tableInfo.getClassName()).append(" record);\n\n");// 批量操作方法sb.append("    int batchInsert(@Param(\"list\") List<").append(tableInfo.getClassName()).append("> list);\n\n");sb.append("    int batchInsertSelective(@Param(\"list\") List<").append(tableInfo.getClassName()).append("> list);\n\n");// 查询方法sb.append("    List<").append(tableInfo.getClassName()).append("> selectAll();\n\n");sb.append("    List<").append(tableInfo.getClassName()).append("> selectByCondition(").append(tableInfo.getClassName()).append(" condition);\n\n");sb.append("    long countByCondition(").append(tableInfo.getClassName()).append(" condition);\n\n");sb.append("}\n");// 写入文件writeToFile(OUTPUT_DIR + BASE_PACKAGE.replace('.', '/') + "/mapper/",tableInfo.getClassName() + "Mapper.java", sb.toString());}/*** 生成Mapper XML文件（完整版）*/private static void generateMapperXml(TableInfo tableInfo) {StringBuilder sb = new StringBuilder();// XML声明和DOCTYPEsb.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");sb.append("<!DOCTYPE mapper PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" \n");sb.append("    \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">\n\n");// Mapper命名空间sb.append("<mapper namespace=\"").append(BASE_PACKAGE).append(".mapper.").append(tableInfo.getClassName()).append("Mapper\">\n\n");// 生成BaseResultMapgenerateResultMap(sb, tableInfo);// 生成所有列generateBaseColumnList(sb, tableInfo);// 生成SQL片段generateBaseColumnListSql(sb, tableInfo);// 基本CRUD操作generateInsert(sb, tableInfo);generateInsertSelective(sb, tableInfo);generateSelectByPrimaryKey(sb, tableInfo);generateUpdateByPrimaryKey(sb, tableInfo);generateUpdateByPrimaryKeySelective(sb, tableInfo);generateDeleteByPrimaryKey(sb, tableInfo);// 批量操作generateBatchInsert(sb, tableInfo);generateBatchInsertSelective(sb, tableInfo);// 查询操作generateSelectAll(sb, tableInfo);generateSelectByCondition(sb, tableInfo);generateCountByCondition(sb, tableInfo);sb.append("</mapper>");// 写入文件writeToFile(XML_OUTPUT_DIR, tableInfo.getClassName() + "Mapper.xml", sb.toString());}/*** 生成ResultMap*/private static void generateResultMap(StringBuilder sb, TableInfo tableInfo) {sb.append("    <resultMap id=\"BaseResultMap\" type=\"").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append("\">\n");for (ColumnInfo column : tableInfo.getColumns()) {if (tableInfo.getPrimaryKeys().contains(column.getColumnName())) {sb.append("        <id column=\"").append(column.getColumnName()).append("\" property=\"").append(column.getPropertyName()).append("\" jdbcType=\"").append(column.getJdbcType()).append("\" />\n");} else {sb.append("        <result column=\"").append(column.getColumnName()).append("\" property=\"").append(column.getPropertyName()).append("\" jdbcType=\"").append(column.getJdbcType()).append("\" />\n");}}sb.append("    </resultMap>\n\n");}/*** 生成基础列列表*/private static void generateBaseColumnList(StringBuilder sb, TableInfo tableInfo) {sb.append("    <sql id=\"Base_Column_List\">\n");sb.append("        ");for (int i = 0; i < tableInfo.getColumns().size(); i++) {ColumnInfo column = tableInfo.getColumns().get(i);sb.append(column.getColumnName());if (i < tableInfo.getColumns().size() - 1) {sb.append(", ");if ((i + 1) % 5 == 0) {sb.append("\n        ");}}}sb.append("\n    </sql>\n\n");}/*** 生成基础列列表SQL*/private static void generateBaseColumnListSql(StringBuilder sb, TableInfo tableInfo) {sb.append("    <sql id=\"Base_Column_List_Sql\">\n");for (int i = 0; i < tableInfo.getColumns().size(); i++) {ColumnInfo column = tableInfo.getColumns().get(i);sb.append("        ").append(column.getColumnName());if (i < tableInfo.getColumns().size() - 1) {sb.append(",\n");}}sb.append("\n    </sql>\n\n");}/*** 生成插入语句*/private static void generateInsert(StringBuilder sb, TableInfo tableInfo) {sb.append("    <insert id=\"insert\" parameterType=\"").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append("\">\n");sb.append("        insert into ").append(tableInfo.getTableName()).append(" (");for (int i = 0; i < tableInfo.getColumns().size(); i++) {ColumnInfo column = tableInfo.getColumns().get(i);sb.append(column.getColumnName());if (i < tableInfo.getColumns().size() - 1) {sb.append(", ");}}sb.append(")\n        values (");for (int i = 0; i < tableInfo.getColumns().size(); i++) {ColumnInfo column = tableInfo.getColumns().get(i);sb.append("#{").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("}");if (i < tableInfo.getColumns().size() - 1) {sb.append(", ");}}sb.append(")\n    </insert>\n\n");}/*** 生成选择性插入语句（修复字符串空字符判断）*/private static void generateInsertSelective(StringBuilder sb, TableInfo tableInfo) {sb.append("    <insert id=\"insertSelective\" parameterType=\"").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append("\">\n");sb.append("        insert into ").append(tableInfo.getTableName()).append("\n");sb.append("        <trim prefix=\"(\" suffix=\")\" suffixOverrides=\",\">\n");for (ColumnInfo column : tableInfo.getColumns()) {// 字符串类型增加空字符判断if ("String".equals(column.getJavaType())) {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null and ").append(column.getPropertyName()).append(" != ''\">\n");} else {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null\">\n");}sb.append("                ").append(column.getColumnName()).append(",\n");sb.append("            </if>\n");}sb.append("        </trim>\n");sb.append("        <trim prefix=\"values (\" suffix=\")\" suffixOverrides=\",\">\n");for (ColumnInfo column : tableInfo.getColumns()) {// 字符串类型增加空字符判断if ("String".equals(column.getJavaType())) {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null and ").append(column.getPropertyName()).append(" != ''\">\n");} else {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null\">\n");}sb.append("                #{").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("},\n");sb.append("            </if>\n");}sb.append("        </trim>\n");sb.append("    </insert>\n\n");}/*** 生成根据主键查询语句*/private static void generateSelectByPrimaryKey(StringBuilder sb, TableInfo tableInfo) {sb.append("    <select id=\"selectByPrimaryKey\" resultMap=\"BaseResultMap\">\n");sb.append("        select \n");sb.append("        <include refid=\"Base_Column_List\" />\n");sb.append("        from ").append(tableInfo.getTableName()).append("\n");sb.append("        where ").append(generatePrimaryKeyWhereClause(tableInfo)).append("\n");sb.append("    </select>\n\n");}/*** 生成更新语句*/private static void generateUpdateByPrimaryKey(StringBuilder sb, TableInfo tableInfo) {sb.append("    <update id=\"updateByPrimaryKey\" parameterType=\"").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append("\">\n");sb.append("        update ").append(tableInfo.getTableName()).append("\n");sb.append("        set \n");boolean first = true;for (ColumnInfo column : tableInfo.getColumns()) {if (!tableInfo.getPrimaryKeys().contains(column.getColumnName())) {if (!first) {sb.append(",\n");}sb.append("            ").append(column.getColumnName()).append(" = #{").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("}");first = false;}}sb.append("\n");sb.append("        where ").append(generatePrimaryKeyWhereClause(tableInfo)) .append("\n");sb.append("    </update>\n\n");}/*** 生成选择性更新语句（修复字符串空字符判断）*/private static void generateUpdateByPrimaryKeySelective(StringBuilder sb, TableInfo tableInfo) {sb.append("    <update id=\"updateByPrimaryKeySelective\" parameterType=\"").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append("\">\n");sb.append("        update ").append(tableInfo.getTableName()).append("\n");sb.append("        <set>\n");for (ColumnInfo column : tableInfo.getColumns()) {if (!tableInfo.getPrimaryKeys().contains(column.getColumnName())) {// 字符串类型增加空字符判断if ("String".equals(column.getJavaType())) {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null and ").append(column.getPropertyName()).append(" != ''\">\n");} else {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null\">\n");}sb.append("                ").append(column.getColumnName()).append(" = #{").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("},\n");sb.append("            </if>\n");}}sb.append("        </set>\n");sb.append("        where ").append(generatePrimaryKeyWhereClause(tableInfo)).append("\n");sb.append("    </update>\n\n");}/*** 生成删除语句*/private static void generateDeleteByPrimaryKey(StringBuilder sb, TableInfo tableInfo) {sb.append("    <delete id=\"deleteByPrimaryKey\">\n");sb.append("        delete from ").append(tableInfo.getTableName()).append("\n");sb.append("        where ").append(generatePrimaryKeyWhereClause(tableInfo)).append("\n");sb.append("    </delete>\n\n");}/*** 生成批量插入语句*/private static void generateBatchInsert(StringBuilder sb, TableInfo tableInfo) {sb.append("    <insert id=\"batchInsert\" parameterType=\"java.util.List\">\n");sb.append("        insert into ").append(tableInfo.getTableName()).append(" (\n");sb.append("            <include refid=\"Base_Column_List\" />\n");sb.append("        ) values\n");sb.append("        <foreach collection=\"list\" item=\"item\" separator=\",\">\n");sb.append("            (");for (int i = 0; i < tableInfo.getColumns().size(); i++) {ColumnInfo column = tableInfo.getColumns().get(i);sb.append("#{item.").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("}");if (i < tableInfo.getColumns().size() - 1) {sb.append(", ");}}sb.append(")\n");sb.append("        </foreach>\n");sb.append("    </insert>\n\n");}/*** 生成批量选择性插入语句（修复字符串空字符判断）*/private static void generateBatchInsertSelective(StringBuilder sb, TableInfo tableInfo) {sb.append("    <insert id=\"batchInsertSelective\" parameterType=\"java.util.List\">\n");sb.append("        insert into ").append(tableInfo.getTableName()).append("\n");sb.append("        <trim prefix=\"(\" suffix=\")\" suffixOverrides=\",\">\n");// 动态列 - 使用第一个元素的非空字段作为列模板for (ColumnInfo column : tableInfo.getColumns()) {// 字符串类型增加空字符判断if ("String".equals(column.getJavaType())) {sb.append("            <if test=\"list[0].").append(column.getPropertyName()).append(" != null and list[0].").append(column.getPropertyName()).append(" != ''\">\n");} else {sb.append("            <if test=\"list[0].").append(column.getPropertyName()).append(" != null\">\n");}sb.append("                ").append(column.getColumnName()).append(",\n");sb.append("            </if>\n");}sb.append("        </trim>\n");sb.append("        values\n");sb.append("        <foreach collection=\"list\" item=\"item\" separator=\",\">\n");sb.append("            <trim prefix=\"(\" suffix=\")\" suffixOverrides=\",\">\n");// 动态值 - 对每个item进行判断for (ColumnInfo column : tableInfo.getColumns()) {// 字符串类型增加空字符判断if ("String".equals(column.getJavaType())) {sb.append("                <if test=\"item.").append(column.getPropertyName()).append(" != null and item.").append(column.getPropertyName()).append(" != ''\">\n");} else {sb.append("                <if test=\"item.").append(column.getPropertyName()).append(" != null\">\n");}sb.append("                    #{item.").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("},\n");sb.append("                </if>\n");}sb.append("            </trim>\n");sb.append("        </foreach>\n");sb.append("    </insert>\n\n");}/*** 生成查询所有语句*/private static void generateSelectAll(StringBuilder sb, TableInfo tableInfo) {sb.append("    <select id=\"selectAll\" resultMap=\"BaseResultMap\">\n");sb.append("        select \n");sb.append("        <include refid=\"Base_Column_List\" />\n");sb.append("        from ").append(tableInfo.getTableName()).append("\n");sb.append("    </select>\n\n");}/*** 生成条件查询语句（修复字符串空字符判断）*/private static void generateSelectByCondition(StringBuilder sb, TableInfo tableInfo) {sb.append("    <select id=\"selectByCondition\" parameterType=\"").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append("\" resultMap=\"BaseResultMap\">\n");sb.append("        select \n");sb.append("        <include refid=\"Base_Column_List\" />\n");sb.append("        from ").append(tableInfo.getTableName()).append("\n");sb.append("        <where>\n");for (ColumnInfo column : tableInfo.getColumns()) {// 字符串类型增加空字符判断if ("String".equals(column.getJavaType())) {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null and ").append(column.getPropertyName()).append(" != ''\">\n");} else {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null\">\n");}sb.append("                and ").append(column.getColumnName()).append(" = #{").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("}\n");sb.append("            </if>\n");}sb.append("        </where>\n");sb.append("    </select>\n\n");}/*** 生成条件计数语句（修复字符串空字符判断）*/private static void generateCountByCondition(StringBuilder sb, TableInfo tableInfo) {sb.append("    <select id=\"countByCondition\" parameterType=\"").append(BASE_PACKAGE).append(".entity.").append(tableInfo.getClassName()).append("\" resultType=\"java.lang.Long\">\n");sb.append("        select count(*) from ").append(tableInfo.getTableName()).append("\n");sb.append("        <where>\n");for (ColumnInfo column : tableInfo.getColumns()) {// 字符串类型增加空字符判断if ("String".equals(column.getJavaType())) {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null and ").append(column.getPropertyName()).append(" != ''\">\n");} else {sb.append("            <if test=\"").append(column.getPropertyName()).append(" != null\">\n");}sb.append("                and ").append(column.getColumnName()).append(" = #{").append(column.getPropertyName()).append(",jdbcType=").append(column.getJdbcType()).append("}\n");sb.append("            </if>\n");}sb.append("        </where>\n");sb.append("    </select>\n\n");}/*** 生成主键WHERE条件子句*/private static String generatePrimaryKeyWhereClause(TableInfo tableInfo) {if (tableInfo.getPrimaryKeys().isEmpty()) {return "1=0"; // 如果没有主键，返回一个false条件}StringBuilder whereClause = new StringBuilder();for (int i = 0; i < tableInfo.getPrimaryKeys().size(); i++) {String pkColumn = tableInfo.getPrimaryKeys().get(i);String pkProperty = convertToCamelCase(pkColumn, false);String jdbcType = getJdbcTypeForColumn(tableInfo, pkColumn);if (i > 0) {whereClause.append("          and ");}whereClause.append(pkColumn).append(" = #{").append(pkProperty).append(",jdbcType=").append(jdbcType).append("}");if (i < tableInfo.getPrimaryKeys().size() - 1) {whereClause.append("\n          ");}}return whereClause.toString();}/*** 获取主键参数（支持复合主键）*/private static String getPrimaryKeyParams(TableInfo tableInfo) {if (tableInfo.getPrimaryKeys().isEmpty()) {return "Long id";}if (tableInfo.getPrimaryKeys().size() == 1) {String pkColumn = tableInfo.getPrimaryKeys().get(0);String pkProperty = convertToCamelCase(pkColumn, false);String javaType = getJavaTypeForColumn(tableInfo, pkColumn);return javaType + " " + pkProperty;} else {// 复合主键，使用@Param注解StringBuilder params = new StringBuilder();for (int i = 0; i < tableInfo.getPrimaryKeys().size(); i++) {String pkColumn = tableInfo.getPrimaryKeys().get(i);String pkProperty = convertToCamelCase(pkColumn, false);String javaType = getJavaTypeForColumn(tableInfo, pkColumn);if (i > 0) {params.append(", ");}params.append("@Param(\"").append(pkProperty).append("\") ").append(javaType).append(" ").append(pkProperty);}return params.toString();}}/*** 工具方法：下划线转驼峰命名*/private static String convertToCamelCase(String name, boolean firstCharUpper) {if (name == null || name.isEmpty()) {return name;}StringBuilder result = new StringBuilder();String[] parts = name.split("_");for (int i = 0; i < parts.length; i++) {String part = parts[i];if (part.isEmpty()) {continue;}if (i == 0 && !firstCharUpper) {result.append(part.toLowerCase());} else {result.append(Character.toUpperCase(part.charAt(0))).append(part.substring(1).toLowerCase());}}return result.toString();}/*** 转换JDBC类型*/private static String convertJdbcType(String dbType) {if (dbType == null) return "VARCHAR";switch (dbType.toUpperCase()) {case "INT":case "INTEGER":return "INTEGER";case "BIGINT":return "BIGINT";case "VARCHAR":case "CHAR":case "TEXT":case "TINYTEXT":case "MEDIUMTEXT":case "LONGTEXT":return "VARCHAR";case "DATETIME":case "TIMESTAMP":return "TIMESTAMP";case "DATE":return "DATE";case "TIME":return "TIME";case "DECIMAL":case "NUMERIC":return "DECIMAL";case "BIT":return "BIT";case "TINYINT":return "TINYINT";case "SMALLINT":return "SMALLINT";case "MEDIUMINT":return "INTEGER";case "FLOAT":return "FLOAT";case "DOUBLE":return "DOUBLE";case "BLOB":case "LONGBLOB":case "MEDIUMBLOB":case "TINYBLOB":return "BLOB";default:return "VARCHAR";}}/*** 转换Java类型（使用LocalDateTime替代Date）*/private static String convertJavaType(String dbType, int size) {if (dbType == null) return "String";String upperDbType = dbType.toUpperCase();switch (upperDbType) {case "INT":case "INTEGER":case "TINYINT":case "SMALLINT":case "MEDIUMINT":return "Integer";case "BIGINT":return "Long";case "VARCHAR":case "CHAR":case "TEXT":case "TINYTEXT":case "MEDIUMTEXT":case "LONGTEXT":case "ENUM":case "SET":return "String";case "DATETIME":case "TIMESTAMP":return "LocalDateTime";case "DATE":return "LocalDate";case "TIME":return "LocalTime";case "DECIMAL":case "NUMERIC":return "BigDecimal";case "BIT":case "BOOL":case "BOOLEAN":return size == 1 ? "Boolean" : "byte[]";case "FLOAT":return "Float";case "DOUBLE":return "Double";case "BINARY":case "VARBINARY":case "BLOB":case "TINYBLOB":case "MEDIUMBLOB":case "LONGBLOB":return "byte[]";default:return "String";}}/*** 获取主键的Java类型*/private static String getPrimaryKeyJavaType(TableInfo tableInfo) {if (tableInfo.getPrimaryKeys().isEmpty()) {return "Long";}String pkColumn = tableInfo.getPrimaryKeys().get(0);return getJavaTypeForColumn(tableInfo, pkColumn);}/*** 获取列的Java类型*/private static String getJavaTypeForColumn(TableInfo tableInfo, String columnName) {for (ColumnInfo column : tableInfo.getColumns()) {if (column.getColumnName().equals(columnName)) {return column.getJavaType();}}return "String";}/*** 获取列的JDBC类型*/private static String getJdbcTypeForColumn(TableInfo tableInfo, String columnName) {for (ColumnInfo column : tableInfo.getColumns()) {if (column.getColumnName().equals(columnName)) {return column.getJdbcType();}}return "VARCHAR";}/*** 写入文件*/private static void writeToFile(String directory, String fileName, String content) {try {java.io.File dir = new java.io.File(directory);if (!dir.exists()) {dir.mkdirs();}try (java.io.FileWriter writer = new java.io.FileWriter(directory + fileName)) {writer.write(content);}System.out.println("生成文件: " + directory + fileName);} catch (Exception e) {e.printStackTrace();}}/*** 表信息类（增强版）*/@Datastatic class TableInfo {private String tableName;private String className;private String objectName;private List<ColumnInfo> columns;private List<String> primaryKeys;private boolean hasBigDecimal;private boolean hasLocalDateTime;}/*** 列信息类*/@Datastatic class ColumnInfo {private String columnName;private String propertyName;private String jdbcType;private String javaType;private String remarks;private boolean nullable;}
}

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.mzph.cn/news/967144.shtml

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

目前市场口碑好的平移门服务商

目前市场口碑好的平移门服务商

摘要 随着智能出入管理系统的快速发展，2025年平移门行业迎来新一轮技术革新。平移门作为智能安防体系的重要组成部分，其安全性、稳定性和智能化水平成为用户关注焦点。本文基于行业数据和技术评测，为您推荐目前市场…
阅读更多...
[AGC030F]Permutation and Minimum

[AGC030F]Permutation and Minimum

Description 给定一个长为 \(2n\) 的序列 \(A\)，其中 \(A\) 的每一位要么是 \(-1\)，要么是 \(1\sim 2n\) 的一个数。将每个 \(A_i=-1\) 替换为 \(1 \sim 2n\) 的某个数，使 \(A\) 成为一个长 \(2n\) 的排列。令长为 …
阅读更多...
2025年安徽伸缩门公司哪家权威：十大品牌综合评测

2025年安徽伸缩门公司哪家权威：十大品牌综合评测

摘要 随着智慧城市建设的加速推进，2025年伸缩门行业迎来技术升级与市场扩容的双重机遇。安徽省作为华东地区制造业重镇，聚集了众多伸缩门生产企业，其中专业技术实力与服务质量成为行业竞争的核心要素。本文基于市场…
阅读更多...
脑机接口

脑机接口

南渡江智慧医疗与康复产业高峰论坛 2025年11月11日 博瑞康 荣脑科技 强脑科技 慧创医疗
阅读更多...
2025年11月阜阳伸缩门供应厂家有哪些

2025年11月阜阳伸缩门供应厂家有哪些

摘要 随着智慧城市建设和安防需求的不断提升，2025年伸缩门行业迎来新一轮发展机遇。阜阳作为安徽省重要工业城市，伸缩门供应市场呈现多元化、专业化发展趋势。本文基于行业数据分析和用户口碑评价，为您推荐阜阳地区…
阅读更多...
【案例实战】多维度视角：鸿蒙2048游戏开发的深度分析与感悟 - 详解

【案例实战】多维度视角：鸿蒙2048游戏开发的深度分析与感悟 - 详解

【案例实战】多维度视角：鸿蒙2048游戏开发的深度分析与感悟 - 详解2025-11-16 15:33 tlnshuju 阅读(0) 评论(0) 收藏 举报pre { white-space: pre !important; word-wrap: normal !important; overflow-x: auto !…
阅读更多...
2025跨境物流/运输公司推荐：中亚/俄罗斯/阿富汗等线路最新top5口碑推荐

2025跨境物流/运输公司推荐：中亚/俄罗斯/阿富汗等线路最新top5口碑推荐

随着跨境电商与国际贸易的蓬勃发展，区域性跨境物流服务商迎来发展机遇。本榜单聚焦临沂及周边地区跨境物流领域，基于线路覆盖能力、时效保障水平、服务附加值、客户口碑四大核心维度，结合行业权威数据与客户反馈，深…
阅读更多...
2025年11月智能床垫品牌TOP5推荐：服务器系统软件办公深度集成

2025年11月智能床垫品牌TOP5推荐：服务器系统软件办公深度集成

本文采用标题：2025年11月智能床垫品牌TOP5推荐：服务器系统软件办公深度集成 摘要 本文基于2025年行业公开数据和推荐对象参考内容，从智能床垫品牌的服务器系统集成能力、软件办公适配性、技术创新及服务品质等维度筛…
阅读更多...
2025年11月载冷剂厂家榜单：性能价格服务综合对比

2025年11月载冷剂厂家榜单：性能价格服务综合对比

入冬在即，冷链、数据中心、新能源电池等温控系统进入高负荷运行期，载冷剂能否在-60℃至+120℃区间持续稳定，直接决定设备能耗与停产风险。多数用户面临三大痛点：一是传统乙二醇溶液腐蚀率高，两年即需换管；二是低…
阅读更多...
Spring 中的 @Configuration 注解

Spring 中的 @Configuration 注解

@Configuration 是 Spring 框架中用于定义配置类的重要注解。它的主要作用是替代传统的 XML 配置文件，通过 Java 代码来定义和管理 Spring 容器中的 Bean。基本用法使用 @Configuration 注解标记一个类，表示该类是一…
阅读更多...
PLUG2：STM32启动流程 - LI,Yi

PLUG2：STM32启动流程 - LI,Yi

不太确定的内容。Step1. STM32上电复位/按键复位此时内核处于已知状态：PC = 0x0000 0000 所有寄存器为默认值，Flash 等待周期按出厂设定1. 硬件复位序列（芯片自动完成） Step2. 取第一个字：主堆栈顶(MSP)内核从 0x…
阅读更多...
C# 封装、继承、抽象、接口

C# 封装、继承、抽象、接口

在C#中，封装、继承、抽象、接口是面向对象编程（OOP）的四大核心特性，它们共同支撑了代码的复用性、扩展性和可维护性。下面分别详解： 一、封装（Encapsulation） 定义：将类的内部数据（字段）和操作数据的方法捆绑…
阅读更多...
python类中的__setattr__

python类中的__setattr__

1. 基本用法 默认情况下，Python 自动处理属性赋值。但你可以重写 setattr 来拦截赋值。 class A:def __setattr__(self, name, value):super().__setattr__(name, value)2. 必须使用 super().setattr 才能真正赋值 错…
阅读更多...
跨域问题解决方案的弃子——JSONP

跨域问题解决方案的弃子——JSONP

今天一位学弟“哭着”来找我，说他面试美团（实习岗）的时候，被问到了跨域问题的解决方案，回答的并不好。我还正想着，这么常见的问题回答不出来，这不就是基础不过关。但谁知道，面试官让他封装 JSONP 来实现跨域！…
阅读更多...
2025年11月智能床垫品牌TOP5推荐：服务器系统软件办公集成优化

2025年11月智能床垫品牌TOP5推荐：服务器系统软件办公集成优化

一、2025年11月智能床垫品牌TOP5全景速览 1、澳洲施华白兰SWAN BEDDING 推荐指数：★★★★★ 口碑评分：9.9分 联系方式：4009936992 基础档案：源自1938年澳大利亚，全球护脊睡眠标杆品牌，依托百年工艺积淀与智能科…
阅读更多...
CPU,GPU,DSP,FPGA,ASIC

CPU,GPU,DSP,FPGA,ASIC

💡 处理器类型概览名称 全称 核心特性 典型应用CPU 中央处理器 (Central Processing Unit) 通用串行计算，高灵活性，擅长复杂逻辑和控制流。 操作系统、日常软件、数据库。GPU 图形处理器 (Graphics Processing Uni…
阅读更多...
智能床垫品牌全面选品指南：2025年11月最新TOP5榜单深度解析

智能床垫品牌全面选品指南：2025年11月最新TOP5榜单深度解析

** 一、2025年11月智能床垫品牌TOP5全景速览 1、澳洲施华白兰SWAN BEDDING 推荐指数：★★★★★ 口碑评分：9.9分 联系方式：4009936992 基础档案：源自1938年澳大利亚，作为全球护脊睡眠标杆品牌，拥有近百年工艺积淀…
阅读更多...
DeepCFD+：一种工业级 CFD 代理模型训练框架【深度学习+流体力学】 - 指南

DeepCFD+：一种工业级 CFD 代理模型训练框架【深度学习+流体力学】 - 指南

pre { white-space: pre !important; word-wrap: normal !important; overflow-x: auto !important; display: block !important; font-family: "Consolas", "Monaco", "Courier New", …
阅读更多...
基于Qt实现的窗口半透明流动背景

基于Qt实现的窗口半透明流动背景

背景基于c++ & Qt实现的半透明渐变色窗口背景先看效果 （gif录制掉帧，请自行编译源码查看，效果比gif流畅）完整源码 GradientWidget.h #ifndef GRADIENTWIDGET_H #define GRADIENTWIDGET_H#include <QWidget&…
阅读更多...
2025河南郑州锅炉设备/改造/安装/维修最新TOP5推荐：质造升级驱动产业新发展，河南中原地区优选

2025河南郑州锅炉设备/改造/安装/维修最新TOP5推荐：质造升级驱动产业新发展，河南中原地区优选

随着工业智能化转型加速，锅炉设备作为能源转化核心装备，其技术创新与安全效能成为企业选型的关键指标。本榜单基于资质认证、技术实力、服务体系三大维度，结合河南省特种设备检测研究院2024年度数据及行业用户调研反…
阅读更多...
最新文章

Copyright @ 2022~2023