Mysql数据库表生成java实体工具类

布满荆棘的人生 2022-05-16 05:09 703阅读 0赞 
  1. import java.io.File;
  2. import java.io.FileWriter;
  3. import java.io.PrintWriter;
  4. import java.sql.Connection;
  5. import java.sql.DatabaseMetaData;
  6. import java.sql.DriverManager;
  7. import java.sql.PreparedStatement;
  8. import java.sql.ResultSet;
  9. import java.sql.ResultSetMetaData;
  10. import java.text.SimpleDateFormat;
  11. import java.util.ArrayList;
  12. import java.util.Date;
  13. import java.util.List;
  15. public class MysqlEntityUtil {
  16.     private static final MysqlEntityUtil INSTANCE = new MysqlEntityUtil();
  18.     private String tableName;// 表名
  19.     private String[] colNames; // 列名数组
  20.     private String[] colTypes; // 列名类型数组
  21.     private int[] colSizes; // 列名大小数组
  22.     private boolean needUtil = false; // 是否需要导入包java.util.*
  23.     private boolean needSql = false; // 是否需要导入包java.sql.*
  24.     private static final SimpleDateFormat SDF = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  25.     private static final String SQL = "SELECT * FROM ";// 数据库操作
  27.     // TODO 需要修改的地方                jdbc:mysql://192.9.8.144:3306/wym_guotu_xc
  28.     private static final String URL = "jdbc:mysql://192.9.8.144:3306/wym_guotu_xc";
  29.     private static final String NAME = "root";
  30.     private static final String PASS = "tj$123456";
  31.     private static final String DRIVER = "com.mysql.jdbc.Driver";
  32.     private String packageOutPath = "com.xc.test.entity.mysql";// 指定实体生成所在包的路径
  33.     private String authorName = "xc";// 作者名字
  34.     private String[] generateTables = null;//指定需要生成的表的表名，全部生成设置为null
  36.     /**
  37.      * 类的构造方法
  38.      */
  39.     private MysqlEntityUtil() {
  40.     }
  42.     /**
  43.      * @return
  44.      * @description 生成class的所有内容
  45.      * @author paul
  46.      * @date 2017年8月18日 下午5:30:07
  47.      * @update 2017年8月18日 下午5:30:07
  48.      * @version V1.0
  49.      */
  50.     private String parse() {
  51.         StringBuffer sb = new StringBuffer();
  52.         sb.append("package " + packageOutPath + ";\r\n");
  53.         sb.append("\r\n");
  54.         // 判断是否导入工具包
  55.         if (needUtil) {
  56.             sb.append("import java.util.Date;\r\n");
  57.         }
  58.         if (needSql) {
  59.             sb.append("import java.sql.*;\r\n");
  60.         }
  61.         // 注释部分
  62.         sb.append("/**\r\n");
  63.         sb.append(" * table name:  " + tableName + "\r\n");
  64.         sb.append(" * author name: " + authorName + "\r\n");
  65.         sb.append(" * create time: " + SDF.format(new Date()) + "\r\n");
  66.         sb.append(" */ \r\n");
  67.         // 实体部分
  68.         sb.append("public class " + getTransStr(tableName, true) + "{\r\n\r\n");
  69.         processAllAttrs(sb);// 属性
  70.         sb.append("\r\n");
  71.         processAllMethod(sb);// get set方法
  72.         //processToString(sb);
  73.         sb.append("}\r\n");
  74.         return sb.toString();
  75.     }
  77.     /**
  78.      * @param sb
  79.      * @description 生成所有成员变量
  80.      * @author paul
  81.      * @date 2017年8月18日 下午5:15:04
  82.      * @update 2017年8月18日 下午5:15:04
  83.      * @version V1.0
  84.      */
  85.     private void processAllAttrs(StringBuffer sb) {
  86.         for (int i = 0; i < colNames.length; i++) {
  87.             sb.append("\tprivate " + sqlType2JavaType(colTypes[i]) + " " + getTransStr(colNames[i], false) + ";\r\n");
  88.         }
  89.     }
  91. //    /**
  92. //     * 重写toString()方法
  93. //     * @param sb
  94. //     */
  95. //    private void processToString(StringBuffer sb) {
  96. //        sb.append("\t@Override\r\n\tpublic String toString() {\r\n");
  97. //        sb.append("\t\treturn \"" +tableName + "{\" + \r\n");
  98. //        for (int i = 0; i < colNames.length; i++) {
  99. //            if (i != 0)
  100. //                sb.append("\t\t\t\", ");
  101. //            if (i == 0)
  102. //                sb.append("\t\t\t\"");
  103. //            sb.append(colNames[i] + "=\" + "
  104. //                    + colNames[i]).append(" + \r\n");
  105. //            if (i == colNames.length - 1) {
  106. //                sb.append("\t\t\t\"}\";\r\n");
  107. //            }
  108. //        }
  109. //        sb.append("\t}\r\n");
  110. //    }
  112.     /**
  113.      * @param sb
  114.      * @description 生成所有get/set方法
  115.      * @author paul
  116.      * @date 2017年8月18日 下午5:14:47
  117.      * @update 2017年8月18日 下午5:14:47
  118.      * @version V1.0
  119.      */
  120.     private void processAllMethod(StringBuffer sb) {
  121.         for (int i = 0; i < colNames.length; i++) {
  122.             sb.append("\tpublic void set" + getTransStr(colNames[i], true) + "(" + sqlType2JavaType(colTypes[i]) + " "
  123.                     + getTransStr(colNames[i], false) + "){\r\n");
  124.             sb.append("\t\tthis." + getTransStr(colNames[i], false) + "=" + getTransStr(colNames[i], false) + ";\r\n");
  125.             sb.append("\t}\r\n");
  126.             sb.append("\tpublic " + sqlType2JavaType(colTypes[i]) + " get" + getTransStr(colNames[i], true) + "(){\r\n");
  127.             sb.append("\t\treturn " + getTransStr(colNames[i], false) + ";\r\n");
  128.             sb.append("\t}\r\n");
  129.         }
  130.     }
  132.     /**
  133.      * @param str 传入字符串
  134.      * @return
  135.      * @description 将传入字符串的首字母转成大写
  136.      * @author paul
  137.      * @date 2017年8月18日 下午5:12:12
  138.      * @update 2017年8月18日 下午5:12:12
  139.      * @version V1.0
  140.      */
  141.     private String initCap(String str) {
  142.         char[] ch = str.toCharArray();
  143.         if (ch[0] >= 'a' && ch[0] <= 'z')
  144.             ch[0] = (char) (ch[0] - 32);
  145.         return new String(ch);
  146.     }
  148.     /**
  149.      * @return
  150.      * @description 将mysql中表名和字段名转换成驼峰形式
  151.      * @author paul
  152.      * @date 2017年8月18日 下午4:55:07
  153.      * @update 2017年8月18日 下午4:55:07
  154.      * @version V1.0
  155.      */
  156.     private String getTransStr(String before, boolean firstChar2Upper) {
  157.         //不带"_"的字符串,则直接首字母大写后返回
  158.         if (!before.contains("_"))
  159.             return firstChar2Upper ? initCap(before) : before;
  160.         String[] strs = before.split("_");
  161.         StringBuffer after = null;
  162.         if (firstChar2Upper) {
  163.             after = new StringBuffer(initCap(strs[0]));
  164.         } else {
  165.             after = new StringBuffer(strs[0]);
  166.         }
  167.         if (strs.length > 1) {
  168.             for (int i=1; i<strs.length; i++)
  169.                 after.append(initCap(strs[i]));
  170.         }
  171.         return after.toString();
  172.     }
  174.     /**
  175.      * @return
  176.      * @description 查找sql字段类型所对应的Java类型
  177.      * @author paul
  178.      * @date 2017年8月18日 下午4:55:41
  179.      * @update 2017年8月18日 下午4:55:41
  180.      * @version V1.0
  181.      */
  182.     private String sqlType2JavaType(String sqlType) {
  183.         if (sqlType.equalsIgnoreCase("bit")) {
  184.             return "boolean";
  185.         } else if (sqlType.equalsIgnoreCase("tinyint")) {
  186.             return "byte";
  187.         } else if (sqlType.equalsIgnoreCase("smallint")) {
  188.             return "short";
  189.         } else if (sqlType.equalsIgnoreCase("int")) {
  190.             return "int";
  191.         } else if (sqlType.equalsIgnoreCase("bigint")) {
  192.             return "long";
  193.         } else if (sqlType.equalsIgnoreCase("float")) {
  194.             return "float";
  195.         } else if (sqlType.equalsIgnoreCase("decimal") || sqlType.equalsIgnoreCase("numeric")
  196.                 || sqlType.equalsIgnoreCase("real") || sqlType.equalsIgnoreCase("money")
  197.                 || sqlType.equalsIgnoreCase("smallmoney")) {
  198.             return "double";
  199.         } else if (sqlType.equalsIgnoreCase("varchar") || sqlType.equalsIgnoreCase("char")
  200.                 || sqlType.equalsIgnoreCase("nvarchar") || sqlType.equalsIgnoreCase("nchar")
  201.                 || sqlType.equalsIgnoreCase("text")) {
  202.             return "String";
  203.         } else if (sqlType.equalsIgnoreCase("datetime")) {
  204.             return "Date";
  205.         } else if (sqlType.equalsIgnoreCase("image")) {
  206.             return "Blod";
  207.         }
  208.         return null;
  209.     }
  211.     /**
  212.      *
  213.      * @description 生成方法
  214.      * @author paul
  215.      * @date 2017年8月18日 下午2:04:20
  216.      * @update 2017年8月18日 下午2:04:20
  217.      * @version V1.0
  218.      * @throws Exception
  219.      */
  220.     private void generate() throws Exception {
  221.         //与数据库的连接
  222.         Connection con;
  223.         PreparedStatement pStemt = null;
  224.         Class.forName(DRIVER);
  225.         con = DriverManager.getConnection(URL, NAME, PASS);
  226.         System.out.println("connect database success...");
  227.         //获取数据库的元数据
  228.         DatabaseMetaData db = con.getMetaData();
  229.         //是否有指定生成表，有指定则直接用指定表，没有则全表生成
  230.         List<String> tableNames = new ArrayList<>();
  231.         if (generateTables == null) {
  232.             //从元数据中获取到所有的表名
  233.             ResultSet rs = db.getTables(null, null, null, new String[] { "TABLE" });
  234.             while (rs.next()) tableNames.add(rs.getString(3));
  235.         } else {
  236.             for (String tableName : generateTables) tableNames.add(tableName);
  237.         }
  238.         String tableSql;
  239.         PrintWriter pw = null;
  240.         for (int j = 0; j < tableNames.size(); j++) {
  241.             tableName = tableNames.get(j);
  242.             tableSql = SQL + tableName;
  243.             pStemt = con.prepareStatement(tableSql);
  244.             ResultSetMetaData rsmd = pStemt.getMetaData();
  245.             int size = rsmd.getColumnCount();
  246.             colNames = new String[size];
  247.             colTypes = new String[size];
  248.             colSizes = new int[size];
  249.             //获取所需的信息
  250.             for (int i = 0; i < size; i++) {
  251.                 colNames[i] = rsmd.getColumnName(i + 1);
  252.                 colTypes[i] = rsmd.getColumnTypeName(i + 1);
  253.                 if (colTypes[i].equalsIgnoreCase("datetime"))
  254.                     needUtil = true;
  255.                 if (colTypes[i].equalsIgnoreCase("image") || colTypes[i].equalsIgnoreCase("text"))
  256.                     needSql = true;
  257.                 colSizes[i] = rsmd.getColumnDisplaySize(i + 1);
  258.             }
  259.             //解析生成class的所有内容
  260.             String content = parse();
  261.             //输出生成文件
  262.             File directory = new File("");
  263.             String dirName = directory.getAbsolutePath() + "/src/main/java/" + packageOutPath.replace(".", "/");
  264.             File dir = new File(dirName);
  265.             if (!dir.exists() && dir.mkdirs()) System.out.println("generate dir 【" + dirName + "】");
  266.             String javaPath = dirName + "/" + getTransStr(tableName, true) + ".java";
  267.             FileWriter fw = new FileWriter(javaPath);
  268.             pw = new PrintWriter(fw);
  269.             pw.println(content);
  270.             pw.flush();
  271.             System.out.println("create class 【" + tableName + "】");
  272.         }
  273.         if (pw != null)
  274.             pw.close();
  275.     }
  277.     /**
  278.      * @param args
  279.      * @description 执行方法
  280.      * @author paul
  281.      * @date 2017年8月18日 下午2:03:35
  282.      * @update 2017年8月18日 下午2:03:35
  283.      * @version V1.0
  284.      */
  285.     public static void main(String[] args) {
  286.         try {
  287.             INSTANCE.generate();
  288.             System.out.println("generate classes success!");
  289.         } catch (Exception e) {
  290.             e.printStackTrace();
  291.         }
  292.     }

发表评论

表情：
评论列表 （有 0 条评论，703人围观）

还没有评论，来说两句吧...

相关阅读