mysql表逆向实体类

mysql表逆向实体类

目标框架springboot,mybatisplus

package com.wql.repackage;

import java.io.FileWriter;
import java.io.IOException;
import java.sql.*;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

public class EntityClassGenerator {

    // todo 需要配置
    private static final String HOST = "127.0.0.1";
    private static final int PORT = 3306;
    // todo 需要配置
    private static final String DATABASE = "数据库名称";
    // todo 需要配置
    private static final String USER = "用户名";
    // todo 需要配置
    private static final String PASSWORD = "密码";
    // todo 需要配置
    private static final String OUTPUTDIR = "C:\\Users\\Admin\\Desktop";

    private static final List<String> TABLE_NAMES = new ArrayList<>();

    private static final List<String> CLASS_CONTENTS = new ArrayList<>();

    private static final String URL = "jdbc:mysql://" + HOST + ":" + PORT + "/" + DATABASE;


    // todo 需要配置
    // 是否增加mybatisplus注解
    private static final Boolean IS_ADD_MYBATISPLUS_ANNOTATION = true;
    private static final List<String> INSERT_FIELD_FILL_COLUMN_NAMES = new ArrayList<>();
    private static final List<String> INSERT_UPDATE_FIELD_FILL_COLUMN_NAMES = new ArrayList<>();

    // todo 需要配置
    // 是否逻辑删除
    private static final Boolean IS_LOGIC_DELETE = true;
    // todo 需要配置
    // 逻辑删除字段
    private static final String LOGIC_DELETE_COLUMN_NAME = "delete_flag";

    static{
        // 表
        TABLE_NAMES.add("i18n_module");
        TABLE_NAMES.add("i18n_keyword");
        // 表统一注释
        CLASS_CONTENTS.add("@author: 作者名称");
        CLASS_CONTENTS.add("@date: " + LocalDateTime.now());

        // 需要insert注解的字段
        INSERT_FIELD_FILL_COLUMN_NAMES.add("create_time");
        INSERT_FIELD_FILL_COLUMN_NAMES.add("create_user");
        // 需要update注解的字段
        INSERT_UPDATE_FIELD_FILL_COLUMN_NAMES.add("update_time");
        INSERT_UPDATE_FIELD_FILL_COLUMN_NAMES.add("update_user");

    }


    public static void main(String[] args) {

        try (Connection conn = DriverManager.getConnection(URL, USER, PASSWORD)) {

            try {
                for (String tableName : TABLE_NAMES) {
                    generateEntityForTable(conn,CLASS_CONTENTS,tableName, OUTPUTDIR);
                }
            } catch (Exception e) {
                throw new RuntimeException(e);
            }

            // 打印结束
            System.out.println("Entity class generated successfully.");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private static void generateEntityForTable(Connection conn, List<String> classContents,String tableName, String outputDir) throws SQLException, IOException {
        System.out.println("=========================>>表"+tableName+"=========");
        String className = toCamelCaseTableName(tableName);
        DatabaseMetaData metaData = conn.getMetaData();

        StringBuilder sb = new StringBuilder();
        sb.append("package com.example;\n\n");

        if (IS_ADD_MYBATISPLUS_ANNOTATION) {
            sb.append("import com.baomidou.mybatisplus.annotation.*;\n");
        }
        sb.append("import com.fasterxml.jackson.annotation.JsonFormat;\n");
        sb.append("import lombok.Data;\n\n");

        // 获取表的comment
        String classComment = generateClassComment(classContents);
        sb.append(classComment);

        sb.append("@Data\n");

        // 判断是否需要添加mybatisplus注解
        if(IS_ADD_MYBATISPLUS_ANNOTATION){
            sb.append("@TableName(\""+tableName+"\")\n");
        }


        sb.append("public class ").append(className).append(" {\n");

        String primaryKeyColumnName = "";
        ResultSet primaryKeys = metaData.getPrimaryKeys(DATABASE, null, tableName);
        if (primaryKeys.next()) {
            primaryKeyColumnName = primaryKeys.getString("COLUMN_NAME");
        }

        ResultSet columns = metaData.getColumns(null, null, tableName, null);

        while (columns.next()) {
            String columnName = columns.getString("COLUMN_NAME");
            String columnType = columns.getString("TYPE_NAME");
            String javaColumnName = toCamelCaseColumnName(columnName);
            System.out.println(columnName+"->DB字段类型:"+columnType+"");

            String comment = columns.getString("REMARKS");
            sb.append("    /** ").append(comment).append(" */\n");

            // 判断是否添加mybatisplus注解，并且判断是否主键，如果是主键，就添加
            // @TableId(type = IdType.ASSIGN_ID)
            // @JsonFormat(shape = JsonFormat.Shape.STRING)
            if(IS_ADD_MYBATISPLUS_ANNOTATION && primaryKeyColumnName.equalsIgnoreCase(columnName)){
                sb.append("    @TableId(type = IdType.ASSIGN_ID)\n");
                sb.append("    @JsonFormat(shape = JsonFormat.Shape.STRING)\n");
            }

            // 判断是否需要添加insert注解，并且判断是否是insert字段
            if(IS_ADD_MYBATISPLUS_ANNOTATION && INSERT_FIELD_FILL_COLUMN_NAMES.contains(columnName)){
                sb.append("    @TableField(fill = FieldFill.INSERT)\n");
            }
            // 判断是否需要添加update注解，并且判断是否是update字段
            if(IS_ADD_MYBATISPLUS_ANNOTATION && INSERT_UPDATE_FIELD_FILL_COLUMN_NAMES.contains(columnName)){
                sb.append("    @TableField(fill = FieldFill.INSERT_UPDATE)\n");
            }

            // 逻辑删除
            if(IS_ADD_MYBATISPLUS_ANNOTATION && IS_LOGIC_DELETE && LOGIC_DELETE_COLUMN_NAME.equalsIgnoreCase(columnName)){
                sb.append("    @TableLogic(value = \"0\", delval = \"1\")\n");
            }

            sb.append("    private ").append(mapSqlTypeToJava(columnType)).append(" ").append(javaColumnName).append(";\n\n");
        }

        sb.append("}\n");

        writeToFile(outputDir, className + ".java", sb.toString());

        System.out.println("=========================<<类"+className+"完毕!=====");
    }


    // 生成类注释
    private static String generateClassComment(List<String> classContents) {
        StringBuilder sb = new StringBuilder();
        sb.append("/** \n * \n");

        if(null != classContents){
            for (String classContent : classContents) {
                sb.append(" * ").append(classContent).append("\n");
            }
        }
        sb.append(" */\n");

        return sb.toString();
    }

    private static String mapSqlTypeToJava(String sqlType) {
        switch (sqlType.toUpperCase()) {
            case "DATE":
                return "LocalDate";
            case "DATETIME":
            case "TIMESTAMP":
                return "LocalDateTime";
            case "INT":
            case "BIT":
            case "TINYINT":
                return "Integer";
            case "BIGINT":
            case "BIGINT UNSIGNED":
                return "Long";
            case "VARCHAR":
            case "CHAR":
            case "TEXT":
                return "String";
            // todo 其他未处理的类型 在此处添加
            default:
                return "Object";
        }
    }


    private static String toCamelCaseTableName(String s) {
        if (s == null || s.isEmpty()) {
            return s;
        }
        String[] parts = s.split("_");
        StringBuilder camelCase = new StringBuilder();

        for (String part : parts) {
            if (!part.isEmpty()) {
                camelCase.append(part.substring(0, 1).toUpperCase()).append(part.substring(1).toLowerCase());
            }
        }

        return camelCase.append("DO").toString();
    }


    private static String toCamelCaseColumnName(String s) {
        if (s == null || s.isEmpty()) {
            return s;
        }
        String[] parts = s.split("_");
        StringBuilder camelCase = new StringBuilder();

        for (String part : parts) {
            if (!part.isEmpty()) {
                if (camelCase.length() == 0) {
                    camelCase.append(part.toLowerCase());
                } else {
                    camelCase.append(part.substring(0, 1).toUpperCase()).append(part.substring(1).toLowerCase());
                }
            }
        }

        return camelCase.toString();
    }


    private static void writeToFile(String dir, String fileName, String content) throws IOException {
        FileWriter writer = new FileWriter(dir + "/" + fileName);
        writer.write(content);
        writer.close();
    }
}