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,期待你的入住，程序员俱乐部,为您提供更多的帮助!");
  }
 
}