【学习笔记】--java通过数据库表名自动生成实体类

//生产类

package com.yangtb.creatclass;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.ParameterMetaData;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;
/**
* 类：创建实体
* 通过表名，创建一个实体
* @author Yang 杨天兵
*时间：20090409
*/
public class CreatClass extends Sql2000 {

//包名
String strpackage ="";
// 数据库表名
String tableName="";

StringBuffer sb = new StringBuffer();
Connection con = null;

ResultSet res = null;
ResultSetMetaData rsmd =null;

public String getTableName() {
return tableName;
}

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

public String getStrpackage() {
return strpackage;
}

public void setStrpackage(String strpackage) {
this.strpackage = strpackage;
}

public void execute() throws IOException{
con = getCoonnection();
String strsql="select * from "+tableName;
Statement st;
try {

st = con.createStatement();

res = st.executeQuery(strsql);

rsmd = res.getMetaData();
String newstrpack = "";
newstrpack = strpackage.replace(".", "/");
System.out.println(newstrpack);
//类名
String className = tableName.substring(0, 1).toUpperCase()+tableName.substring(1);
//创建这个类文件strpackage
File file = new File("src"+"/"+newstrpack+"/"+className+".java");
sb.append("package "+strpackage+";/n");
sb.append("");
sb.append("/*/n *");
sb.append("这是一个类：/n *@author yangtianb/n *//n");
sb.append("public class "+className+" {");

sb.append("/n");
for(int i=1;i<rsmd.getColumnCount();i++){
String type = typename(rsmd.getColumnTypeName(i));
//打印属性
sb.append("/tprivate/t");
sb.append(type+"/t");
sb.append(rsmd.getColumnName(i)+";");
sb.append("/n");
//打印get方法
sb.append("/t//get方法/n");
sb.append("/tpublic/t"+type+"/tget"+rsmd.getColumnName(i)+"(){/n");
sb.append("/t/treturn "+rsmd.getColumnName(i)+";/n/t/n");
sb.append("}/n");
//打印set方法
sb.append("/t//set方法/n");
//
sb.append("/tpublic/t"+type+"/tset"+rsmd.getColumnName(i)+"("+type+"/t"+rsmd.getColumnName(i)+")"+"{/n");
sb.append("/treturn/tthis."+rsmd.getColumnName(i)+"/t = "+rsmd.getColumnName(i)+";/n");
sb.append("}/n");
}
sb.append("}");
String strsb = sb.toString();
//创建一个FileWriter对象
FileWriter fw = new FileWriter(file);
//创建一个BufferedWriter对象
BufferedWriter bw = new BufferedWriter(fw);
bw.write(strsb);
bw.newLine();
//关闭文件流
bw.flush();
fw.close();

} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}finally{
//关闭数据库连接
closeResultSet(res);
closeConneciton(con);
}

System.out.println(sb);
}

public String typename(String typename){

String tystr=null;
if(typename.equals("varchar")){
tystr="String";
}
if(typename.equals("int")){
tystr="int";
}
else{
tystr="Object";
}
return tystr;
}

}

在上面工程中的类，以后在其他工程中导入这个类，在写入xml传入参数就能得到相应数据库表名的实体

//build.xml

<?xml version="1.0"?>

<project name="project" default="default">

<target name="default" depends="depends" description="--> description">
<taskdef name="test" classname="com.yangtb.creatclass.CreatClass"></taskdef>
<test tableName="Bloginfo" strpackage="entity"></test>
</target>
<target name="depends">
</target>
</project>

记得导入