数据库表转java对象代码

闲的无聊，看到myeclipse自带的hibernate reverse不够给力，就自己写了个，留作纪念。
package com.qianzheni.ssh.common.engine.crepo;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;

public class CreatePoEngine {
private String driverName="org.gjt.mm.mysql.Driver";
private String sqlUrl="jdbc:mysql://localhost:3306/test";
private String userName="root";
private String pwd="weichuang";
private Connection connection;
private DatabaseMetaData databaseMetaData;
public void initConnection(){
try {
//mysql
Class.forName(this.driverName);
this.connection=DriverManager.getConnection(this.sqlUrl,this.userName,this.pwd);
} catch (Exception e) {
System.out.println("数据库连接失败！请检查数据连接参数！");
e.printStackTrace();
}
}
public void initDatabaseMetaData(){
try {
this.initConnection();
this.databaseMetaData=this.connection.getMetaData();
} catch (SQLException e) {
e.printStackTrace();
}
}
public void go(String fileDir,String packageName){
try {
this.initDatabaseMetaData();
ResultSet resultSet=this.databaseMetaData.getTables(null, null, null, new String[]{"TABLE"});
while (resultSet.next()) {
String tableName=resultSet.getString(3);
ResultSet columnResultSet=this.databaseMetaData.getColumns(this.connection.getCatalog(), null, tableName, "");
BufferedWriter bufferedWriter=new BufferedWriter(new FileWriter(new File(
fileDir+"/"+this.reverseTableName(tableName)+".java")));
bufferedWriter.write("package "+packageName+";\r\n");
bufferedWriter.write("/**\r\n");
bufferedWriter.write(" *\r\n");
bufferedWriter.write(" * @author weichuang\r\n");
bufferedWriter.write(" * @version 0.01.01\r\n");
bufferedWriter.write(" *\r\n");
bufferedWriter.write(" */\r\n");
bufferedWriter.write("public class "+this.reverseTableName(tableName)+"{\r\n");
while (columnResultSet.next()) {
String colName=columnResultSet.getString(4);
String colType=columnResultSet.getString(6);
System.out.println(this.reverseCol(colName, colType));
bufferedWriter.write(this.reverseCol(colName, colType));
}
bufferedWriter.write("}");
bufferedWriter.close();
}
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}finally{
this.close();
}
}
public String reverseCol(String colName,String colType){
if (colType!=null&&"int".equalsIgnoreCase(colType)) {
String name=reverseColName(colName);
String type="Integer";
return "\tprivate "+type+" "+name+";\r\n"+reverseToGetSetMethod(name, type);
}else if (colType!=null&&"varchar".equalsIgnoreCase(colType)) {
String name=reverseColName(colName);
String type="String";
return "\tprivate "+type+" "+name+";\r\n"+reverseToGetSetMethod(name, type);
}else if (colType!=null&&"datetime".equalsIgnoreCase(colType)) {
String name=reverseColName(colName);
String type="java.util.Date";
return "\tprivate "+type+" "+name+";\r\n"+reverseToGetSetMethod(name, type);
}else if (colType!=null&&"date".equalsIgnoreCase(colType)) {
String name=reverseColName(colName);
String type="java.util.Date";
return "\tprivate "+type+" "+name+";\r\n"+reverseToGetSetMethod(name, type);
}else {
throw new RuntimeException("还有没有匹配的列数据类型("+colType+")！");
}
}
public String reverseToGetSetMethod(String name,String type){
StringBuilder stringBuilder=new StringBuilder();
stringBuilder.append("\tpublic "+type+" get"+Character.toUpperCase(name.charAt(0))+name.substring(1)+"(){\r\n");
stringBuilder.append("\t\treturn this."+name+";\r\n");
stringBuilder.append("\t}\r\n");
stringBuilder.append("\tpublic "+"void"+" set"+Character.toUpperCase(name.charAt(0))+name.substring(1)+"("+type+" "+name+"){\r\n");
stringBuilder.append("\t\tthis."+name+"="+name+";\r\n");
stringBuilder.append("\t}\r\n");
return stringBuilder.toString();
}
public String reverseColName(String colName){
StringBuilder stringBuilder=new StringBuilder();
String[] names=colName.split("_");
for (int i = 0; i < names.length; i++) {
String name=names[i].trim();
if (i==0) {
stringBuilder.append(name);
}else {
stringBuilder.append(Character.toUpperCase(name.charAt(0))+name.substring(1));
}
}
return stringBuilder.toString();
}
public String reverseTableName(String colName){
StringBuilder stringBuilder=new StringBuilder();
String[] names=colName.split("_");
for (int i = 0; i < names.length; i++) {
String name=names[i].trim();
stringBuilder.append(Character.toUpperCase(name.charAt(0))+name.substring(1));
}
return stringBuilder.toString();
}
public void close(){
try {
if (this.connection!=null) {
this.connection.close();
}
} catch (SQLException e) {
e.printStackTrace();
}
}
public static void main(String[] args) {
new CreatePoEngine().go("C:\\weichuang\\projects\\myeclipse\\ssh\\src\\main\\java\\com\\qianzheni\\ssh\\common\\entity"
,"com.qianzheni.ssh.common.entity");
}
}