通过java实体类逆向生成sql

时间:2025-02-21 07:09:18
package ;

import .slf4j.Slf4j;

import ;
import ;
import ;
import ;
import ;
import ;
import ;
import ;

@Slf4j
public class SqlGenerator {
    //-1小写,1大写
    public static final int format = -1;
    /**
     * 用来存储Java等属性类型与sql中属性类型的对照
     * </br>
     * 例如: 对应 integer
     */
    public static Map<String, String> map = new HashMap<>();

    public static void main(String[] args) {
        ("boolean", "tinyint(1)");
        ("int", "int");
        ("short", "int");
        ("long", "bigint");
        ("float", "float");
        ("double", "double");
        ("char", "varchar(255)");
        ("", "varchar(255)");
        ("", "int");
        ("[]", "blob");
        ("", "bit");
        ("", "integer unsigned");
        ("", "bigint unsigned");
        ("", "float");
        ("", "float");
        ("", "datetime");
        ("", "time");
        ("", "datetime");
        ("", "datetime");
        ("", "tinyint");
        sqlConstruction();
    }

    /**
     * 生成sql建库语句
     */
    private static void sqlConstruction() {
        //实体类所在的package在磁盘上的绝对路径
        String packageName = "C:\\Users\\Administrator\\Desktop\\dletc-qpt\\etc-em-api\\src\\main\\java\\com\\etc\\major\\domain";
        //生成sql的文件夹
        String filePath = "D:\\tmp4";
        //项目中实体类的路径
        String prefix = ".";
        String className = "";

        StringBuffer sqls = new StringBuffer();
        //获取包下的所有类名称
        List<String> list = getAllClasses(packageName);
        for (String str : list) {
            className = prefix + (0, ("."));
            String sql = generateSql(className, filePath);
            (sql);
        }
        StringToSql((), filePath + ".sql");
    }

    /**
     * 根据实体类生成建表语句
     *
     * @param className 全类名
     * @param filePath  磁盘路径
     * @return
     */
    public static String generateSql(String className, String filePath) {
        try {
            Class<?> clz = (className);
            className = ();
            // 表表名adminUser → tb_admin_user
            className = getStandardFields(className).toLowerCase();
            Field[] fields = ();
            StringBuffer column = new StringBuffer();
            String varchar = " CHARACTER SET utf8 COLLATE utf8_general_ci DEFAULT NULL,";
            for (Field f : fields) {
                (().getName());
                String typeName = ().getName();
                if (().equals("serialVersionUID")) {
                    continue;
                }
                if ("".equals(typeName)) {
                    (" \n `" + getStandardFields(()) + "`" + " " + (typeName)).append(varchar);
                } else {
                    if(().equalsIgnoreCase("id")){
                        (" \n `id` int(11) NOT NULL AUTO_INCREMENT,");
                    }else {
                        (" \n `" + getStandardFields(()) + "`" + " " + (typeName)).append(",");
                    }
                }
            }
            StringBuffer sql = new StringBuffer();
            ("\n DROP TABLE IF EXISTS `" + className + "`; ")
                    .append(" \n CREATE TABLE `" + className + "`  (")
//                    .append(" \n `id` int(11) NOT NULL AUTO_INCREMENT,")
                    .append(column)
                    .append(" \n PRIMARY KEY (`id`) USING BTREE,")
                    .append("\n INDEX `id`(`id`) USING BTREE")
                    .append(" \n ) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = utf8 COLLATE = utf8_general_ci;\n");
            return ();
        } catch (ClassNotFoundException e) {
            ("该类未找到!");
            return null;
        }

    }

    /**
     * 转换为标准等sql字段 例如 adminUser → admin_user
     *
     * @param str 转换为字符串的字段名
     * @return
     */
    public static String getStandardFields(String str) {
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < (); i++) {
            char c = (i);
            if (i != 0 && (c > 'A' && c < 'Z')) {
                ("_");
            }
            (c);
        }
        return ();
    }

    /**
     * 获取包下面等所有实体类名称,类似于获取 
     *
     * @param packageName 全类名
     * @return
     */
    public static List<String> getAllClasses(String packageName) {
        List<String> classList = new ArrayList();
        String className = "";
        File f = new File(packageName);
        if (() && ()) {
            File[] files = ();
            // 遍历实体类下面等所有.java文件 获取其类名
            for (File file : files) {
                className = ();
                (className);
            }
            return classList;
        } else {
            ("包路径未找到!");
            return null;
        }
    }

    /**
     * 将生成等String字符串 写进sql文件
     *
     * @param str  String字符串
     * @param path sql文件路径路径
     */
    public static void StringToSql(String str, String path) {
        if (format < 0) {
            str = ();
        } else {
            str = ();
        }
        (str);
        byte[] sourceByte = ();
        FileOutputStream os = null;
        if (null != sourceByte) {
            try {
                //文件路径(路径+文件名)
                File file = new File(path);
                //文件不存在则创建文件,先创建目录
                if (!()) {
                    File dir = new File(());
                    ();
                    ();
                }
                //文件输出流用于将数据写入文件
                os = new FileOutputStream(file);
                (sourceByte);
                ();
                ("生成成功!!");
            } catch (Exception e) {
                ();
            } finally {
                // 关闭文件输出流
                try {
                    if (os != null) {
                        ();
                    }
                } catch (IOException e) {
                    ();
                }
            }
        }
    }
}