如何通过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 = 0i <sizei++) {

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 = 0i <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 = 0i <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)

}

}

}

}

1.读取数据：从数据库中读取数据的同时，拿到元数据，即metadata

2.实例化：通过类名，利用反射 实例你的实体类(类名可以和表名保持一致，这样就不用写额外的

映射了)，如果你只是想通过反射赋值的话，你也可以直接new个类；

3.方法调用：通过元数据，利用反射 得到你的set方法(方法名可以和字段名保持一致，否则你要写

出映射关系)；