java工具类–自动将数据库表生成javabean

最近和数据库的表打交道挺多的，因为暂时做的是接口活。

在这过程中发现要把表转换成对应的javabean类型，字段少的表还行，如果不小心碰到几十个字段的他妈的写起来就有点麻烦了，万一碰到几百个的呢，那不是要崩溃。

于是想写个工具类，自动生成javabean。

先说下思路：

1.读取数据库表，获取里面的字段名。

准备连接数据库的驱动包，这里只是针对了oracle数据库和mysql数据库

2.构建一个stringBuffer字符串，用来生成我们需要的类。

3.写入文件

要求具备相应的文件流知识。

好了，准备工作就这些，下面附上代码，

package com.tool;
/*
 * 给出数据库JAR包，数据库链接路径，数据库表空间名，数据库名，数据库密码，表名
 *可以提取出来创建表属性的javaBean文件，并且提供标准的get,set方法。
 * 此程序将所有字段和数据提取出来定义为String类型，如有别的需要可以提取表中字段的类型和别的表信息，自动生成
 * java文件
 * \t 表示 空格
 * \r 表示换行 等价于 \n
 * ResultSetMetaData 关键
 * */
import java.io.BufferedWriter;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;
 
public class SqlToBean {
 private Connection conn = null;    //保存链接路径
 private Statement stmt = null;     //建立连接
 private ResultSetMetaData meta = null;  //保存表属性信息
 private ResultSet rs = null;  //查询结果集
 private OutputStreamWriter osw = null;
 private BufferedWriter bw = null;
 private FileOutputStream fos = null;
 private static StringBuffer coding = new StringBuffer();  //字符串缓冲区
 private String driver = null;    //数据库包名
 private String url = null;          //路径名
 private String table = null;        //表空间名
 private String password = null;     //密码
 private String tableName = null;    //表名
 
 public SqlToBean(String driver, String url, String table, String password, String tableName) {
  this.driver = driver;
  this.url = url;
  this.table = table;
  this.password = password;
  this.tableName = tableName;
 }
 
 private String getCoding(StringBuffer code) {
  return code.toString();
 }
 
 private StringBuffer createGenerate(String property) {
  String prop = property.toLowerCase();
  coding.append("\r \t private String " + prop + ";");
  return coding;
 }
 
 private StringBuffer createMethod(String[] str){
  for(int i=0;i<str.length;i++){
   //str[i].charAt(0) - 32)转成大写   思路
   str[i] = str[i].toLowerCase();
   coding.append("\r \t public void set" + (char)(str[i].charAt(0) - 32)+ str[i].substring(1)+"(String " + str[i] +"){");
   coding.append("\r \t\t this."+str[i] + "=" + str[i] + ";");
   coding.append("\r \t }");
   coding.append("\r \t public String get" + (char)(str[i].charAt(0) - 32)+ str[i].substring(1)+"(){");
   coding.append("\r \t\t return this."+str[i] +  ";");
   coding.append("\r \t }\n");
  }
  return coding;
 }
 /*
  * 关闭与数据库的所有链接
  * */
 private void destroy() {
  try {
   if(conn != null){
    conn.close();
    conn = null;
   }
   if(stmt != null){
    stmt.close();
    stmt = null;
   }
   if(rs != null){
    rs.close();
    rs = null;
   }
 
   if(bw != null){
    bw.close();
    bw = null;
   }
   if(fos != null) {
    fos.close();
    fos = null;
   }
   if(osw != null) {
    osw.close();
    osw = null;
   }
 
  } catch (SQLException e) {
   e.printStackTrace();
  }  catch (IOException e) {
   e.printStackTrace();
  }
 }
 /*
  * 数据库连接发生异常就关闭链接
  * */
 private  void connect () {
  try {
   Class.forName(driver);
   conn = DriverManager.getConnection(url,table,password);
   stmt = conn.createStatement();
 
   rs = stmt.executeQuery("select  * from " + tableName ); //查询下确定结果集是那个表的
   meta = rs.getMetaData();                         //调用结果集的记录表信息的方法
  } catch (ClassNotFoundException e) {
   e.printStackTrace();
   try {
    if(conn != null){
     conn.close();
     conn = null;
    }
    if(stmt != null){
     stmt.close();
     stmt = null;
    }
    if(rs != null){
     rs.close();
     rs = null;
    }
   } catch (SQLException e1) {
    e.printStackTrace();
   }
  }  catch (SQLException e) {
   e.printStackTrace();
   try {
    if(conn != null){
     conn.close();
     conn = null;
    }
    if(stmt != null){
     stmt.close();
     stmt = null;
    }
    if(rs != null){
     rs.close();
     rs = null;
    }
   } catch (SQLException e1) {
    e.printStackTrace();
   }
  }
 
 }
 
 private String[] getColumenName() {
  /*得到表的所有列名以字符串数组的形式返回
   * */
  int count;
  String[] str = null;
  try {
   count = meta.getColumnCount();
   String[] strColumenName = new String[count];
   for(int i = 1;i <= count; i++) {
    strColumenName[i-1] = meta.getColumnName(i);
   }
   str = strColumenName;
  } catch (SQLException e) {
   e.printStackTrace();
  }
  return str;
 }
 /**
  * 写入指定的文件中
  * @param message
  */
 private void writeData(String message,String className) {
  String file = "C:\\"+className+".java";
  try {
   fos = new FileOutputStream(file);
   osw = new OutputStreamWriter(fos);
   bw = new BufferedWriter(osw);
   bw.write(message);
   bw.flush();
  } catch (FileNotFoundException e) {
   e.printStackTrace();
  } catch (IOException e) {
   e.printStackTrace();
  }
 
 }
 
 public StringBuffer createClassName(String className){
  coding.append("public class " + className + "{\n");
  return coding;
 }
 
 public static void main(String[] args) {
  String className = "Hellow";
  //SqlToBean sqlToBean = new SqlToBean("oracle.jdbc.driver.OracleDriver","jdbc:oracle:thin:@192.168.3.11:1521:orcl","mamibon","mamibon","my_standard_data2");
  SqlToBean sqlToBean = new SqlToBean("org.gjt.mm.mysql.Driver","jdbc:mysql://117.79.84.144:3306/wordpress","wangjun","wangjun123","wp_users");
  //连接数据库
  sqlToBean.connect();
  sqlToBean.createClassName(className);
  //获取表的字段
  String[] str ;
  str = sqlToBean.getColumenName();
  for(int i = 0;i<str.length ;i++) {
   sqlToBean.createGenerate(str[i]);
  }
  coding.append("\n");
  sqlToBean.createMethod(str);
  coding.append("\n}");
  //写入文件
  sqlToBean.writeData(sqlToBean.getCoding(coding),className);
  sqlToBean.destroy();
 
  System.out.println("如果觉得这工具类不错,请关注我们的网站:http://www.itbuluoge.com,期待你的入住，程序员俱乐部,为您提供更多的帮助!");
  System.out.println("如果觉得这工具类不错,请关注我们的网站:http://www.itbuluoge.com,期待你的入住，程序员俱乐部,为您提供更多的帮助!");
 }
 
}