jfinal java类型和数据库类型_JFinal 连接数据库生成Model类

很喜欢JFinal 框架的，简单，不用像SSH那样乱七八糟的注入，配置。不过用惯了Hibernate，Eclipse 可以直接连接数据库，批量生成表对象类。嘿嘿……“懒惰”驱使，那就自己写一个吧，肯定以后的学习中，JFinal是很有用的，就写了这个CreateJfinalEntityUtil 工具类。用的是POSTGRESQL数据库，这个数据库非常不错，呵呵……写的不好，大家勿喷。

package xidian.wwf.utils;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.ArrayList;

import java.util.List;

import java.util.Locale;

/**

* 映射数据库，自动生成Entity

* @author WWF

*/

public class CreateJfinalEntityUtil {

private static Connection conn = null;

private static final String URL;

private static final String JDBC_DRIVER;

private static final String USER_NAME;

private static final String PASSWORD;

private static final String DATABASENAME;

private static final String PACKAGENAME;

static {

DATABASENAME = "数据库名";

URL = "jdbc:postgresql://localhost:5432/"+DATABASENAME;

JDBC_DRIVER = "org.postgresql.Driver";

USER_NAME = "数据库帐号";

PASSWORD = "密码";

PACKAGENAME = "xidian.wwf.entity";

}

/**

* 获得连接

* @return

*/

public static Connection getConnection() {

try {

Class.forName(JDBC_DRIVER);

conn = DriverManager.getConnection(URL, USER_NAME, PASSWORD);

} catch (ClassNotFoundException e) {

e.printStackTrace();

} catch (SQLException e) {

e.printStackTrace();

}

return conn;

}

/**

* 关闭数据库

*/

public static void closeConnection(){

try {

if (conn != null && !conn.isClosed()) {

conn.close();

}

} catch (Exception e) {

e.printStackTrace();

}

}

/**

* 生成字段静态声明

* @param connection

* @param tableName

* @return

*/

private static String CreateEntityString(String tableName){

String result = "package "+PACKAGENAME+";\n\n";

result += "import com.jfinal.plugin.activerecord.Model; \n\n";

result += "public class "+StringUtil.toLowerCaseTheFristChar(tableName)+" extends Model{\n\n";

result += " private static final long serialVersionUID = 1L;\n";

result += " public static final "+StringUtil.toLowerCaseTheFristChar(tableName)+" dao = new "+StringUtil.toLowerCaseTheFristChar(tableName)+"();\n\n";

result += " /**表名**/ \n" ;

result += " public static String TABLE = \""+tableName+"\";" ;

String sql = "select column_name from information_schema.columns where table_name = '"+tableName+"';";

try {

Statement statement = conn.createStatement();

ResultSet resultSet = statement.executeQuery(sql);

while (resultSet.next()) {

if (resultSet.getString(1)!=null&&!resultSet.getString(1).isEmpty()) {

String string = resultSet.getString(1);

String row = " public static final String "+ string.toUpperCase(Locale.CHINA) +" = \""+string+"\";";

String note = " /****/";

result += "\n"+note + "\n" +row;

}

}

resultSet.close();

statement.close();

result+="\n }";

return result;

} catch (SQLException e) {

e.printStackTrace();

return null;

}

}

/**

* 获得数据库的所有表名

* @return

*/

public static List getAllTables(){

String sql = "select relname as TABLE_NAME from pg_class c " +

"where relkind = 'r' and relname not like 'pg_%' and relname not like 'sql_%' order by relname";

try {

List result = new ArrayList();

Statement statement = conn.createStatement();

ResultSet resultSet = statement.executeQuery(sql);

while (resultSet.next()) {

if (resultSet.getString(1)!=null&&!resultSet.getString(1).isEmpty()){

result.add(resultSet.getString(1));

}

}

return result;

} catch (Exception e) {

e.printStackTrace();

return null;

}

}

/**

* 生成Entity

*/

public static void CreateEntity(){

try {

getConnection();

List tables = getAllTables();

for (int i = 0; i < tables.size(); i++) {

File createFolder = new File(System.getProperty("user.dir")+"/src/"+PACKAGENAME.replace(".", "/"));

//路径不存在，生成文件夹

if (!createFolder.exists()) {

createFolder.mkdirs();

}

String entityString = CreateEntityString(tables.get(i).trim());

File entity = new File(createFolder.getAbsolutePath()+"/"+StringUtil.toLowerCaseTheFristChar(tables.get(i))+".java");

if (!entity.exists())

{

//写入文件

BufferedWriter out = new BufferedWriter(new FileWriter(entity, true));

out.write(entityString);

out.close();

out = null;

entity = null;

}

}

closeConnection();

System.out.println("生成成功");

} catch (Exception e) {

e.printStackTrace();

}

}

public static void main(String[] args) {

CreateEntity();

}

}