利用Java将Mysql数据表生成JPA实体对象_MySQL

package cn.lry.xp;

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;

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;

/**

* 从数据库表反射出实体类，自动生成实体类

* @author lry

*

*/

public class GenEntityMysql {

private String packageOutPath = "cn.lry.po";//指定实体生成所在包的路径

private String authorName = "lry";//作者名字

private String tablename = "";//表名

private String[] colnames; // 列名数组

private String[] colTypes; //列名类型数组

private int[] colSizes; //列名大小数组

private boolean f_util = false; // 是否需要导入包java.util.*

private boolean f_sql = false; // 是否需要导入包java.sql.*

private boolean f_jpa = true; // 是否需要生成基于注解的JPA实体对象

//数据库连接

private static final String URL ="jdbc:mysql://localhost:3306/mytest";

private static final String NAME = "root";

private static final String PASS = "123456";

private static final String DRIVER ="com.mysql.jdbc.Driver";

/*

* 构造函数

*/

public GenEntityMysql(){

List list=getTableName();

for(int p=0;p

tablename=list.get(p);

//创建连接

Connection con;

//查要生成实体类的表

String sql = "select * from " + tablename;

PreparedStatement pStemt = null;

try {

try {

Class.forName(DRIVER);

} catch (ClassNotFoundException e1) {

e1.printStackTrace();

}

con = DriverManager.getConnection(URL,NAME,PASS);

pStemt = con.prepareStatement(sql);

ResultSetMetaData rsmd = pStemt.getMetaData();

int size = rsmd.getColumnCount();//统计列

colnames = new String[size];

colTypes = new String[size];

colSizes = new int[size];

for (int i = 0; i

colnames[i] = rsmd.getColumnName(i + 1);

colTypes[i] = rsmd.getColumnTypeName(i + 1);

if(colTypes[i].equalsIgnoreCase("datetime")){

f_util = true;

}

if(colTypes[i].equalsIgnoreCase("image") || colTypes[i].equalsIgnoreCase("text")){

f_sql = true;

}

colSizes[i] = rsmd.getColumnDisplaySize(i + 1);

}

String content = parse(colnames,colTypes,colSizes);

try {

File directory = new File("");

String outputPath = directory.getAbsolutePath()+ "/src/"+this.packageOutPath.replace(".", "/")+"/"+initcap(tablename) + ".java";

FileWriter fw = new FileWriter(outputPath);

PrintWriter pw = new PrintWriter(fw);

pw.println(content);

pw.flush();

pw.close();

} catch (IOException e) {

e.printStackTrace();

}

} catch (SQLException e) {

e.printStackTrace();

} finally{

//try {

//con.close();

/