Java中Oracle生成實體類的工具代碼

java將ORACLE數(shù)據(jù)庫中的表生成對應(yīng)的實體類


package com.util;

import java.io.FileWriter;

import java.io.IOException;

import java.io.PrintWriter;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.SQLException;

import java.sql.Statement;

/**

* 根據(jù)數(shù)據(jù)庫表結(jié)構(gòu)生成相應(yīng)的JavaBean

*

* 思路

* 1、創(chuàng)建數(shù)據(jù)庫連接

* 2、獲取數(shù)據(jù)庫表、表注釋

* 3、獲取數(shù)據(jù)庫表中字段名、字段類型、字段注釋

* 4、構(gòu)建StringBuffer緩存

* 5、寫入文件

* 6、關(guān)閉連接、輸入流等等

*

* @author ella_li

*

*/

public class GenEntityOracle {

? ? private String packageOutPath = "D:/";// 指定實體生成所在包的路徑

? ? private String tableName = "HOUSE_REGISTER";//數(shù)據(jù)庫中的表名

? ? private String packageName = "com.ufgov.wcx.entity";//實體類所在的包

? ? //private String authorName = "wcx";// 作者名字

? ? private String[] colsPre = { "F_NB_", "F_VC_", "F_CR_", "F_DT_"};

? ? private String[] colnames; // 列名數(shù)組

? ? private String[] colTypes; // 列名類型數(shù)組

? ? private int[] colSizes; // 列名大小數(shù)組

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

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

? ? // 數(shù)據(jù)庫連接

? ? private static final String URL = "jdbc:oracle:thin:@127.0.0.1:1521:orcl";

? ? private static final String NAME = "scott";

? ? private static final String PASS = "1";

? ? private static final String DRIVER = "oracle.jdbc.driver.OracleDriver";

? ? private Connection con = null;

? ? private Statement pStemt = null;

? ? private ResultSet rs = null;

? ? private ResultSetMetaData rsmd = null;

? ? public GenEntityOracle() {

? ? ? ? connect();

? ? ? ? getTableInfo(tableName);

? ? ? ? colseConnect();

? ? }

? ? /**

? ? * 1、創(chuàng)建數(shù)據(jù)庫連接

? ? */

? ? private void connect() {

? ? ? ? try {

? ? ? ? ? ? Class.forName(DRIVER);

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

? ? ? ? ? ? pStemt = (Statement) con.createStatement();

? ? ? ? } catch (ClassNotFoundException | SQLException e) {

? ? ? ? ? ? e.printStackTrace();

? ? ? ? }

? ? }

? ? private void colseConnect() {

? ? ? ? try {

? ? ? ? ? ? if (con != null) {

? ? ? ? ? ? ? ? con.close();

? ? ? ? ? ? ? ? con = null;

? ? ? ? ? ? }

? ? ? ? ? ? if (pStemt != null) {

? ? ? ? ? ? ? ? pStemt.close();

? ? ? ? ? ? ? ? pStemt = null;

? ? ? ? ? ? }

? ? ? ? } catch (SQLException e) {

? ? ? ? ? ? e.printStackTrace();

? ? ? ? }

? ? }

? ? /**

? ? * 獲取單個數(shù)據(jù)庫表信息

? ? */

? ? private void getTableInfo(String tableName) {

? ? ? ? int size = 0;

? ? ? ? String sql = "SELECT * FROM " + tableName;

? ? ? ? try {

? ? ? ? ? ? rs = pStemt.executeQuery(sql);

? ? ? ? ? ? rsmd = rs.getMetaData();

? ? ? ? ? ? size = rsmd.getColumnCount();

? ? ? ? ? ? colnames = new String[size];

? ? ? ? ? ? colTypes = new String[size];

? ? ? ? ? ? colSizes = new int[size];

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

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

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

? ? ? ? ? ? ? ? if (colTypes[i].equalsIgnoreCase("date")

? ? ? ? ? ? ? ? ? ? ? ? || colTypes[i].equalsIgnoreCase("timestamp")) {

? ? ? ? ? ? ? ? ? ? f_util = true;

? ? ? ? ? ? ? ? }

? ? ? ? ? ? ? ? if (colTypes[i].equalsIgnoreCase("blob")

? ? ? ? ? ? ? ? ? ? ? ? || colTypes[i].equalsIgnoreCase("char")) {

? ? ? ? ? ? ? ? ? ? f_sql = true;

? ? ? ? ? ? ? ? }

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

? ? ? ? ? ? }

? ? ? ? ? ? //獲取單張數(shù)據(jù)庫表注釋

? ? ? ? ? ? String tableComment = getTableComment(tableName);

? ? ? ? ? ? //獲取單張數(shù)據(jù)庫表的所有列信息

? ? ? ? ? ? StringBuffer tempSb = getColsInfo(tableName);

? ? ? ? ? ? //生成JavaBean文件

? ? ? ? ? ? genFile(tableName, getSb(tableName, tableComment, tempSb));

? ? ? ? } catch (SQLException e) {

? ? ? ? ? ? e.printStackTrace();

? ? ? ? }

? ? }

? ? /**

? ? * 獲取單張數(shù)據(jù)庫表注釋

? ? *

? ? * @param tableName

? ? * @return

? ? */

? ? private String getTableComment(String tableName) {

? ? ? ? String str = "";

? ? ? ? String sql = "select * from user_tab_comments where table_name = '"

? ? ? ? ? ? ? ? + tableName + "'";

? ? ? ? try {

? ? ? ? ? ? rs = pStemt.executeQuery(sql);

? ? ? ? ? ? while (rs.next()) {

? ? ? ? ? ? ? ? str = rs.getString("comments");

? ? ? ? ? ? ? ? if (null != str && str.indexOf("\r\n") != -1) {

? ? ? ? ? ? ? ? ? ? str = str.replace("\r\n", "");

? ? ? ? ? ? ? ? }

? ? ? ? ? ? ? ? if (null != str && str.indexOf("\n") != -1) {

? ? ? ? ? ? ? ? ? ? str = str.replace("\n", "");

? ? ? ? ? ? ? ? }

? ? ? ? ? ? }

? ? ? ? } catch (SQLException e) {

? ? ? ? ? ? e.printStackTrace();

? ? ? ? }

? ? ? ? return str;

? ? }

? ? /**

? ? * 獲取單張數(shù)據(jù)庫表的所有列信息

? ? *

? ? * @param tableName

? ? */

? ? private StringBuffer getColsInfo(String tableName) {

? ? ? ? StringBuffer temp = new StringBuffer();

? ? ? ? for (int i = 0; i < colnames.length; i++) {

? ? ? ? ? ? temp.append("\tprivate " + getColsType(colTypes[i]) + " "

? ? ? ? ? ? ? ? ? ? + getColsName(colnames[i]) + ";? //"

? ? ? ? ? ? ? ? ? ? + getColComment(tableName, colnames[i]) + "\r\n");

? ? ? ? ? ? String colname = getColsName(colnames[i]);

? ? ? ? ? ? String colnameUp = captureName(colname);

? ? ? ? ? ? temp.append("\n\tpublic void set" + colnameUp + "("

? ? ? ? ? ? ? ? ? ? + getColsType(colTypes[i]) + " " + colname + "){\r\n");

? ? ? ? ? ? temp.append("\t\tthis." + colname + "=" + colname + ";\r\n");

? ? ? ? ? ? temp.append("\t}\r\n");

? ? ? ? ? ? temp.append("\n\tpublic " + getColsType(colTypes[i]) + " get"

? ? ? ? ? ? ? ? ? ? + colnameUp + "(){\r\n");

? ? ? ? ? ? temp.append("\t\treturn " + colname + ";\r\n");

? ? ? ? ? ? temp.append("\t}\r\n");

? ? ? ? }

? ? ? ? return temp;

? ? }

? ? /**

? ? * 獲取列類型

? ? *

? ? * @param sqlType

? ? * @return

? ? */

? ? private String getColsType(String sqlType) {

? ? ? ? if (sqlType.equalsIgnoreCase("binary_double")) {

? ? ? ? ? ? return "double";

? ? ? ? } else if (sqlType.equalsIgnoreCase("binary_float")) {

? ? ? ? ? ? return "float";

? ? ? ? } else if (sqlType.equalsIgnoreCase("blob")) {

? ? ? ? ? ? return "byte[]";

? ? ? ? } else if (sqlType.equalsIgnoreCase("blob")) {

? ? ? ? ? ? return "byte[]";

? ? ? ? } else if (sqlType.equalsIgnoreCase("char")

? ? ? ? ? ? ? ? || sqlType.equalsIgnoreCase("nvarchar2")

? ? ? ? ? ? ? ? || sqlType.equalsIgnoreCase("varchar2")) {

? ? ? ? ? ? return "String";

? ? ? ? } else if (sqlType.equalsIgnoreCase("date")

? ? ? ? ? ? ? ? || sqlType.equalsIgnoreCase("timestamp")

? ? ? ? ? ? ? ? || sqlType.equalsIgnoreCase("timestamp with local time zone")

? ? ? ? ? ? ? ? || sqlType.equalsIgnoreCase("timestamp with time zone")) {

? ? ? ? ? ? return "Date";

? ? ? ? } else if (sqlType.equalsIgnoreCase("number")) {

? ? ? ? ? ? return "Long";

? ? ? ? }

? ? ? ? return "String";

? ? }

? ? /**

? ? * 獲取列名

? ? *

? ? * @param str

? ? * @return

? ? */

? ? private String getColsName(String str) {

? ? ? ? for (String temp : colsPre) {

? ? ? ? ? ? int preIndex = str.indexOf(temp);

? ? ? ? ? ? if (preIndex >= 0) {

? ? ? ? ? ? ? ? str = str.substring(preIndex + temp.length());

? ? ? ? ? ? ? ? str = str.replace("_", "").toLowerCase();

? ? ? ? ? ? }

? ? ? ? }

? ? ? ? return str;

? ? }

? ? /**

? ? * 獲取列注釋

? ? *

? ? * @param tableName

? ? * @param columnName

? ? * @return

? ? */

? ? private String getColComment(String tableName, String columnName) {

? ? ? ? String str = "";

? ? ? ? String sql = "select comments from USER_COL_COMMENTS where table_name= '"

? ? ? ? ? ? ? ? + tableName + "' and column_name= '" + columnName + "'";

? ? ? ? try {

? ? ? ? ? ? rs = pStemt.executeQuery(sql);

? ? ? ? ? ? while (rs.next()) {

? ? ? ? ? ? ? ? str = rs.getString("comments");

? ? ? ? ? ? ? ? if (null != str && str.indexOf("\r\n") != -1) {

? ? ? ? ? ? ? ? ? ? str = str.replace("\r\n", "");

? ? ? ? ? ? ? ? }

? ? ? ? ? ? ? ? if (null != str && str.indexOf("\n") != -1) {

? ? ? ? ? ? ? ? ? ? str = str.replace("\n", "");

? ? ? ? ? ? ? ? }

? ? ? ? ? ? }

? ? ? ? } catch (SQLException e) {

? ? ? ? ? ? e.printStackTrace();

? ? ? ? }

? ? ? ? return str;

? ? }

? ? /**

? ? * 構(gòu)建StringBuffer緩存

? ? *

? ? * @param tableName

? ? * @param tableComment

? ? * @param colSb

? ? * @return

? ? */

? ? private StringBuffer getSb(String tableName, String tableComment,

? ? ? ? ? ? StringBuffer colSb) {

? ? ? ? StringBuffer sb = new StringBuffer();

? ? ? ? sb.append("package " + this.packageName + ";\r\n");

? ? ? ? // 判斷是否導入工具包

? ? ? ? if (f_util) {

? ? ? ? ? ? sb.append("import java.util.Date;\r\n");

? ? ? ? }

? ? ? ? if (f_sql) {

? ? ? ? ? ? sb.append("import java.sql.*;\r\n");

? ? ? ? }

? ? ? ? sb.append("\r\n");

? ? ? ? // 注釋部分

? ? ? ? //sb.append("? /**\r\n");

? ? ? ? //sb.append("? ? * " + tableName + " 實體類? ? ? ? ? ? ? " + tableComment + "\r\n");

? ? ? ? //sb.append("? ? * " + new Date() + "\r\n");

? ? ? ? //sb.append("? ? * @author " + this.authorName + "\r\n");

? ? ? ? //sb.append("? ? */ \r\n");

? ? ? ? // 實體部分

? ? ? ? sb.append("\r\n\r\npublic class " + tableName + "{\r\n");

? ? ? ? sb.append("\tprivate static final long serialVersionUID = 1L;\r\n");

? ? ? ? sb.append(colSb);

? ? ? ? sb.append("}\r\n");

? ? ? ? return sb;

? ? }

? ? /**

? ? * 生成JavaBean文件

? ? *

? ? * @param content

? ? */

? ? private void genFile(String tableName, StringBuffer content) {

? ? ? ? try {

? ? ? ? ? ? String outputPath = this.packageOutPath + tableName

? ? ? ? ? ? ? ? ? ? + ".java";

? ? ? ? ? ? FileWriter fw = new FileWriter(outputPath);

? ? ? ? ? ? PrintWriter pw = new PrintWriter(fw);

? ? ? ? ? ? pw.println(content);

? ? ? ? ? ? pw.flush();

? ? ? ? ? ? pw.close();

? ? ? ? } catch (IOException e) {

? ? ? ? ? ? e.printStackTrace();

? ? ? ? }

? ? }

? ? ? ? /**

? ? * 首字母大寫

? ? * @param name

? ? * @return

? ? */

? ? public static String captureName(String name) {

? ? ? ? char[] cs = name.toCharArray();

? ? ? ? //首字母是小寫的需要大寫

? ? ? ? if(cs[0] >= 97 && cs[0] <= 122) {

? ? ? ? ? ? cs[0] -= 32;

? ? ? ? }

? ? ? ? return String.valueOf(cs);

? ? }

? ? /**

? ? * 出口 TODO

? ? *

? ? * @param args

? ? */

? ? public static void main(String[] args) {

? ? ? ? new GenEntityOracle();?

? ? }

}

最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容