通过实体类生成建表SQL语句实现方法

import java.io.File;

import java.io.FileOutputStream;

import java.lang.reflect.Field;

import java.util.ArrayList;

import java.util.List;

import org.hibernate.hql.internal.ast.SqlGenerator;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class Test {

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

public static void main(String[] args) {

//实体类所在的package在磁盘上的绝对路径

String packageName = "d:/workspace/";

//生成sql的文件夹

String filePath = "I:/create/";

//项目中实体类的路径

String prefix = "com.zh.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 + "report.sql");

}

/**

* 根据实体类生成建表语句

* @author

* @date 2019年1月14日

* @param className 全类名

* @param filePath 磁盘路径 如 : d:/workspace/

*/

public static String generateSql(String className,String filePath){

try {

Class<?> clz = Class.forName(className);

className = clz.getSimpleName();

Field[] fields = clz.getDeclaredFields();

StringBuffer column = new StringBuffer();

String varchar = " varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci DEFAULT NULL,";

for (Field f : fields) {

column.append(" \n `"+f.getName()+"`").append(varchar);

}

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;

}

}

/**

* 获取包下的所有类名称,获取的结果类似于 XXX.java

* @author

* @date 2019年1月14日

* @param packageName

* @return

*/

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文件

* @author

* @date 2019年1月14日

* @param str

* @param path

*/

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();

}

}

}

}

生成的代码截图如下：

原文地址：https://blog.csdn.net/zgsdzczh/article/details/86487149

转载于:https://www.cnblogs.com/jpfss/p/11358320.html