通过实体类生成对应的建表语句

最近的工作是把json解析后存入数据库,要求根据其结构创建对应的表去存放这些数据,然后就开始建起了实体类,花了整整一天时间,见了60多个类。。。想着要把这些类再创建成表简直是要疯掉。。还好我机智的想到了可以用反射解决这个问题。
我要做的很简单 1.根据类名称获取其字段名称;
2.根据其字段名称拼接成sql语句即可。
接着就在第二天花了半小时捣鼓出了以下代码:

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

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


/**
 * sql自动生成
 * @author 
 * @date 2018年4月11日
 */
public class SqlGenerator {
	private static final Logger logger = LoggerFactory.getLogger(SqlGenerator.class);
	
	public static void main(String[] args) {
		//实体类所在的package在磁盘上的绝对路径
		String packageName = "E:/report";
		//生成sql的文件夹
		String filePath = "E:/create/";
		//项目中实体类的路径
		String prefix = "com.example.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	2018年4月11日
	 * @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	2018年4月11日
	 * @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	2018年4月11日
	 * @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();
		    }  
		}  
	}
}

写这段代码的时候还顺便做了个package下的类名扫描,这样就不用一个一个生成建表语句了。

  • 4
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值