【无标题】vue-viewUI-admin

import com.baomidou.mybatisplus.annotation.KeySequence;
import com.baomidou.mybatisplus.annotation.TableName;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileOutputStream;
import java.lang.reflect.Field;
import java.util.*;


/**
 * 根据 java 类自动生成sql 语句 - mysql
 */
public class SqlGenerator {
    private static final Logger logger = LoggerFactory.getLogger(SqlGenerator.class);
    public static Map<String, String> javaProperty2SqlColumnMap = new HashMap<>();

    static {
        javaProperty2SqlColumnMap.put("Integer", "number");
        javaProperty2SqlColumnMap.put("Short", "tinyint(4)");
        javaProperty2SqlColumnMap.put("Long", "number");
        javaProperty2SqlColumnMap.put("BigDecimal", "decimal(19,2)");
        javaProperty2SqlColumnMap.put("Double", "double precision not null");
        javaProperty2SqlColumnMap.put("Float", "float");
        javaProperty2SqlColumnMap.put("boolean", "tinyint(1)");
        javaProperty2SqlColumnMap.put("Boolean", "tinyint(1)");
        javaProperty2SqlColumnMap.put("Timestamp", "timestamp");
        javaProperty2SqlColumnMap.put("Date", "timestamp");
        javaProperty2SqlColumnMap.put("String", "varchar2(255)");
    }

    public static void main(String[] args) {
        //实体类所在的package在磁盘上的绝对路径
        String packageName = "";
        //生成sql的文件夹
        //项目中实体类的路径
        String prefix = "com.it..";
        String className = "";

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

    }

    /**
     * 根据实体类生成建表语句
     *
     * @param className 全类名
     * @author
     * @date
     */
    public static String generateSql(String className) {
        try {
            Class<?> clz = Class.forName(className);
            className = clz.getSimpleName();

            TableName tableAnnotation = clz.getAnnotation(TableName.class);
            KeySequence sequenceAnnotation = clz.getAnnotation(KeySequence.class);
            String tableName = tableAnnotation.value();
            String sequenceName = sequenceAnnotation.value();

            List<Field> columns = new ArrayList<>();
            // 获取继承来的属性
            Field[] fields = clz.getSuperclass().getDeclaredFields();
            Field[] subFields = clz.getDeclaredFields();

            Collections.addAll(columns, fields);
            Collections.addAll(columns, subFields);

            String param = "";
            String column = "";
            StringBuilder sql = null;
            sql = new StringBuilder(50);
            if (tableName == null || ("").equals(tableName)) {
                tableName = clz.getName();
                tableName = tableName.substring(tableName.lastIndexOf(".") + 1);
            }

            sql.append("\n\n/*========= ").append(tableName).append(" ==========*/\n");
            sql.append("DROP TABLE IF EXISTS `").append(className).append("`; \n");
            sql.append("CREATE TABLE `").append(tableName).append("` ( \n");

            for (int i = 0; i < columns.size(); i++) {
                Field f = columns.get(i);
                column = f.getName();
                if (("serialVersionUID").equals(column)) {
                    continue;
                }
                param = f.getType().getSimpleName();

                //将大写字母转小写，并添加下划线
                column = toUnderScoreCase(column);

                sql.append("`").append(column).append("`");
                sql.append(" ").append(javaProperty2SqlColumnMap.get(param));
                if (i != columns.size() - 1){
                    sql.append(",");
                }
                sql.append("\n");
            }
            sql.append(");").append("\n");
            sql.append("CREATE SEQUENCE ").append(sequenceName).append(";");

            return sql.toString();

        } catch (ClassNotFoundException e) {
            logger.debug("该类未找到！");
            return null;
        }
    }

    /**
     * 获取包下的所有类名称,获取的结果类似于 XXX.java
     *
     * @param packageName
     * @return
     * @author
     * @date
     */
    public static List<String> getAllClasses(String packageName) {
        List<String> classList = new ArrayList<String>();
        String className = "";
        File f = new File(packageName);
        if (f.exists() && f.isDirectory()) {
            File[] files = f.listFiles();
            for (File file : files) {
                className = file.getName();
                classList.add(className);
            }
            return classList;
        } else {
            logger.debug("包路径未找到！");
            return null;
        }
    }

    /**
     * 将string 写入sql文件
     *
     * @param str
     * @param path
     * @author
     * @date
     */
    public static void StringToSql(String str, String path) {
        byte[] sourceByte = str.getBytes();
        if (null != sourceByte) {
            try {
                //文件路径（路径+文件名）
                File file = new File(path);
                //文件不存在则创建文件，先创建目录
                if (!file.exists()) {
                    File dir = new File(file.getParent());
                    dir.mkdirs();
                    file.createNewFile();
                }
                //文件输出流用于将数据写入文件
                FileOutputStream outStream = new FileOutputStream(file);
                outStream.write(sourceByte);
                outStream.flush();
                outStream.close();
                System.out.println("生成成功");
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    public static String toUnderScoreCase(String camelCaseStr) {
        StringBuilder builder = new StringBuilder();
        for (int i = 0; i < camelCaseStr.length(); i++) {
            char c = camelCaseStr.charAt(i);
            if (Character.isUpperCase(c)) {
                if (i > 0) {
                    builder.append('_');
                }
                builder.append(Character.toLowerCase(c));
            } else {
                builder.append(c);
            }
        }
        return builder.toString().toUpperCase();
    }
}