利用DB2数据字典自动创建表对应的Java Model

本代码是利用DB2数据字典自动创建表对应的Java Model，提高了表格太多，字段太多时手动创建Model时的效率。由于代码是本人写的，所以有任何问题敬请谅解。

代码中提供了创建单表Model的方法和对应schema下所有表格model的方法。请用到的网友自行测试。

下面直接贴代码：

package com.test.util;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

public class DBModelCreateManager {

private Connection con = null;
private String dbIP;
private String dbPort;
private String dbName;
private String userName;
private String password;

public DBModelCreateManager(String dbIP,String dbPort,String userName,String password,String dbName){
this.dbIP = dbIP;
this.dbName = dbName;
this.dbPort = dbPort;
this.userName = userName;
this.password = password;
getConnection();
}

private void getConnection(){
try {
Class.forName("com.ibm.db2.jcc.DB2Driver");
String url = "jdbc:db2://"+dbIP+":"+dbPort+"/"+dbName;
con = DriverManager.getConnection(url, userName, password);
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (SQLException e) {
e.printStackTrace();
}
}

/**
* 获取对应表的字段信息
* @param tableName
* @return
*/
private List<Map<String,String>> getTableColumns(String tableName,String schema){
List<Map<String,String>> colList = new ArrayList<Map<String,String>>(0);
if(tableName==null){
System.err.println("Neither tableName nor schema can be null!");
}else{
if(con == null) getConnection();
if(con != null){
String sql = "select colname,typename from SYSCAT.COLUMNS where TABNAME='"+tableName.toUpperCase()+"' and TABSCHEMA='"+schema.toUpperCase()+"'"
+" order by COLNO";
PreparedStatement st;
try {
st = con.prepareStatement(sql);
ResultSet rs = st.executeQuery();

if(rs != null){
while(rs.next()){
Map<String,String> map = new HashMap<String, String>();
map.put("colName",rs.getString("colname"));
map.put("className",  getTypeClass(rs.getString("typename")));
colList.add(map);
}
}
} catch (SQLException e) {
e.printStackTrace();
}
}
}
return colList;
}
/**
* 根据数据库类型获取对应java类型
* @param type
* @return
*/
private String getTypeClass(String type){
if(type != null){
if(type.equalsIgnoreCase("VARCHAR")||type.equalsIgnoreCase("CHARACTER")){
return "String";
}else if(type.equalsIgnoreCase("SAMLLINT")||type.equalsIgnoreCase("INTEGER")){
return "Integer";
}else if(type.equalsIgnoreCase("DOUBLE")){
return "double";
}else if(type.equalsIgnoreCase("BIGINT")){
return "long";
}else if(type.equalsIgnoreCase("Date")){
return "Date";
}else if(type.equalsIgnoreCase("TIMESTAMP")){
return "Timestamp";
}
}
return "Object";
}

/**
* 创建单个表对应的model
* @param tableName
* @param packagePath
* @return
* @throws IOException
*/
public void createDBModel(String tableName,String packagePath,String schema) throws IOException{

//创建目录
File f = new File("src/"+packagePath.replace(".", "/")+"/");
f.mkdir();

//因为文件是追加写入，所以文件存在则删除
String tabName = tableName.toLowerCase();
tabName = tabName.substring(0, 1).toUpperCase()+tabName.substring(1);
if(tabName.contains("_")){
String[] items = tabName.toLowerCase().split("_");
tabName = items[0].substring(0, 1).toUpperCase()+items[0].substring(1);
for(int i = 1;i<items.length;i++){
tabName = tabName+items[i].substring(0, 1).toUpperCase()+items[i].substring(1);
}
}
File file = new File("src/"+packagePath.replace(".", "/")+"/"+tabName+".java");
if(file.exists()){
file.delete();
}
OutputStreamWriter out = new FileWriter(file,true);

List<Map<String,String>> colList = getTableColumns(tableName,schema);

StringBuffer fileContent = new StringBuffer("package "+packagePath+";/n");

fileContent.append("import java.util.Date;/n");
fileContent.append("import java.sql.Timestamp;/n");
fileContent.append("public class "+tabName+"{/n");

StringBuffer varStr = new StringBuffer();
StringBuffer methods = new StringBuffer();
for(Iterator<Map<String,String>> it = colList.iterator();it.hasNext();){
Map<String,String> obj = it.next();
String className = obj.get("className");
String colName = obj.get("colName");
String[] items = colName.toLowerCase().split("_");
colName = items[0];
for(int i = 1;i<items.length;i++){
colName = colName+items[i].substring(0, 1).toUpperCase()+items[i].substring(1);
}
//变量定义部分
varStr.append("/t"+"private "+className+" "+colName+";/n");

//set/get方法部分
methods.append("/t"+"public void set"+colName.substring(0, 1).toUpperCase()+colName.substring(1)+"("+className+" "+colName+"){/n");
methods.append("/t/t this."+colName+" = "+colName+";/n");
methods.append("/t}/n/n");
methods.append("/t"+"public "+className+" get"+colName.substring(0, 1).toUpperCase()+colName.substring(1)+"(){/n");
methods.append("/t/t return this."+colName+";/n");
methods.append("/t}/n/n");
}
fileContent.append(varStr.toString()+"/n");
fileContent.append(methods.toString());
fileContent.append("}");

//写入文件
out.write(fileContent.toString());

out.close();

}
/**
* 为对应schema下的所有表创建model
* @param schema
* @param packagePath
* @throws IOException
*/
public void createModelForSchema(String schema,String packagePath) throws IOException{
List<String> tableList = new ArrayList<String>(0);

if(schema==null){
System.err.println("Schema can not be null!");
}else{
if(con == null) getConnection();
if(con != null){
String sql = "select tabname from SYSCAT.TABLES where TABSCHEMA='"+schema.toUpperCase()+"'";
PreparedStatement st;
try {
st = con.prepareStatement(sql);
ResultSet rs = st.executeQuery();

if(rs != null){
while(rs.next()){
tableList.add(rs.getString("tabname"));
}
}
} catch (SQLException e) {
e.printStackTrace();
}
}
}

for(Iterator<String> it = tableList.iterator();it.hasNext();){
String tableName = it.next();
createDBModel(tableName, packagePath,schema);
}
}
}

下面这个是我自己测试生成的一个model的类代码。

package com.test.model;
import java.util.Date;
import java.sql.Timestamp;
public class PatchAdvisory{
private long uid;
private Integer packageSid;
private long swcatUid;
private String vendorId;
private String vendorTitle;
private String rating;
private Timestamp vendorReleaseDate;
private Timestamp receiveDate;
private String sourceAvailable;
private Timestamp publishDate;
private String productType;
private String vendor;
private String product;
private String version;
private String releasename;
private String underlyingOs;
private String architecture;
private String createdby;
private String datasource;
private String lastmodifiedby;
private String deleted;
private String audUser;
private String audType;
private Timestamp audTime;
private Timestamp lastmodifiedtime;
private String validFlag;
private String vendorUrl;
private String sourceUrl;
private String primaryUrl;
private String dq;
private String updatePending;
private Timestamp ratingDate;
private String securityflag;
private String crcCode;
private String ratingDesignator;
private Integer varianceCode;

public void setUid(long uid){
this.uid = uid;
}

public long getUid(){
return this.uid;
}

public void setPackageSid(Integer packageSid){
this.packageSid = packageSid;
}

public Integer getPackageSid(){
return this.packageSid;
}

public void setSwcatUid(long swcatUid){
this.swcatUid = swcatUid;
}

public long getSwcatUid(){
return this.swcatUid;
}

public void setVendorId(String vendorId){
this.vendorId = vendorId;
}

public String getVendorId(){
return this.vendorId;
}

public void setVendorTitle(String vendorTitle){
this.vendorTitle = vendorTitle;
}

public String getVendorTitle(){
return this.vendorTitle;
}

public void setRating(String rating){
this.rating = rating;
}

public String getRating(){
return this.rating;
}

public void setVendorReleaseDate(Timestamp vendorReleaseDate){
this.vendorReleaseDate = vendorReleaseDate;
}

public Timestamp getVendorReleaseDate(){
return this.vendorReleaseDate;
}

public void setReceiveDate(Timestamp receiveDate){
this.receiveDate = receiveDate;
}

public Timestamp getReceiveDate(){
return this.receiveDate;
}

public void setSourceAvailable(String sourceAvailable){
this.sourceAvailable = sourceAvailable;
}

public String getSourceAvailable(){
return this.sourceAvailable;
}

public void setPublishDate(Timestamp publishDate){
this.publishDate = publishDate;
}

public Timestamp getPublishDate(){
return this.publishDate;
}

public void setProductType(String productType){
this.productType = productType;
}

public String getProductType(){
return this.productType;
}

public void setVendor(String vendor){
this.vendor = vendor;
}

public String getVendor(){
return this.vendor;
}

public void setProduct(String product){
this.product = product;
}

public String getProduct(){
return this.product;
}

public void setVersion(String version){
this.version = version;
}

public String getVersion(){
return this.version;
}

public void setReleasename(String releasename){
this.releasename = releasename;
}

public String getReleasename(){
return this.releasename;
}

public void setUnderlyingOs(String underlyingOs){
this.underlyingOs = underlyingOs;
}

public String getUnderlyingOs(){
return this.underlyingOs;
}

public void setArchitecture(String architecture){
this.architecture = architecture;
}

public String getArchitecture(){
return this.architecture;
}

public void setCreatedby(String createdby){
this.createdby = createdby;
}

public String getCreatedby(){
return this.createdby;
}

public void setDatasource(String datasource){
this.datasource = datasource;
}

public String getDatasource(){
return this.datasource;
}

public void setLastmodifiedby(String lastmodifiedby){
this.lastmodifiedby = lastmodifiedby;
}

public String getLastmodifiedby(){
return this.lastmodifiedby;
}

public void setDeleted(String deleted){
this.deleted = deleted;
}

public String getDeleted(){
return this.deleted;
}

public void setAudUser(String audUser){
this.audUser = audUser;
}

public String getAudUser(){
return this.audUser;
}

public void setAudType(String audType){
this.audType = audType;
}

public String getAudType(){
return this.audType;
}

public void setAudTime(Timestamp audTime){
this.audTime = audTime;
}

public Timestamp getAudTime(){
return this.audTime;
}

public void setLastmodifiedtime(Timestamp lastmodifiedtime){
this.lastmodifiedtime = lastmodifiedtime;
}

public Timestamp getLastmodifiedtime(){
return this.lastmodifiedtime;
}

public void setValidFlag(String validFlag){
this.validFlag = validFlag;
}

public String getValidFlag(){
return this.validFlag;
}

public void setVendorUrl(String vendorUrl){
this.vendorUrl = vendorUrl;
}

public String getVendorUrl(){
return this.vendorUrl;
}

public void setSourceUrl(String sourceUrl){
this.sourceUrl = sourceUrl;
}

public String getSourceUrl(){
return this.sourceUrl;
}

public void setPrimaryUrl(String primaryUrl){
this.primaryUrl = primaryUrl;
}

public String getPrimaryUrl(){
return this.primaryUrl;
}

public void setDq(String dq){
this.dq = dq;
}

public String getDq(){
return this.dq;
}

public void setUpdatePending(String updatePending){
this.updatePending = updatePending;
}

public String getUpdatePending(){
return this.updatePending;
}

public void setRatingDate(Timestamp ratingDate){
this.ratingDate = ratingDate;
}

public Timestamp getRatingDate(){
return this.ratingDate;
}

public void setSecurityflag(String securityflag){
this.securityflag = securityflag;
}

public String getSecurityflag(){
return this.securityflag;
}

public void setCrcCode(String crcCode){
this.crcCode = crcCode;
}

public String getCrcCode(){
return this.crcCode;
}

public void setRatingDesignator(String ratingDesignator){
this.ratingDesignator = ratingDesignator;
}

public String getRatingDesignator(){
return this.ratingDesignator;
}

public void setVarianceCode(Integer varianceCode){
this.varianceCode = varianceCode;
}

public Integer getVarianceCode(){
return this.varianceCode;
}

}