遇到百張數據表也不怕，Java自動生成實體、Controller、DAO

一、說明

<code>該工具類實現以下功能： 

1、簡單的controller方法
2、自動生成Dao類
2、自動生成Service類
2、自動生成ServiceImpl類/<code>

二、連接數據庫

<code>
    // 數據庫配置信息
    private static final String URL = "jdbc:mysql://127.0.0.1:3306/test?useUnicode=true&characterEncoding=utf-8&autoReconnect=true";
    private static final String NAME = "root";
    private static final String PASS = "123456";
    private static final String DRIVER = "com.mysql.jdbc.Driver";
/<code>

第二步：其他相關配置

<code> //表名
    private String tableName;
    //列名數組
    private String[] colNames;
    //列名類型數組
    private String[] colTypes;
    //列名大小數組
    private int[] colSizes;
    //列名註釋
    private Map colNamesComment = new HashMap();
    //是否需要導入包java.util.*
    private boolean needUtil = false;
    //是否需要導入包java.sql.*
    private boolean needSql = false;
    //是否需要導入包java.math.BigDecimal
    private boolean needBigDecimal = false;
    //是否創建EntityHelper
    private boolean needEntityHelper = true;
    private static final SimpleDateFormat SDF = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
    private static final String SQL = "SELECT * FROM ";// 數據庫操作
    
    
    //指定實體生成所在包的路徑 

    private static String basePath = new File("").getAbsolutePath();
    //指定包名
    private static String packageEntityOutPath = "com.qinshengfei.system.entity";

    //指定包名
    private static String packageControllerOutPath = "com.qinshengfei.system.controller";

    private static String packageDaoOutPath = "com.qinshengfei.system.dao";

    private static String packageServiceOutPath = "com.qinshengfei.system.service";
    private static String packageServiceimpleOutPath = "com.qinshengfei.system.service.impl";

    //作者名字
    private String authorName = "qinshengfei";
    //指定需要生成的表的表名，全部生成設置為null
    private String[] generateTables = null;
    //主鍵
    private static String pk;
/<code>

第三步：main方法編寫

<code>
    public static void main(String[] args) {
        MySQLGeneratorEntityUtil instance = new MySQLGeneratorEntityUtil();
        //instance.basePath=""; //指定生成的位置,默認是當前工程
        try {
            instance.generate();
            System.out.println("generate Entity to classes successful!");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
/<code>

第四步：瞭解instance.generate()方法

<code> /**
     * @description 生成方法
     */
    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..."+con);
        //獲取數據庫的元數據
        DatabaseMetaData db = con.getMetaData();
        //是否有指定生成表，有指定則直接用指定表，沒有則全表生成
        List<string> tableNames = new ArrayList<>();
        if (generateTables == null) {
            //從元數據中獲取到所有的表名
            //ResultSet rs = db.getTables(null, null, null, new String[] { "TABLE" });
            ResultSet rsTables = db.getTables(null, null, "%", null);
            while (rsTables .next()) {
                String tablename = rsTables.getString(3) ;
                if (tablename.contains("bc_")){
                    tableNames.add(rsTables.getString(3));
                }
            }

           // while (rs.next()) tableNames.add(rs.getString(3));
        } else {
            for (String tableName : generateTables) tableNames.add(tableName);
        }
        if(needEntityHelper) {
            EntityHelper();
        }
        String tableSql;
        PrintWriter pw = null;
        for (int j = 0; j < tableNames.size(); j++) {

            tableName = tableNames.get(j);
            tableSql = SQL + tableName;
            pStemt = con.prepareStatement(tableSql);
            ResultSetMetaData rsmd = pStemt.getMetaData();
            ResultSet rsk = con.getMetaData().getPrimaryKeys(con.getCatalog().toLowerCase(), null, tableName);
            if (rsk.next()) {
                String primaryKey = rsk.getString("COLUMN_NAME");
                pk=primaryKey;
            }
            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);
            }
            //獲取字段註釋
            ResultSet rsComment = pStemt.executeQuery("show full columns from " + tableName);
            while (rsComment.next()) {
                colNamesComment.put(rsComment.getString("Field"), rsComment.getString("Comment"));
            }

            //輸出生成文件
            String content = makeEntity();
            String dirName = MySQLGeneratorEntityUtil.pkgEntityDirName();
            String javaPath = dirName + "/" + under2camel(tableName, true) + ".java";
            FileWriter fw = new FileWriter(javaPath);
            pw = new PrintWriter(fw);
            pw.println(content);
            pw.flush();

            ///controller
            String makeControllerContent = makeController();
            String dirControllerName = MySQLGeneratorEntityUtil.pkgControllerDirName();
            String javaCPath = dirControllerName + "/" + under2camel(tableName, true) + "Controller.java";
            FileWriter fwcontroller = new FileWriter(javaCPath);
            pw = new PrintWriter(fwcontroller);
            pw.println(makeControllerContent);
            pw.flush();
            ///dao

            String daoContent = makeDao();
            String dirDaoName = MySQLGeneratorEntityUtil.pkgDaoDirName();
            String javadaoPath = dirDaoName + "/" + under2camel(tableName, true) + "Dao.java";
            FileWriter fwdao = new FileWriter(javadaoPath);
            pw = new PrintWriter(fwdao);
            pw.println(daoContent);
            pw.flush();

            ///service
            String serviceContent = makeService();
            String dirServiceName = MySQLGeneratorEntityUtil.pkgDaoDirName();
            String javaServicePath = dirServiceName + "/" + under2camel(tableName, true) + "Service.java";
            FileWriter fwService = new FileWriter(javaServicePath);
            pw = new PrintWriter(fwService);
            pw.println(serviceContent);
            pw.flush(); 


            ///serviceImpl
            String serviceImplContent = makeServiceImpl();
            String javaServiceImplPath = dirName + "/service/impl/" + under2camel(tableName, true) + "ServiceImpl.java";
            FileWriter fwServiceimpl = new FileWriter(javaServiceImplPath);
            pw = new PrintWriter(fwServiceimpl);
            pw.println(serviceImplContent);
            pw.flush();



            System.out.println("create class 【" + tableName + "】");
        }
        if (pw != null)
            pw.close();
    }
/<string>/<code>

第五步 瞭解entity的生成

<code>所有自定義的Entity都是這編寫。。。/<code>

<code> /**
     * @description 生成class的所有內容
     */
    private String makeEntity() {
        StringBuffer sb = new StringBuffer();
        sb.append("package " + packageEntityOutPath + ";\\r\\n");
        sb.append("\\r\\n");
        // 判斷是否導入工具包
        if (needUtil) {
            sb.append("import java.util.Date;\\r\\n");
        }
        if (needSql) {
            sb.append("import java.sql.*;\\r\\n");
        }

        for (int i = 0; i < colNames.length; i++) {
            String hasbd = sqlType2JavaType(colTypes[i]);
            if(hasbd =="BigDecimal" || "BigDecimal".equals(hasbd)) {needBigDecimal=true;}
        }
        if(needBigDecimal) {
            sb.append("import java.math.BigDecimal;\\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");
        // 實體部分
        String classExtends = "";
        if(needEntityHelper) {
            classExtends=" extends EntityHelper";
        }
        sb.append("public class " + under2camel(tableName, true) + classExtends + "{\\r\\n\\r\\n");

        processAllAttrs(sb);// 屬性
        sb.append("\\r\\n");
        processConstructor(sb);//構造函數
        processAllMethod(sb);// get set方法
        processToString(sb);
        if(needEntityHelper) {
            processEntityHelper(sb,pk);
        }
        sb.append("}\\r\\n");
        return sb.toString();
    }
/<code>

第六步：編寫Dao層

<code>
    /**
     * @description 生成class的所有內容
     */
    private String makeDao() {
        StringBuffer sb = new StringBuffer();
        sb.append("package " + packageDaoOutPath + ";\\r\\n");
        sb.append("import org.apache.ibatis.annotations.Insert;  \\r\\n");
        sb.append("import org.apache.ibatis.annotations.Update;  \\r\\n");
        sb.append("import org.springframework.stereotype.Component;  \\r\\n");
        sb.append("import com.qinshengfei.system.entity."+under2camel(tableName, true)+" ; \\r\\n");
        sb.append("import org.apache.ibatis.annotations.Select;\\r\\n");
        sb.append("import java.util.List;\\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");
        // 實體部分
        String classExtends = "";
        if(needEntityHelper) {
            classExtends=" ";
        }
        sb.append("@Component\\n");

        sb.append("public interface " + under2camel(tableName, true) + "Dao" + "{\\r\\n\\r\\n");

        sb.append("\\r\\n");
        processDaoMethed(sb);
        sb.append("}\\r\\n");
        return sb.toString();
    }
/<code>

第七步：編寫Service

<code>
    /**
     * @description 生成class的所有內容
     */
    private String makeService() {
        StringBuffer sb = new StringBuffer();
        sb.append("package " + packageServiceOutPath + ";\\r\\n");
        sb.append("\\r\\n");
        // 判斷是否導入工具包
        sb.append("import com.qinshengfei.system.entity."+under2camel(tableName, true)+" ; \\r\\n");
        sb.append("import java.util.List;\\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");
        // 實體部分
        String classExtends = "";
        if(needEntityHelper) {
            classExtends=" extends EntityHelper";
        }
        sb.append("public interface " + under2camel(tableName, true) + "Service" + "{\\r\\n\\r\\n");
        processServiceMethed(sb);
        sb.append("}\\r\\n");
        return sb.toString();
    } 

/<code>

第八：編寫Service實現類

<code>
    /**
     * @description 生成class的所有內容
     */
    private String makeServiceImpl() {
        StringBuffer sb = new StringBuffer();
        sb.append("package " + packageServiceimpleOutPath + ";\\r\\n");
        sb.append("\\r\\n");
        // 判斷是否導入工具包
        sb.append("import com.qinshengfei.system.entity."+under2camel(tableName, true)+" ; \\r\\n");
        sb.append("import org.springframework.beans.factory.annotation.Autowired;\\r\\n");
        sb.append("import com.qinshengfei.system.service."+under2camel(tableName, true)+"Service ; \\r\\n");
        sb.append("import com.qinshengfei.system.dao."+under2camel(tableName, true)+"Dao ; \\r\\n");
        sb.append("import org.springframework.stereotype.Service;\\r\\n");
        sb.append("import java.util.List;\\r\\n");
        sb.append("import org.springframework.stereotype.Service;\\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");
        // 實體部分
        String classExtends = "";
        //if(needEntityHelper) {
            classExtends=" implements "+under2camel(tableName, true) +"Service";
       // }
        sb.append("@Service\\r\\n");
        sb.append("public class " + under2camel(tableName, true) + "ServiceImpl  " +classExtends+  "{\\r\\n\\r\\n");
        processServiceImplMethed(sb);
        sb.append("}\\r\\n");
        return sb.toString();
    }
/<code>

最後，因為還有一些工具方法，最完整代碼參見：

https://gitee.com/Qinux/MySQLGeneratorEntityUtil

閱讀更多 一凡碼農 的文章

關鍵字: 百張 Java 數據庫