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

最近的工作是把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);

    /**
     * 根据实体类生成建表语句
     * @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下的类名扫描,这样就不用一个一个生成建表语句了。30分钟完美解决了一天的工作量,真是深深的为自己的智慧所折服(●ˇ∀ˇ●),最后说一句:

**

懒惰使人进步~

**

猜你喜欢

转载自blog.csdn.net/ly690226302/article/details/79920319