加强版JPA entities、repository、service、controller 生成器-java-ICode联盟

IntelliJ IDEA 社区版生成JPA代码
在不借助工具从mysql生成Spring JPA entity实体 GenEntityForJPA 这篇基础上添加了repository、service、controller 生成
package com.iface.tools;

import java.io.File;
import java.io.FileWriter;
import java.io.PrintWriter;
import java.sql.*;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

public class GenEntityForJPA {
    private static final GenEntityForJPA INSTANCE = new GenEntityForJPA();

    private String tableName;// 表名
    private String[] colNames; // 列名数组
    private String[] colTypes; // 列名类型数组
    private int[] colSizes; // 列名大小数组
    private boolean needUtil = false; // 是否需要导入包java.util.*
    private boolean needSql = false; // 是否需要导入包java.sql.*
    private static final SimpleDateFormat SDF = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
    private static final String SQL = "SELECT * FROM ";// 数据库操作

    // TODO 需要修改的地方
    private static final String URL = "jdbc:mysql://yy123:3306/iface";
    private static final String NAME = "sa";
    private static final String PASS = "123456";
    private static final String DRIVER = "com.mysql.jdbc.Driver";
    private String entitiesOutPath = "com.iface.model.entities";// 指定实体生成所在包的路径
    private String repositoryOutPath = "com.iface.repository";// 指定实体生成所在包的路径
    private String serviceOutPath = "com.iface.service";// 指定实体生成所在包的路径
    private String controllerOutPath = "com.iface.web";// 指定实体生成所在包的路径

    private String authorName = "Damon";// 作者名字
    private String[] generateTables = new String[]{"video1", "videoLine", "videoPlayInfo"};//指定需要生成的表的表名，全部生成设置为null

    /**
     * 类的构造方法
     */
    private GenEntityForJPA() {
    }

    /**
     * @return
     * @description 生成class的所有内容
     * @author paul
     * @date 2017年8月18日 下午5:30:07
     * @update 2017年8月18日 下午5:30:07
     * @version V1.0
     */
    private String parse() {
        StringBuffer sb = new StringBuffer();
        sb.append("package " + entitiesOutPath + ";\r\n");
        sb.append("\r\n");

        sb.append("import lombok.Data;\r\n");
        sb.append("import javax.persistence.*;\r\n");
        // 判断是否导入工具包
        if (needUtil) {
            sb.append("import java.util.Date;\r\n");
        }
        if (needSql) {
            sb.append("import java.sql.*;\r\n");
        }
        // 注释部分
        sb.append("/**\r\n");
        sb.append(" * table name:  " + tableName + "\r\n");
        sb.append(" * author name: " + authorName + "\r\n");
        sb.append(" * create time: " + SDF.format(new Date()) + "\r\n");
        sb.append(" */ \r\n");
        // 实体部分


        sb.append(" @Data \r\n");
        sb.append(" @Entity \r\n");
        sb.append("public class " + getTransStr(tableName, true) + "{\r\n\r\n");
        processAllAttrs(sb);// 属性
        sb.append("\r\n");
        //processAllMethod(sb);// get set方法
        sb.append("}\r\n");
        return sb.toString();
    }

    /**
     * @param sb
     * @description 生成所有成员变量
     * @author paul
     * @date 2017年8月18日 下午5:15:04
     * @update 2017年8月18日 下午5:15:04
     * @version V1.0
     */
    private void processAllAttrs(StringBuffer sb) {
        for (int i = 0; i < colNames.length; i++) {
            if (i == 0) {
                sb.append("\t@Id\r\n");
                sb.append("\t@Column(name=\"" + colNames[i] + "\")\r\n");
                sb.append("\t@GeneratedValue(strategy = GenerationType.AUTO)\r\n");
            } else {
                sb.append("\t@Column(name=\"" + colNames[i] + "\")\r\n");
            }
            sb.append("\tprivate " + sqlType2JavaType(colTypes[i]) + " " + getTransStr(colNames[i], false) + ";\r\n\r\n");
        }
    }

    /**
     * @param sb
     * @description 生成所有get/set方法
     * @author paul
     * @date 2017年8月18日 下午5:14:47
     * @update 2017年8月18日 下午5:14:47
     * @version V1.0
     */
    private void processAllMethod(StringBuffer sb) {
        for (int i = 0; i < colNames.length; i++) {
            sb.append("\tpublic void set" + getTransStr(colNames[i], true) + "(" + sqlType2JavaType(colTypes[i]) + " "
                    + getTransStr(colNames[i], false) + "){\r\n");
            sb.append("\t\tthis." + getTransStr(colNames[i], false) + "=" + getTransStr(colNames[i], false) + ";\r\n");
            sb.append("\t}\r\n");
            sb.append("\tpublic " + sqlType2JavaType(colTypes[i]) + " get" + getTransStr(colNames[i], true) + "(){\r\n");
            sb.append("\t\treturn " + getTransStr(colNames[i], false) + ";\r\n");
            sb.append("\t}\r\n");
        }
    }

    /**
     * @param str 传入字符串
     * @return
     * @description 将传入字符串的首字母转成大写
     * @author paul
     * @date 2017年8月18日 下午5:12:12
     * @update 2017年8月18日 下午5:12:12
     * @version V1.0
     */
    private String initCap(String str) {
        char[] ch = str.toCharArray();
        if (ch[0] >= 'a' && ch[0] <= 'z')
            ch[0] = (char) (ch[0] - 32);
        return new String(ch);
    }

    /**
     * @return
     * @description 将mysql中表名和字段名转换成驼峰形式
     * @author paul
     * @date 2017年8月18日 下午4:55:07
     * @update 2017年8月18日 下午4:55:07
     * @version V1.0
     */
    private String getTransStr(String before, boolean firstChar2Upper) {
        //不带"_"的字符串,则直接首字母大写后返回
        if (!before.contains("_"))
            return firstChar2Upper ? initCap(before) : before;
        String[] strs = before.split("_");
        StringBuffer after = null;
        if (firstChar2Upper) {
            after = new StringBuffer(initCap(strs[0]));
        } else {
            after = new StringBuffer(strs[0]);
        }
        if (strs.length > 1) {
            for (int i = 1; i < strs.length; i++)
                after.append(initCap(strs[i]));
        }
        return after.toString();
    }

    /**
     * @return
     * @description 查找sql字段类型所对应的Java类型
     * @author paul
     * @date 2017年8月18日 下午4:55:41
     * @update 2017年8月18日 下午4:55:41
     * @version V1.0
     */
    private String sqlType2JavaType(String sqlType) {
        if (sqlType.equalsIgnoreCase("bit")) {
            return "boolean";
        } else if (sqlType.equalsIgnoreCase("tinyint")) {
            return "byte";
        } else if (sqlType.equalsIgnoreCase("smallint")) {
            return "short";
        } else if (sqlType.equalsIgnoreCase("int")) {
            return "int";
        } else if (sqlType.equalsIgnoreCase("bigint")) {
            return "long";
        } else if (sqlType.equalsIgnoreCase("float")) {
            return "float";
        } else if (sqlType.equalsIgnoreCase("decimal") || sqlType.equalsIgnoreCase("numeric")
                || sqlType.equalsIgnoreCase("real") || sqlType.equalsIgnoreCase("money")
                || sqlType.equalsIgnoreCase("smallmoney")) {
            return "double";
        } else if (sqlType.equalsIgnoreCase("varchar") || sqlType.equalsIgnoreCase("char")
                || sqlType.equalsIgnoreCase("nvarchar") || sqlType.equalsIgnoreCase("nchar")
                || sqlType.equalsIgnoreCase("text")) {
            return "String";
        } else if (sqlType.equalsIgnoreCase("datetime")) {
            return "Date";
        } else if (sqlType.equalsIgnoreCase("image")) {
            return "Blod";
        }
        return null;
    }

    /**
     * @throws Exception
     * @description 生成方法
     * @author paul
     * @date 2017年8月18日 下午2:04:20
     * @update 2017年8月18日 下午2:04:20
     * @version V1.0
     */
    private void generate() throws Exception {
        //与数据库的连接
        Connection con;
        PreparedStatement pStemt = null;
        Class.forName(DRIVER);
        con = DriverManager.getConnection(URL, NAME, PASS);
        System.out.println("connect database success...");
        //获取数据库的元数据
        DatabaseMetaData db = con.getMetaData();
        //是否有指定生成表，有指定则直接用指定表，没有则全表生成
        List<String> tableNames = new ArrayList<>();
        if (generateTables == null) {
            //从元数据中获取到所有的表名
            ResultSet rs = db.getTables(null, null, null, new String[]{"TABLE"});
            while (rs.next()) tableNames.add(rs.getString(3));
        } else {
            for (String tableName : generateTables) tableNames.add(tableName);
        }
        generateTables = tableNames.toArray(new String[tableNames.size()]);
        String tableSql;

        for (int j = 0; j < tableNames.size(); j++) {
            tableName = tableNames.get(j);
            tableSql = SQL + tableName;
            pStemt = con.prepareStatement(tableSql);
            ResultSetMetaData rsmd = pStemt.getMetaData();
            int size = rsmd.getColumnCount();
            colNames = new String[size];
            colTypes = new String[size];
            colSizes = new int[size];
            //获取所需的信息
            for (int i = 0; i < size; i++) {
                colNames[i] = rsmd.getColumnName(i + 1);
                colTypes[i] = rsmd.getColumnTypeName(i + 1);
                if (colTypes[i].equalsIgnoreCase("datetime"))
                    needUtil = true;
                if (colTypes[i].equalsIgnoreCase("image") || colTypes[i].equalsIgnoreCase("text"))
                    needSql = true;
                colSizes[i] = rsmd.getColumnDisplaySize(i + 1);
            }
            //解析生成class的所有内容
            String content = parse();
            genFile(content, entitiesOutPath, tableName);
        }

    }

    private void genFile(String content, String outPath, String classNamw) {
        PrintWriter pw = null;
        try {
            //输出生成文件
            File directory = new File("");
            String dirName = directory.getAbsolutePath() + "/src/main/java/" + outPath.replace(".", "/");
            File dir = new File(dirName);
            if (!dir.exists() && dir.mkdirs()) System.out.println("generate dir 【" + dirName + "】");
            String javaPath = dirName + "/" + getTransStr(classNamw, true) + ".java";
            FileWriter fw = new FileWriter(javaPath);
            pw = new PrintWriter(fw);
            pw.println(content);
            pw.flush();
            System.out.println("create class 【" + classNamw + "】");
        } catch (Exception ex) {
            ex.printStackTrace();
        } finally {
            if (pw != null) pw.close();
        }
    }

    private void genRepository() throws Exception {
        Arrays.stream(generateTables).forEach(tableName -> {

            StringBuffer sb = new StringBuffer();
            sb.append("package " + repositoryOutPath + ";\r\n");
            sb.append("\r\n");
            String entityName = getTransStr(tableName, true);
            sb.append("import " + entitiesOutPath + "." + entityName + ";\r\n");
            sb.append("import org.springframework.data.jpa.repository.JpaRepository;\r\n");

            // 注释部分
            sb.append("\r\n");
            sb.append("/**\r\n");
            sb.append(" * table name:  " + tableName + "\r\n");
            sb.append(" * author name: " + authorName + "\r\n");
            sb.append(" * create time: " + SDF.format(new Date()) + "\r\n");
            sb.append(" */ \r\n");
            // 实体部分

            sb.append("public interface " + entityName + "Repository extends JpaRepository<" + entityName + ", Long>{\r\n\r\n");
            sb.append("}\r\n");
            genFile(sb.toString(), repositoryOutPath, entityName + "Repository");
        });
    }

    private void genService() throws Exception {
        Arrays.stream(generateTables).forEach(tableName -> {
            String entityName = getTransStr(tableName, true);

            StringBuffer sb = new StringBuffer();
            sb.append("package " + serviceOutPath + ";\r\n");
            sb.append("\r\n");
            sb.append("import " + repositoryOutPath + "." + entityName + "Repository;\r\n");
            sb.append("import lombok.extern.slf4j.Slf4j;\r\n");
            sb.append("import org.springframework.stereotype.Service;\r\n");
            sb.append("\r\n");
            sb.append("import javax.inject.Inject;\r\n");

            // 注释部分
            sb.append("\r\n");
            sb.append("/**\r\n");
            sb.append(" * table name:  " + tableName + "\r\n");
            sb.append(" * author name: " + authorName + "\r\n");
            sb.append(" * create time: " + SDF.format(new Date()) + "\r\n");
            sb.append(" */ \r\n");

            sb.append("@Slf4j\r\n");
            sb.append("@Service\r\n");
            sb.append("public class " + entityName + "Service {\r\n\r\n");
            sb.append("\t@Inject\r\n");
            sb.append("\tprivate "+entityName+"Repository "+tableName+"Repository;\r\n");
            sb.append("}\r\n");
            genFile(sb.toString(), serviceOutPath, entityName + "Service");
        });
    }

    private void genController() throws Exception {
        Arrays.stream(generateTables).forEach(tableName -> {
            String entityName = getTransStr(tableName, true);

            StringBuffer sb = new StringBuffer();
            sb.append("package " + controllerOutPath + ";\r\n");
            sb.append("\r\n");
            sb.append("import " + serviceOutPath + "." + entityName + "Service;\r\n");
            sb.append("import lombok.extern.slf4j.Slf4j;\r\n");
            sb.append("import org.springframework.web.bind.annotation.RestController;\r\n");
            sb.append("\r\n");
            sb.append("import javax.inject.Inject;\r\n");

            // 注释部分
            sb.append("/**\r\n");
            sb.append(" * table name:  " + tableName + "\r\n");
            sb.append(" * author name: " + authorName + "\r\n");
            sb.append(" * create time: " + SDF.format(new Date()) + "\r\n");
            sb.append(" */ \r\n");

            sb.append("@Slf4j\r\n");

            sb.append("@RestController(\""+entityName+"Resource\")\r\n");
            sb.append("public class " + entityName + "Resource {\r\n\r\n");
            sb.append("\t@Inject\r\n");
            sb.append("\tprivate "+entityName+"Service "+tableName+"Service;\r\n");
            sb.append("}\r\n");
            genFile(sb.toString(), controllerOutPath, entityName + "Resource");
        });
    }

    /**
     * @param args
     * @description 执行方法
     * @author paul
     * @date 2017年8月18日 下午2:03:35
     * @update 2017年8月18日 下午2:03:35
     * @version V1.0
     */
    public static void main(String[] args) {
        try {
            INSTANCE.generate();
            INSTANCE.genRepository();
            INSTANCE.genService();
            INSTANCE.genController();
            System.out.println("generate classes success!");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}