JAVA依据数据库表结构生成javabean工具类
JAVA根据数据库表结构生成javabean工具类
package com.zzx.study.jdbc;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.text.SimpleDateFormat;
import java.util.Date;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
*
* @ClassName: GenerateJavaBeanUtil
* @Description: TODO 根据表生成JavaBean
* @author A18ccms a18ccms_gmail_com
* @date 2012-11-11 上午12:55:54
*
*/
public class GenerateJavaBeanUtil {
private static final Log log = LogFactory.getLog(GenerateJavaBeanUtil.class);
private String tableName;// 表名
private String[] colnames; // 列名数组
private String[] colTypes; // 列名类型数组
private int[] colSizes; // 列名大小数组
private boolean f_util = false; // 是否需要导入包java.util.Date
private boolean f_sql = false; // 是否需要导入包java.sql.*
private String dataBaseUrl = "jdbc:mysql://localhost:3306/ismp0";
private String userName = "root";
private String password = "mysql";
private String driver = "com.mysql.jdbc.Driver";
private String packagePath;
public void generateJavaBean() {
Connection con = null;
String sql = "select * from " + tableName;
PreparedStatement pStemt = null;
PrintWriter pw =null;
try {
try {
Class.forName(driver);
} catch (ClassNotFoundException e1) {
log.error(e1);
}
con = DriverManager.getConnection(dataBaseUrl, userName, password);
pStemt = con.prepareStatement(sql);
ResultSetMetaData rsmd = pStemt.getMetaData();
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")) {
f_util = true;
}
if (colTypes[i].equalsIgnoreCase("image")
|| colTypes[i].equalsIgnoreCase("text")) {
f_sql = true;
}
colSizes[i] = rsmd.getColumnDisplaySize(i + 1);
}
String content = parse(colnames, colTypes, colSizes, packagePath);
try {
File directory = new File("");
String path = this.getClass().getResource("").getPath();
log.info("Generate JavaBean path="+path);
log.info("src/?/"
+ path.substring(path.lastIndexOf("/com/",
path.length())));
FileWriter fw = new FileWriter(directory.getAbsolutePath()
+ "/src/"
+ path.substring(
path.lastIndexOf("/com/", path.length()),
path.length()) + initcap(tableName) + ".java");
pw = new PrintWriter(fw);
pw.println(content);
pw.flush();
pw.close();
log.info("Generate Success!");
} catch (IOException e) {
log.error(e);
}
} catch (SQLException e) {
log.error(e);
} finally {
try {
con.close();
pStemt.close();
pw.close();
} catch (SQLException e) {
log.error(e);
}
}
}
/**
* @Title: parse
* @Description: TODO 更加列，类型，生成 JAVABEAN
* @param colnames
* @param colTypes
* @param colSizes
* @param packagePath
* @return
* @return String 返回类型
* @throws
*/
private String parse(String[] colnames, String[] colTypes, int[] colSizes,
String packagePath) {
StringBuffer sb = new StringBuffer();
if (f_util) {
sb.append("import java.util.Date;\r\n");
}
if (f_sql) {
sb.append("import java.sql.*;\r\n");
}
sb.append("package ").append(packagePath).append(";");
sb.append("\r\n\r\n");
sb.append("/**\r\n");
sb.append("* " + tableName + " 实体类\r\n");
sb.append("* " + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()) + "\r\n");
sb.append("* create by GenerateJavaBeanUtil " + "\r\n");
sb.append("*/ \r");
sb.append("public class " + initcap(tableName) + "{\r\n");
processAllAttrs(sb);// 属性
processAllMethod(sb);// get set方法
sb.append("}\r\n");
return sb.toString();
}
/**
* @Title: processAllAttrs
* @Description: TODO 生成属性
* @param sb
* @return void 返回类型
* @throws
*/
private void processAllAttrs(StringBuffer sb) {
for (int i = 0; i < colnames.length; i++) {
sb.append("\tprivate " + sqlType2JavaType(colTypes[i]) + " "
+ colnames[i] + ";\r\n");
}
}
/**
* @Title: processAllMethod
* @Description: TODO 生成方法类
* @param sb
* @return void 返回类型
* @throws
*/
private void processAllMethod(StringBuffer sb) {
for (int i = 0; i < colnames.length; i++) {
sb.append("\tpublic void set" + initcap(colnames[i]) + "("
+ sqlType2JavaType(colTypes[i]) + " " +
colnames[i] + "){\r\n");
sb.append("\t\tthis." + colnames[i] + "=" + colnames[i] + ";\r\n");
sb.append("\t}\r\n");
sb.append("\tpublic " + sqlType2JavaType(colTypes[i]) + " get"
+ initcap(colnames[i]) + "(){\r\n");
sb.append("\t\treturn " + colnames[i] + ";\r\n");
sb.append("\t}\r\n");
}
}
/**
* @Title: initcap
* @Description: TODO 首字母大写
* @param str
* @return
* @return String 返回类型
* @throws
*/
private String initcap(String str) {
char[] ch = str.toCharArray();
if (ch[0] >= 'a' && ch[0] <= 'z') {
ch[0] = (char) (ch[0] - 32);
}
return new String(ch);
}
/**
* @Title: sqlType2JavaType
* @Description: TODO 数据库类型转换为java类型
* @param sqlType
* @return
* @return String 返回类型
* @throws
*/
private String 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 "int";
} else if (sqlType.equalsIgnoreCase("bigint")
|| sqlType.equalsIgnoreCase("timestamp")) {
return "long";
} else if (sqlType.equalsIgnoreCase("float")) {
return "float";
} else if (sqlType.equalsIgnoreCase("decimal")
|| 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";
}else if(sqlType.equalsIgnoreCase("tinyint")){
return "byte";
}
return null;
}
/**
*
* @Title: main
* @Description: TODO main测试
* @param args
* @return void 返回类型
* @throws
*/
public static void main(String[] args) {
GenerateJavaBeanUtil generate =new GenerateJavaBeanUtil();
generate.setDataBaseUrl("jdbc:mysql://localhost:3306/ismp0");
generate.setDriver("com.mysql.jdbc.Driver");
generate.setUserName("root");
generate.setPassword("mysql");
generate.setPackagePath("com.zzx.study.jdbc");
generate.setTableName("T_Function");
generate.generateJavaBean();
}
public String getDataBaseUrl() {
return dataBaseUrl;
}
public void setDataBaseUrl(String dataBaseUrl) {
this.dataBaseUrl = dataBaseUrl;
}
public String getUserName() {
return userName;
}
public void setUserName(String userName) {
this.userName = userName;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
public String getDriver() {
return driver;
}
public void setDriver(String driver) {
this.driver = driver;
}
public String getPackagePath() {
return packagePath;
}
public void setPackagePath(String packagePath) {
this.packagePath = packagePath;
}
public String getTableName() {
return tableName;
}
public void setTableName(String tableName) {
this.tableName = tableName;
}
}
这个代码应该很全了，测试的也有，其中用到commons的log，具体的规则很简单，像mysql中的tinyint转换成java的byte类型，其中有一个timestamp笔者是把它转换为long类型，因为笔者项目中不支持date类型，这个可以修改下即可。