通过数据库表自动生成POJO(JavaBean)对象

2023-08-19 21:07:16

主类:

package bqw.tool;

import java.util.ResourceBundle;
import java.sql.DriverManager;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.Statement;
import java.sql.Types;
import java.io.File;
import java.io.FileWriter;
import java.sql.DatabaseMetaData;
import java.util.ArrayList;

public class DB2JavaBean {

private String driver = ""; //驱动
private String url = ""; //数据库访问串
private String userName = ""; //数据库用户名
private String password = ""; //数据库密码
private String tableName = ""; //要生成jopo对象的表名,使用;进行分割
private String tableMatchPattern = ""; //数据库表名匹配模式
private String matchPattern = ""; //是否启用数据库表名匹配模式功能,启用后tableName属性不被使用

public DB2JavaBean(){}

public DB2JavaBean(boolean init){
   if(init){
    ResourceBundle rb = ResourceBundle.getBundle("bqw.tool.DB2JavaBean");
    this.driver = rb.getString("driver");
    this.url = rb.getString("url");
    this.userName = rb.getString("userName");
    this.password = rb.getString("password");
    this.tableName = rb.getString("tableName");
    this.matchPattern = rb.getString("matchPattern");
    this.tableMatchPattern = rb.getString("tableMatchPattern");
   }

}

public DB2JavaBean(String baseName){
   ResourceBundle rb = ResourceBundle.getBundle(baseName);
   this.driver = rb.getString("driver");
   this.url = rb.getString("url");
   this.userName = rb.getString("userName");
   this.password = rb.getString("password");
   this.tableName = rb.getString("tableName");
   this.tableMatchPattern = rb.getString("tableMatchPattern");
   this.matchPattern = rb.getString("matchPattern");

}

public DB2JavaBean(String driver,String url,String userName,String password,String tableName,String tableMatchPattern,String matchPattern){
   this.driver = driver;
   this.password = password;
   this.userName = userName;
   this.url = url;
   this.tableName = tableName;
   this.tableMatchPattern = tableMatchPattern;
   this.matchPattern = matchPattern;
}

public void setDriver(String driver){
this.driver = driver;
}

public void setUrl(String url){
this.url = url;
}

public void setUserName(String userName){
this.userName = userName;
}

public void setPassword(String password){
this.password = password;
}

public void setTableName(String tableName){
this.tableName = tableName;
}

public void setTableMatchPattern(String tableMatchPattern){
this.tableMatchPattern=tableMatchPattern;
}

public void setMatchPattern(String matchPattern){
this.matchPattern = matchPattern;
}

public String getDriver(){
return this.driver;
}

public String getUrl(){
return this.url;
}

public String getUserName(){
return this.userName;
}

public String getPassword(){
return this.password;
}

public String getTableName(){
return this.tableName;
}

public String getTableMatchPattern(){
return this.tableMatchPattern;
}

public String getMatchPattern(){
   return this.matchPattern;
}
public void init(int ObjectTypeOrCommonlyType){
   try{
    Class.forName(this.driver).newInstance();
    Connection conn = DriverManager.getConnection(this.url,this.userName, this.password);
    String [] tables = new String[0];
    ArrayList tableal = new ArrayList(20);
    if("true".equals(this.matchPattern)){
     DatabaseMetaData dbmd = conn.getMetaData();
     ResultSet dbmdrs = dbmd.getTables(null, this.userName.toUpperCase(), this.tableMatchPattern, new String[]{"TABLE"});
     while(dbmdrs.next()){
      tableal.add(dbmdrs.getString(3));
     }
     dbmdrs.close();
     if(tableal.size()==0){
      dbmdrs = dbmd.getTables(null, this.userName.toLowerCase(), this.tableMatchPattern, new String[]{"TABLE"});
      while(dbmdrs.next()){
       tableal.add(dbmdrs.getString(3));
      }
      dbmdrs.close();
     }
     if(tableal.size()==0){
      dbmdrs = dbmd.getTables(null, this.userName, this.tableMatchPattern, new String[]{"TABLE"});
      while(dbmdrs.next()){
       tableal.add(dbmdrs.getString(3));
      }
      dbmdrs.close();
     }
     tables = new String[tableal.size()];
     for(int ti = 0;ti
      tables[ti] = tableal.get(ti);
     }
    }
    else{
     tables = this.tableName.split(";");
    }
    String strType;
    String strName;
    String className;
    String [] nameSect;
    StringBuilder tbn = new StringBuilder();
    StringBuilder tstr1 = new StringBuilder();
    StringBuilder tstr2 = new StringBuilder();
    File file = new File("JavaBean");
    if(!file.exists())file.mkdir();
    if(!file.isDirectory())file.mkdir();
    for(int i=0;i
     nameSect = tables[i].split("_");
     for(String ns : nameSect){
      tbn.append(ns.substring(0,1).toUpperCase()+ns.substring(1).toLowerCase());
     }

     className = tbn.toString();
     tbn.delete(0,tbn.length());
     tstr1.append("import java.sql.*; ");
     tstr1.append("\n");
     tstr1.append("import javax.sql.*; ");
     tstr1.append("\n");
     tstr1.append("import java.io.*; ");
     tstr1.append("\n");
     tstr1.append("public class "+className+" implements Serializable{ ");
     tstr1.append("\n");
     try{
      System.out.println(tables[i]);
      Statement statement = conn.createStatement();
      ResultSet rs = statement.executeQuery("select * from "+tables[i]);
      ResultSetMetaData rsd = rs.getMetaData();
      int cc = rsd.getColumnCount();
      for(int j=1;j<=cc;j++){
       if(ObjectTypeOrCommonlyType == StaticVar.OBJECTTYPE){
        strType = this.getObjectType(rsd.getColumnType(j));
       }
       else{
        strType = this.getCommonlyType(j);
       }
       if(strType == null)continue;
       strName = rsd.getColumnName(j);
       tstr1.append(" private "+strType+" "+strName.toLowerCase()+";");
       tstr1.append("\n");
       tstr2.append(" public void set"+strName.substring(0,1).toUpperCase()+strName.substring(1).toLowerCase()+"("+strType+" "+strName.toLowerCase()+"){");
       tstr2.append("\n");
       tstr2.append("   this."+strName.toLowerCase()+" = "+strName.toLowerCase()+";");
       tstr2.append("\n");
       tstr2.append(" }");
       tstr2.append("\n");
       tstr2.append(" public "+strType+" get"+strName.substring(0,1).toUpperCase()+strName.substring(1).toLowerCase()+"(){");
       tstr2.append("\n");
       tstr2.append("   return this."+strName.toLowerCase()+";");
       tstr2.append("\n");
       tstr2.append(" }");
       tstr2.append("\n");


      }
      rs.close();
      statement.close();

     }
     catch(Exception tableE)
     {
      tableE.printStackTrace();
     }
     tstr2.append("} ");
     tstr2.append("\n");
     tstr1.append(tstr2.toString());
     tstr1.append("\n");
     file = new File("JavaBean/"+className+".java");
     FileWriter fw = new FileWriter(file);
     fw.write(tstr1.toString());
     fw.flush();
     fw.close();
     tstr1.delete(0, tstr1.length());
     tstr2.delete(0, tstr2.length());
    }
    conn.close();
   }
   catch(Exception driverE){
    driverE.printStackTrace();
   }

}

public String getObjectType(int type){
   switch(type){
   case Types.ARRAY:return null;
   case Types.BIGINT:return "Long";
   case Types.BINARY:return null;
   case Types.BIT:return "Byte";
   case Types.BLOB:return "Blob";
   case Types.BOOLEAN:return "Boolean";
   case Types.CHAR:return "String";
   case Types.CLOB:return "Clob";
   case Types.DATALINK:return null;
   case Types.DATE:return "Date";
   case Types.DECIMAL:return "Double";
   case Types.DISTINCT:return null;
   case Types.DOUBLE:return "Double";
   case Types.FLOAT:return "Float";
   case Types.INTEGER:return "Integer";
   case Types.NUMERIC:return "Integer";
   case Types.JAVA_OBJECT:return null;
   case Types.LONGVARBINARY:return null;
   case Types.LONGVARCHAR:return null;
   case Types.NULL:return null;
   case Types.OTHER:return null;
   case Types.REAL:return null;
   case Types.REF:return null;
   case Types.SMALLINT:return "Short";
   case Types.STRUCT:return null;
   case Types.TIME:return "Time";
   case Types.TIMESTAMP:return "Timestamp";
   case Types.TINYINT:return "Short";
   case Types.VARBINARY:return null;
   case Types.VARCHAR:return "String";
   default :return null;
   }
}

public String getCommonlyType(int type){
   switch(type){
   case Types.ARRAY:return null;
   case Types.BIGINT:return "long";
   case Types.BINARY:return null;
   case Types.BIT:return "byte";
   case Types.BLOB:return "String";
   case Types.BOOLEAN:return "boolean";
   case Types.CHAR:return "String";
   case Types.CLOB:return "String";
   case Types.DATALINK:return null;
   case Types.DATE:return "Date";
   case Types.DECIMAL:return "double";
   case Types.DISTINCT:return null;
   case Types.DOUBLE:return "double";
   case Types.FLOAT:return "float";
   case Types.INTEGER:return "int";
   case Types.NUMERIC:return "int";
   case Types.JAVA_OBJECT:return null;
   case Types.LONGVARBINARY:return null;
   case Types.LONGVARCHAR:return null;
   case Types.NULL:return null;
   case Types.OTHER:return null;
   case Types.REAL:return null;
   case Types.REF:return null;
   case Types.SMALLINT:return "short";
   case Types.STRUCT:return null;
   case Types.TIME:return "Time";
   case Types.TIMESTAMP:return "Timestamp";
   case Types.TINYINT:return "short";
   case Types.VARBINARY:return null;
   case Types.VARCHAR:return "String";
   default :return null;
   }
}

public static void main(String[] args) {
   // TODO Auto-generated method stub
   DB2JavaBean d2j = new DB2JavaBean(true);
   System.out.println(d2j.getDriver());
   System.out.println(d2j.getUrl());
   System.out.println(d2j.getUserName());
   System.out.println(d2j.getPassword());
   System.out.println(d2j.getTableName());
   System.out.println(d2j.getTableMatchPattern());
   System.out.println(d2j.getMatchPattern());
   d2j.init(StaticVar.OBJECTTYPE);
   //d2j.init(StaticVar.COMMONLYTYPE);
   System.out.println("OK");
}

}

静态类:

package bqw.tool;

public class StaticVar {
public static final int OBJECTTYPE = 0;
public static final int COMMONLYTYPE = 1;
}

配置文件:

#驱动
driver=org.apache.derby.jdbc.ClientDriver
#数据库访问串
url=jdbc:derby://localhost:1527/myeclipse
#用户名
userName=classiccars
#密码
password=classiccars
#需要产生pojo文件的表名,使用;号分隔
tableName=customer;employee;office;payment;
#如果matchPattern为true,则tableMatchPattern起作用,tableName不起作用
matchPattern=true
#表名匹配模式,使用like语句的匹配模式,自动选择属于此用户且表名称与tableMatchPattern匹配的表
tableMatchPattern=%

#驱动
#driver=oracle.jdbc.driver.OracleDriver
#数据库访问串
#url=jdbc:oracle:thin:@127.0.0.1:1521:orcl
#用户名
#userName=test
#密码
#password=test
#需要产生pojo文件的表名,使用;号分隔
#tableName=t_files;t_sys_wordtemplate;t_sys_wordmodel;t_sys_user;t_sys_user_dept;
#如果matchPattern为true,则tableMatchPattern起作用,tableName不起作用
#matchPattern=true
#表名匹配模式,使用like语句的匹配模式,自动选择属于此用户且表名称与tableMatchPattern匹配的表,区分大小写
#tableMatchPattern=T_%

转自：http://hi.baidu.com/bianqiwei/item/e37c4eaa913acf14a8cfb7dd

码农公寓

相关文章