根据数据库生成实体类 如何根据数据库中的表自动生成实体类
大家好,如果您还对根据数据库生成实体类不太了解,没有关系,今天就由本站为大家分享根据数据库生成实体类的知识,包括如何根据数据库中的表自动生成实体类的问题都会给大家分析到,还望可以解决大家的问题,下面我们就开始吧!
如何通过java反射将数据库表生成实体类
首先有几点声明:
1、代码是在别人的基础进行改写的;
2、大家有什么改进的意见可以告诉我,也可以自己改好共享给其他人;
3、刚刚毕业,水平有限,肯定有许多不足之处;
4、希望刚刚学习java的同学能有所启发。
//这个是做转换的类,里面的DB只是封装了数据库的连接,大家可以用自己的,随意
package com.tt.util.gen.entity.tool;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Properties;
import com.tt.util.DB;
public class GenEntityMysql{
private String packageOutPath;//指定实体生成所在包的路径
private String authorName;//作者名字
private String tablename;//表名
private String databasename;//数据库名
private List<String> tablenames;//拿到对应数据库中所有的实体类(实体类需要与其他表明做区分)
private List<String> colnames;//列名集合
private List<String> colTypes;//列名类型集合
private boolean f_util= false;//是否需要导入包java.util.*
private boolean f_sql= false;//是否需要导入包java.sql.*
/*
*构造函数
*/
public GenEntityMysql(){
//使用properties读取配置文件
Properties prop= new Properties();
try{
InputStream genentity= getClass().getResourceAsStream(
"/genentity.properties");
prop.load(genentity);
if(genentity!= null){
genentity.close();
}
} catch(Exception e){
System.out.println("file"+"catalogPath.properties"
+" not found!\n"+ e);
}
this.databasename= prop.getProperty("databasename").toString();
this.tablename= prop.getProperty("tablename").toString();
this.packageOutPath= prop.getProperty("packageOutPath").toString();
this.authorName= prop.getProperty("authorName").toString();
}
//创建多个实体类
private void genEntity(List<String> tablenames, Connection conn){
//使用第归生成文件
for(String tablename: tablenames){
this.genEntity(tablename, conn);
}
}
//创建单个实体类
private void genEntity(String tablename, Connection conn){
String sql="select* from"+ tablename;
PreparedStatement pstmt= null;
ResultSetMetaData rsmd= null;
try{
pstmt= DB.getPStmt(conn, sql);
rsmd= pstmt.getMetaData();
int size= rsmd.getColumnCount();//统计列
colnames= new ArrayList<String>();
colTypes= new ArrayList<String>();
for(int i= 0; i< size; i++){
colnames.add(rsmd.getColumnName(i+ 1));
colTypes.add(rsmd.getColumnTypeName(i+ 1));
if(colTypes.get(i).equalsIgnoreCase("datetime")){
f_util= true;
}
if(colTypes.get(i).equalsIgnoreCase("image")
|| colTypes.get(i).equalsIgnoreCase("text")){
f_sql= true;
}
}
System.out.println(colnames);
System.out.println(colTypes);
} catch(SQLException e1){
// TODO Auto-generated catch block
e1.printStackTrace();
} finally{
DB.close(pstmt);
}
//在内存中生成代码
String content= parse(tablename);
//写入到文件中
try{
File directory= new File("");
String outputPath= directory.getAbsolutePath()+"/src/"
+ this.packageOutPath.replace(".","/")+"/";
System.out.println("写出的路径:"+ outputPath);
//检测路径是否存在,不存在就创建路径
File path= new File(outputPath);
if(!path.exists()&&!path.isDirectory()){
path.mkdir();
System.out.println(path.exists());
}
//创建文件
outputPath+= initcap(tablename)+".java";
File file= new File(outputPath);
if(!file.exists()){
file.createNewFile();
}
//写出到硬盘
FileWriter fw= new FileWriter(file);
PrintWriter pw= new PrintWriter(fw);
pw.println(content);
pw.flush();
pw.close();
} catch(IOException e){
e.printStackTrace();
}
}
private void getAllEntityTable(Connection conn, List<String> tablenames){
ResultSet rs= null;
try{
DatabaseMetaData dmd=(DatabaseMetaData) conn.getMetaData();
/*
* TABLE_CAT String=>表类别(可为 null)
* TABLE_SCHEM String=>表模式(可为null)
* TABLE_NAME String=>表名称
* TABLE_TYPE String=>表类型
*/
rs= dmd.getTables(null, null,"%", null);
while(rs.next()){
tablenames.add(rs.getString("TABLE_NAME"));
}
} catch(SQLException e){
e.printStackTrace();
} finally{
try{
rs.close();
} catch(SQLException e){
e.printStackTrace();
}
}
}
/**
*
*@param tablename
*@return
*/
private String parse(String tablename){
StringBuffer sb= new StringBuffer();
//判断是否导入工具包
if(f_util){
sb.append("import java.util.Date;\r\n");
}
if(f_sql){
sb.append("import java.sql.*;\r\n");
}
sb.append("package"+ this.packageOutPath+";\r\n");
sb.append("\r\n");
//注释部分
sb.append("/**\r\n");
sb.append("*"+ tablename+"实体类\r\n");
sb.append("*"+ new Date()+""+ this.authorName+"\r\n");
sb.append("*/\r\n");
//实体部分
sb.append("\r\n\r\npublic class"+ initcap(tablename)+"{\r\n");
processAllAttrs(sb);//属性
processAllMethod(sb);// get set方法
sb.append("}\r\n");
return sb.toString();
}
/**
*功能:生成所有属性
*
*@param sb
*/
private void processAllAttrs(StringBuffer sb){
for(int i= 0; i< colnames.size(); i++){
sb.append("\tprivate"+ sqlType2JavaType(colTypes.get(i))+""
+ colnames.get(i)+";\r\n");
}
}
/**
*功能:生成所有方法
*
*@param sb
*/
private void processAllMethod(StringBuffer sb){
for(int i= 0; i< colnames.size(); i++){
sb.append("\tpublic void set"+ initcap(colnames.get(i))+"("
+ sqlType2JavaType(colTypes.get(i))+""+ colnames.get(i)
+"){\r\n");
sb.append("\t\tthis."+ colnames.get(i)+"="+ colnames.get(i)
+";\r\n");
sb.append("\t}\r\n");
sb.append("\tpublic"+ sqlType2JavaType(colTypes.get(i))+" get"
+ initcap(colnames.get(i))+"(){\r\n");
sb.append("\t\treturn"+ colnames.get(i)+";\r\n");
sb.append("\t}\r\n");
}
}
/**
*功能:将输入字符串的首字母改成大写
*
*@param str
*@return
*/
private String initcap(String str){
char[] ch= str.toCharArray();
if(ch[0]>='a'&& ch[0]<='z'){
ch[0]=(char)(ch[0]- 32);
}
return new String(ch);
}
/**
*功能:获得列的数据类型
*
*@param sqlType
*@return
*/
private String sqlType2JavaType(String sqlType){
if(sqlType.equalsIgnoreCase("bit")){
return"boolean";
} else if(sqlType.equalsIgnoreCase("tinyint")){
return"byte";
} else if(sqlType.equalsIgnoreCase("smallint")){
return"short";
} else if(sqlType.equalsIgnoreCase("int")){
return"int";
} else if(sqlType.equalsIgnoreCase("bigint")){
return"long";
} else if(sqlType.equalsIgnoreCase("float")){
return"float";
} else if(sqlType.equalsIgnoreCase("decimal")
|| sqlType.equalsIgnoreCase("numeric")
|| sqlType.equalsIgnoreCase("real")
|| sqlType.equalsIgnoreCase("money")
|| sqlType.equalsIgnoreCase("smallmoney")){
return"double";
} else if(sqlType.equalsIgnoreCase("varchar")
|| sqlType.equalsIgnoreCase("char")
|| sqlType.equalsIgnoreCase("nvarchar")
|| sqlType.equalsIgnoreCase("nchar")
|| sqlType.equalsIgnoreCase("text")){
return"String";
} else if(sqlType.equalsIgnoreCase("datetime")){
return"Date";
} else if(sqlType.equalsIgnoreCase("image")){
return"Blod";
}
return null;
}
/**
*出口 TODO
*
*@param args
*/
public static void main(String[] args){
new GenEntityMysql().start();
}
private void start(){
//创建连接
Connection conn= DB.getConn();
if(databasename!= null&&!databasename.equals("")
&& tablename!= null&&!tablename.equals("")){
System.out.println("databasename和 tablename不能同时存在");
} else{
//如果配置文件中有数据库名字,则可以拿到其中所有的实体类
if(databasename!= null&&!databasename.equals("")){
//获取所有实体表名字
tablenames= new ArrayList<String>();
getAllEntityTable(conn, tablenames);
System.out.println(tablenames);
//为每个实体表生成实体类
genEntity(tablenames, conn);
} else{
//为指定实体表生成实体类
genEntity(tablename, conn);
}
//关闭数据库连接
if(conn!= null){
DB.close(conn);
}
}
}
}
如何根据数据库中的表自动生成实体类
最好的做法是每张表对应一个实体类bean,如果一张表关联其他表,把这张表对应的实体类加上就可以了。如果使用了Hibernate,就可以使用配置文件进行映射。
A.实体类就是属性类,通常定义在model层里面。
B.一般的实体类对应一个数据表,其中的属性对应数据表中的字段,这样做对数据库操作起来方便。
这样做的好处:
1.对对象实体的封装,体现OO思想。
2.属性可以对字段定义和状态进行判断和过滤
3.把相关信息用一个实体类封装后,我们在程序中可以把实体类作为参数传递,更加方便。
C.说白了就是为了让程序员在对数据库操作的时候不用写SQL语句
D.就是一个数据库表生成一个类
代码生成器是什么如何根据数据库去产生实体类
我们在编写代码的过程中会发现有很多代码是有规律的,编写起来无味、繁琐。
这个时候我们可以根据规律来编写一些程序来生成这些代码,这个生成代码的程序就是代码生成器。
现阶段代码生成器有很多,其中我个人认为CodeSmith的功能最为强大。基本上你想要生成什么样的代码都可以根据模版来生成,模版有自带的,有别人编写的,当然,你觉得不够,也可以自己编写模版。
至于如何根据数据库生成实体类。他的原理是数据库中的字段就是将来生成实体类中的私有变量和属性。
所以我们可以编写模版来读取数据库中的字段,然后来根据数据库中的字段来生成你所需要的变量和属性。
至于Codesmith的具体使用,比较简单,可以到网上搜索一下,有很多的介绍。
关于根据数据库生成实体类的内容到此结束,希望对大家有所帮助。