??在我們參考別人項目時候特別是SSM項目沒有sql文件的時候我們?nèi)绾斡蓪嶓w類生成數(shù)據(jù)庫表
接下來就和大家一起探討這個問題

package ltd.newbee.mall;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/***
 *
 * @Author : 吟雪情楓
 * @Description 根據(jù)實體類生成建庫語句
 * @Version : v1.0.0
 * @Date : Created in 2020/7/31 10:12
 * @Modified By : 
 **/
public class SqlGenerator {

    private static final Logger logger = LoggerFactory.getLogger(SqlGenerator.class);

    /**
     * 用來存儲Java等屬性類型與sql中屬性類型的對照
     * </br>
     * 例如：java.lang.Integer 對應(yīng) integer
     */
    public static Map<String, String> map = new HashMap<>();

    public static void main(String[] args) {
        map.put("class java.lang.String", "varchar(255)");
        map.put("class java.lang.Integer", "int");
        map.put("class java.lang.Long", "integer unsigned");
        map.put("class java.lang.byte[]", "blob");
        map.put("class java.lang.Boolean", "bit");
        map.put("class java.math.BigInteger", "bigint unsigned");
        map.put("class java.lang.Float", "float");
        map.put("class java.lang.Double", "float");
        map.put("class java.sql.Date", "datetime");
        map.put("class java.sql.Time", "time");
        map.put("class java.sql.Timestamp", "datetime");
        map.put("class java.util.Date", "datetime");
        map.put("class java.lang.Byte", "tinyint");
        sqlConstruction();
    }

    /**
     * 生成sql建庫語句
     */
    private static void sqlConstruction() {
        //實體類所在的package在磁盤上的絕對路徑
        String packageName = "F:/article/newbee-mall/src/main/java/ltd/newbee/mall/entity";
        //生成sql的文件夾
        String filePath = "F:/article/newbee-mall/newbee";
        //項目中實體類的路徑
        String prefix = "ltd.newbee.mall.entity.";
        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, filePath);
            sqls.append(sql);
        }
        System.out.println(sqls.toString());
        StringToSql(sqls.toString(), filePath + ".sql");
    }

    /**
     * 根據(jù)實體類生成建表語句
     *
     * @param className 全類名
     * @param filePath  磁盤路徑
     * @return
     */
    public static String generateSql(String className, String filePath) {
        try {
            Class<?> clz = Class.forName(className);
            className = clz.getSimpleName();
            // 表表名adminUser → tb_admin_user
            className = "tb_" + getStandardFields(className);
            Field[] fields = clz.getDeclaredFields();
            StringBuffer column = new StringBuffer();
            String varchar = " CHARACTER SET utf8 COLLATE utf8_general_ci DEFAULT NULL,";
            for (Field f : fields) {
                if ("class java.lang.String".equals(f.getType().toString())){
                    column.append(" \n `" + getStandardFields(f.getName()) + "`" + " " + map.get(f.getType().toString())).append(varchar);
                }else {
                    column.append(" \n `" + getStandardFields(f.getName()) + "`" + " " + map.get(f.getType().toString())).append(",");
                }
            }
            StringBuffer sql = new StringBuffer();
            sql.append("\n DROP TABLE IF EXISTS `" + className + "`; ")
                    .append(" \n CREATE TABLE `" + className + "`  (")
                    .append(" \n `id` int(11) NOT NULL AUTO_INCREMENT,")
                    .append(" \n " + 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;");
            return sql.toString();
        } catch (ClassNotFoundException e) {
            logger.debug("該類未找到！");
            return null;
        }

    }

    /**
     * 轉(zhuǎn)換為標(biāo)準(zhǔn)等sql字段 例如 adminUser → admin_user
     *
     * @param str 轉(zhuǎn)換為字符串的字段名
     * @return
     */
    public static String getStandardFields(String str) {
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < str.length(); i++) {
            char c = str.charAt(i);
            if (i != 0 && (c > 'A' && c < 'Z')) {
                sb.append("_");
            }
            sb.append(c);
        }
        return sb.toString();
    }

    /**
     * 獲取包下面等所有實體類名稱，類似于獲取 XXX.java
     *
     * @param packageName 全類名
     * @return
     */
    public static List<String> getAllClasses(String packageName) {
        List<String> classList = new ArrayList();
        String className = "";
        File f = new File(packageName);
        if (f.exists() && f.isDirectory()) {
            File[] files = f.listFiles();
            // 遍歷實體類下面等所有.java文件 獲取其類名
            for (File file : files) {
                className = file.getName();
                classList.add(className);
            }
            return classList;
        } else {
            logger.debug("包路徑未找到！");
            return null;
        }
    }

    /**
     * 將生成等String字符串 寫進(jìn)sql文件
     *
     * @param str  String字符串
     * @param path sql文件路徑路徑
     */
    public static void StringToSql(String str, String path) {
        byte[] sourceByte = str.getBytes();
        FileOutputStream os = null;
        if (null != sourceByte) {
            try {
                //文件路徑（路徑+文件名）
                File file = new File(path);
                //文件不存在則創(chuàng)建文件，先創(chuàng)建目錄
                if (!file.exists()) {
                    File dir = new File(file.getParent());
                    dir.mkdirs();
                    file.createNewFile();
                }
                //文件輸出流用于將數(shù)據(jù)寫入文件
                os = new FileOutputStream(file);
                os.write(sourceByte);
                os.flush();
                System.out.println("生成成功!!");
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                // 關(guān)閉文件輸出流
                try {
                    if (os != null) {
                        os.close();
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}

歡迎大家在下方留言一起談?wù)撘黄鸶纳?/p>

一起完善該代碼塊 歡迎一起討論?。。。?/p>