該工具類實現以下功能：1、簡單的controller方法2、自動生成Dao類2、自動生成Service類2、自動生成ServiceImpl類
二、連接數據庫// 數據庫配置信息 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";
第二步：其他相關配置//表名 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;
第三步：main方法編寫public static void main(String[] args) { MySQLGeneratorEntityUtil instance = new MySQLGeneratorEntityUtil(); //instance.basePath=""; //指定生成的位置,默認是當前工程 try { instance.generate(); System.out.println("generate Entity toes successful!"); } catch (Exception e) { e.printStackTrace(); } }
第四步：了解instance.generate()方法/** * @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 【" + tableName + "】"); } if (pw != null) pw.close(); }
第五步 了解entity的生成所有自定義的Entity都是這編寫。。。
/** * @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"); // 實體部分 StringExtends = ""; if(needEntityHelper) { classExtends=" extends EntityHelper"; } sb.append("public " + under2camel(tableName, true) +Extends + "{\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(); }
第六步：編寫Dao層/** * @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"); // 實體部分 StringExtends = ""; 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(); }
第七步：編寫Service/** * @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"); // 實體部分 StringExtends = ""; 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(); }
第八：編寫Service實現類/** * @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"); // 實體部分 StringExtends = ""; //if(needEntityHelper) { classExtends=" implements "+under2camel(tableName, true) +"Service"; // } sb.append("@Service\r\n"); sb.append("public " + under2camel(tableName, true) + "ServiceImpl " +classExtends+ "{\r\n\r\n"); processServiceImplMethed(sb); sb.append("}\r\n"); return sb.toString(); }
最後，因爲還有一些工具方法，最完整代碼參見：https://gitee.com/Qinux/MySQLGeneratorEntityUtil
