mysql 生成类_mysql逆向生成 java 实体类

importjava.io.File;importjava.io.FileWriter;importjava.io.PrintWriter;importjava.sql.Connection;importjava.sql.DatabaseMetaData;importjava.sql.DriverManager;importjava.sql.PreparedStatement;importjava.sql.ResultSet;importjava.sql.ResultSetMetaData;importjava.text.SimpleDateFormat;importjava.util.ArrayList;importjava.util.Date;importjava.util.List;public classGenEntityMysql {private static final GenEntityMysql INSTANCE = newGenEntityMysql();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://localhost:3306/库名";private static final String NAME = "root";private static final String PASS = "12345678";private static final String DRIVER = "com.mysql.jdbc.Driver";private String packageOutPath = "com.zdz.entity";//指定实体生成所在包的路径

private String authorName = "zdz";//作者名字

private String[] generateTables = {"表名"};//指定需要生成的表的表名，全部生成设置为null

/*** 类的构造方法*/

privateGenEntityMysql() {

}/***@return* @description 生成class的所有内容

*@authorpaul

* @date 2017年8月18日 下午5:30:07

* @update 2017年8月18日 下午5:30:07

*@versionV1.0*/

privateString parse() {

StringBuffer sb= newStringBuffer();

sb.append("package " + packageOutPath + ";\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");

}//注释部分

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 class " + getTransStr(tableName, true) + "{\r\n\r\n");

processAllAttrs(sb);//属性

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

processAllMethod(sb);//get set方法

sb.append("}\r\n");returnsb.toString();

}/***@paramsb

* @description 生成所有成员变量

*@authorpaul

* @date 2017年8月18日 下午5:15:04

* @update 2017年8月18日 下午5:15:04

*@versionV1.0*/

private voidprocessAllAttrs(StringBuffer sb) {for (int i = 0; i < colNames.length; i++) {

sb.append("\tprivate " + sqlType2JavaType(colTypes[i]) + " " + getTransStr(colNames[i], false) + ";\r\n");

}

}/***@paramsb

* @description 生成所有get/set方法

*@authorpaul

* @date 2017年8月18日 下午5:14:47

* @update 2017年8月18日 下午5:14:47

*@versionV1.0*/

private voidprocessAllMethod(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");

}

}/***@paramstr 传入字符串

*@return* @description 将传入字符串的首字母转成大写

*@authorpaul

* @date 2017年8月18日 下午5:12:12

* @update 2017年8月18日 下午5:12:12

*@versionV1.0*/

privateString initCap(String str) {char[] ch =str.toCharArray();if (ch[0] >= 'a' && ch[0] <= 'z')

ch[0] = (char) (ch[0] - 32);return newString(ch);

}/***@return* @description 将mysql中表名和字段名转换成驼峰形式

*@authorpaul

* @date 2017年8月18日 下午4:55:07

* @update 2017年8月18日 下午4:55:07

*@versionV1.0*/

private String getTransStr(String before, booleanfirstChar2Upper) {//不带"_"的字符串,则直接首字母大写后返回

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

after.append(initCap(strs[i]));

}returnafter.toString();

}/***@return* @description 查找sql字段类型所对应的Java类型

*@authorpaul

* @date 2017年8月18日 下午4:55:41

* @update 2017年8月18日 下午4:55:41

*@versionV1.0*/

privateString 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 "Integer";

}else if (sqlType.equalsIgnoreCase("bigint")) {return "Long";

}else if (sqlType.equalsIgnoreCase("float")) {return "float";

}else if (sqlType.equalsIgnoreCase("decimal")) {return "BigDecimal";

}else if (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;

}/***

* @description 生成方法

*@authorpaul

* @date 2017年8月18日 下午2:04:20

* @update 2017年8月18日 下午2:04:20

*@versionV1.0

*@throwsException*/

private void generate() throwsException {//与数据库的连接

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 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);

}

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();int size =rsmd.getColumnCount();

colNames= newString[size];

colTypes= newString[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();//输出生成文件

File directory = new File("");

String dirName= directory.getAbsolutePath() + "/src/main/java/" + packageOutPath.replace(".", "/");

File dir= newFile(dirName);if (!dir.exists() && dir.mkdirs()) System.out.println("generate dir 【" + dirName + "】");

String javaPath= dirName + "/" + getTransStr(tableName, true) + ".java";

FileWriter fw= newFileWriter(javaPath);

pw= newPrintWriter(fw);

pw.println(content);

pw.flush();

System.out.println("create class 【" + tableName + "】");

}if (pw != null)

pw.close();

}/***@paramargs

* @description 执行方法

*@authorpaul

* @date 2017年8月18日 下午2:03:35

* @update 2017年8月18日 下午2:03:35

*@versionV1.0*/

public static voidmain(String[] args) {try{

INSTANCE.generate();

System.out.println("generate classes success!");

}catch(Exception e) {

e.printStackTrace();

}

}

}