如何根据Mapper的Class反向生成SQL文件创建数据库表

微信公众号：大数据高性能计算

问题：已经根据数据库表生成了一堆的Mapper Class，但是当另一个新同学复制项目的时候没有SQL建表文件，导致无法创建数据库，那么这时候我们就需要通过反射以及注解相关的解析自动去生成数据库表文件。

看一下我们的工程的Mapper：

可以看到我们已经有大量的数据库实体对象了
针对每一个实体对象，我们看下通用的结构：

我们可以看到这个实体Class里面有所有的属性字段，也有表名的注解，自增主键的注解，所有借助反射以及注解我们可以拿到我们想要的信息。 同时这里的属性字段都是驼峰命名的，所以还需要一个操作类将驼峰字段转为下划线字段命名，比如userId 转成 user_id，之后就是Java数据类型与数据库数据类型的基础映射。主要的几个Class类如下：
CamelToSnake: 用于做转换成下划线命名的

public class CamelToSnake {
    public static void main(String[] args) {
        String camelCase = "myVariableName";
        String snakeCase = camelToSnake(camelCase);
        System.out.println(snakeCase);
    }

    public static String camelToSnake(String camelCase) {
        StringBuilder snakeCase = new StringBuilder();
        for (int i = 0; i < camelCase.length(); i++) {
            char currentChar = camelCase.charAt(i);
            if (Character.isUpperCase(currentChar) && i > 0) {
                snakeCase.append('_');
            }
            snakeCase.append(Character.toLowerCase(currentChar));
        }
        return snakeCase.toString();
    }
}

PackageScanner: 扫描我们的实体类所在的对象的路径，拿到所有Class对象，供后面解析所用、

import com.baomidou.mybatisplus.annotation.TableName;

import java.io.File;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

public class PackageScanner {
    public static void main(String[] args) {
        String packageName = "com.fuint.repository.model"; // 替换成您要扫描的包名
        List<Class<?>> classes = scanPackage(packageName);

        for (Class<?> clazz : classes) {
            System.out.println(clazz.getName());
            // 使用 getAnnotation 方法来获取 @TableName 注解
            TableName tableNameAnnotation = clazz.getAnnotation(TableName.class);

            if (tableNameAnnotation != null) {
                String tableName = tableNameAnnotation.value();
                System.out.println("Table Name: " + tableName);
            } else {
                System.out.println("@TableName annotation not found.");
            }
        }
    }

    public static List<Class<?>> scanPackage(String packageName) {
        List<Class<?>> classes = new ArrayList<>();
        String packagePath = packageName.replace('.', '/');
        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
        URL packageURL = classLoader.getResource(packagePath);

        if (packageURL != null) {
            File packageDir = new File(packageURL.getFile());
            if (packageDir.isDirectory()) {
                File[] files = packageDir.listFiles();
                if (files != null) {
                    for (File file : files) {
                        if (file.isFile() && file.getName().endsWith(".class")) {
                            String className = packageName + "." + file.getName().replace(".class", "");
                            try {
                                Class<?> clazz = Class.forName(className);
                                classes.add(clazz);
                            } catch (ClassNotFoundException e) {
                                // 处理异常
                                e.printStackTrace();
                            }
                        }
                    }
                }
            }
        }
        return classes;
    }
}

最后就是实体转化成SQL的全文件解析：

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import io.swagger.annotations.ApiModel;

import java.lang.reflect.Field;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Entity2Sql {


    public static void main(String[] args) {

        List<Class<?>> classes = PackageScanner.scanPackage("com.fuint.repository.model");
        for (Class<?> entityClass : classes) {

            TableName tableNameAnnotation = entityClass.getAnnotation(TableName.class);
            String tableName = null;
            if (tableNameAnnotation != null) {
                tableName = tableNameAnnotation.value();
                //System.out.println("Table Name: " + tableName);
            } else {
                throw new RuntimeException("@TableName annotation not found.");
            }

            StringBuilder sql = new StringBuilder();

            sql.append("DROP TABLE IF EXISTS" ).append(" `").append(tableName).append("`;\n");

            sql.append("CREATE TABLE ").append(tableName).append(" (\n");

            Field[] fields = entityClass.getDeclaredFields();

            String primaryKeyFieldName = "";
            for (Field field : fields) {

                /*ApiModelProperty annotation = field.getAnnotation(ApiModelProperty.class);
                if (annotation == null) {
                    continue;
                }*/
                if (java.lang.reflect.Modifier.isStatic(field.getModifiers())) {
                    continue;
                }

                TableId tableIdAnnotation = field.getAnnotation(TableId.class);
                IdType type = null;
                String value = null;
                if (tableIdAnnotation != null) {
                    type = tableIdAnnotation.type();
                    value = tableIdAnnotation.value();
                }
                String fieldCamel2Snake = CamelToSnake.camelToSnake(field.getName());
                String fieldType = field.getType().getSimpleName();

                // 根据字段类型转换为数据库对应的数据类型
                String dbType = convertToDbType(fieldType);

                // 添加字段和数据类型到SQL语句中
                if (type != null && type == IdType.AUTO) {
                    primaryKeyFieldName = value;
                    sql.append("\t").append(fieldCamel2Snake).append(" ").append(dbType).append(" NOT NULL AUTO_INCREMENT COMMENT '自增ID'").append(",\n");

                } else {
                    sql.append("\t").append(fieldCamel2Snake).append(" ").append(dbType).append(",\n");
                }
            }

            // 使用 getAnnotation 方法来获取 @ApiModel 注解
            ApiModel apiModelAnnotation = entityClass.getAnnotation(ApiModel.class);
            String description = null;
            if (apiModelAnnotation != null) {
                description = apiModelAnnotation.description();
                //System.out.println("Description: " + description);
            } else {
                throw new RuntimeException("@ApiModel annotation not found.");
            }
            // 添加主键约束
            sql.append("\tPRIMARY KEY (" + primaryKeyFieldName + ")\n");
            if (description != null) {

                sql.append(")ENGINE=InnoDB DEFAULT CHARSET=utf8 ROW_FORMAT=COMPRESSED COMMENT='" + description + "';");
            } else {
                sql.append(")ENGINE=InnoDB DEFAULT CHARSET=utf8;");
            }


            System.out.println(sql.toString());
        }
    }

    private static String convertToDbType(String fieldType) {
        Map<String, String> typeMap = new HashMap<>();
        typeMap.put("boolean", "BOOLEAN");
        typeMap.put("byte", "TINYINT");
        typeMap.put("short", "SMALLINT");
        typeMap.put("int", "INT");
        typeMap.put("long", "BIGINT");
        typeMap.put("float", "FLOAT");
        typeMap.put("double", "DOUBLE");
        typeMap.put("char", "CHAR(1)");
        typeMap.put("String", "VARCHAR(255)");
        typeMap.put("Date", "DATETIME");

        typeMap.put("Boolean", "BOOLEAN");
        typeMap.put("Byte", "TINYINT");
        typeMap.put("Short", "SMALLINT");
        typeMap.put("Integer", "INT");
        typeMap.put("Long", "BIGINT");
        typeMap.put("Float", "FLOAT");
        typeMap.put("Double", "DOUBLE");
        typeMap.put("Character", "CHAR(1)");

        // 可以继续添加其他类型的映射关系

        String dbType = typeMap.get(fieldType);

        if (dbType == null) {
            dbType = "VARCHAR(255)";
        }

        return dbType;
    }

}

最后我们看看我们生成的结果：