通过java代码读取数据库表反向生成相应的pojo类

最新推荐文章于 2021-03-14 22:49:01 发布

weixin_33701617

最新推荐文章于 2021-03-14 22:49:01 发布

阅读量191

点赞数

文章标签：数据库 java python

原文链接：https://my.oschina.net/linuxred/blog/30368

版权

2019独角兽企业重金招聘Python工程师标准>>>

下午要根据数据库的一个表写一个pojo，那个累啊，80列名。写了大概10列就没心情写了。于是就想到通过数据库反向生成java类，但没有 myeclipse又不懂生成，于是手动写了一个大概的方法，暂时仅仅在sqlserver测试通过，对于包名跟引用类就没写了，把代码复制到 eclipse直接ctrl+shift+o就导入了

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;

import com.onedear.util.database.DataBaseType;

/**
* 数据库工具类
*
* @author onedear
* @data:2010-10-21 下午06:12:39
*/
public class DBUtil {高中英语重点语法有哪些

    public static final int SQLSERVER = 1;
    public static final int MYSQL = 2;
    public static final int ORACLE = 3;
    private static final String LINE = "\r\n";
    private static final String TAB = "\t";欠款

    /**
    * 用于调试,懒得写相应的参数
    *
    * @return
    * @throws SQLException
    * @throws ClassNotFoundException
    */
    public static Connection getConnection() throws ClassNotFoundException, SQLException {
            return getConnection("192.168.0.161:1997", "ETForMonitor_V2", "sa",
                    "password", DBUtil.SQLSERVER);
    }

    /**
    *
    * 通过jdbc获取相应的数据库链接connection
    *
    * @param ipport
    *            ip+port ,eg.: 192.168.0.161:1997
    * @param dbName
    *            databaseName ,eg. : ETForMonitor_V2
    * @param username
    *            eg.:sa
    * @param password
    *            eg. :password
    * @param type
    *            请看本类的静态变量
    * @return
    * @throws ClassNotFoundException
    * @throws SQLException
    */
    public static Connection getConnection(String ipport, String dbName,
            String username, String password, int type)
            throws ClassNotFoundException, SQLException {
        String jdbcString = null;
        if (type == SQLSERVER) {
            jdbcString = "jdbc:jtds:sqlserver://" + ipport + ";databaseName="
                    + dbName;
            Class.forName("net.sourceforge.jtds.jdbc.Driver");
        } else if (type == MYSQL) {
            jdbcString = "jdbc:mysql://" + ipport + "/" + dbName;
            Class.forName("org.gjt.mm.mysql.Driver");
        } else if (type == ORACLE) {
            jdbcString = "jdbc:oracle:thin:@" + ipport + ":" + dbName;
            Class.forName("oracle.jdbc.driver.OracleDriver");
        }

        Connection connection = null;
        connection = DriverManager
                .getConnection(jdbcString, username, password);
        return connection;
    }

    /**
    * 数据库表生成相应的java类，生成规则
    * 类名=        表名（第一个字母大写）
    * 属性名=   数据库列名
    * get/set方法 = 根据标准生成
    * 其中生成的基本类型均为包装类，例如Integer , Long , Boolean , String
    * @param connection
    * @param tableName
    * @param dbType
    * @param path
    * @param isCreateFile
    * @return
    * @throws SQLException
    */
    public static String table2pojo(Connection connection, String tableName,
            int dbType, String path , boolean isCreateFile) throws SQLException {
        String sql = "select * from " + tableName + " where 1 <> 1";
        PreparedStatement ps = null;
        ResultSet rs = null;
        ps = connection.prepareStatement(sql);
        rs = ps.executeQuery();
        ResultSetMetaData md = rs.getMetaData();
        int columnCount = md.getColumnCount();

        StringBuffer sb = new StringBuffer();
        tableName = tableName.substring(0, 1).toUpperCase() +tableName.subSequence(1, tableName.length());
        sb.append("public class " + tableName + " {");
        sb.append(LINE);

        for (int i = 1; i <= columnCount; i++) {
            sb.append(TAB);
            sb.append("private "
                    + DataBaseType.getPojoType(md.getColumnTypeName(i)) + " "
                    + md.getColumnName(i) + ";");
            // System.out.println("name : " + md.getColumnName(i) +
            // "   , type :"
            // + md.getColumnTypeName(i));
            sb.append(LINE);
        }

        for (int i = 1; i <= columnCount; i++) {
            sb.append(TAB);

            String pojoType = DataBaseType.getPojoType(md.getColumnTypeName(i));
            String columnName = md.getColumnName(i);
            String getName = null;
            String setName = null;
            if (columnName.length() > 1) {
                getName = "public "+pojoType+" get" + columnName.substring(0, 1).toUpperCase()
                        + columnName.substring(1, columnName.length()) + "() {";
                setName = "public void set" + columnName.substring(0, 1).toUpperCase()
                        + columnName.substring(1, columnName.length()) + "("
                        + pojoType + " " + columnName + ") {";
            } else {
                getName = "public get" + columnName.toUpperCase() + "() {";
                setName = "public set" + columnName.toUpperCase() + "(" + pojoType
                        + " " + columnName + ") {";
            }

            sb.append(LINE).append(TAB).append(getName);
            sb.append(LINE).append(TAB).append(TAB);
            sb.append("return " + columnName +";");
            sb.append(LINE).append(TAB).append("}");
            sb.append(LINE);
            sb.append(LINE).append(TAB).append(setName);
            sb.append(LINE).append(TAB).append(TAB);
            sb.append("this." + columnName + " = " + columnName +";" );
            sb.append(LINE).append(TAB).append("}");
            sb.append(LINE);

        }
        sb.append("}");

        System.out.println(sb.toString());

        if(isCreateFile)
            FileUtils.stringToFile(null,tableName +".java" , sb.toString());
        return null;
    }

    public static void main(String[] args) throws SQLException, ClassNotFoundException {
        Connection con = getConnection();
        table2pojo(con, "PlayBackVisitorControl", DBUtil.SQLSERVER, "" , true);

    }

}