通过数据表自动生成实体

àì夳堔傛蜴生んèń 2022-05-26 13:47 242阅读 0赞

import java.sql.*;
    import java.text.ParseException;
    import java.text.SimpleDateFormat;
    import java.util.*;
    import java.util.Date;
    
    /**
     * @author liaofh
     * @version 1.0  2016/5/20
     * @modified liaofh  2016/5/20  <创建>
     */
    public class DB2JavaBean {
        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";
    
    
        private static Map<String, String> map;
    
        static {
            map = new HashMap<String, String>();
            map.put("BIGINT", "String");
            map.put("CHAR", "String");
            map.put("DATE", "Date");
            map.put("DECIMAL", "String");
            map.put("DOUBLE", "String");
            map.put("FLOAT", "String");
            map.put("INTEGER", "String");
    
            map.put("NUMBER", "String");
    //      map.put("NUMERIC", "String");
            map.put("SMALLINT", "String");
            map.put("TIME", "String");
            map.put("DATETIME", "Date");
            map.put("TIMESTAMP", "String");
    
            map.put("VARCHAR2", "String");
            map.put("VARCHAR", "String");
            map.put("BLOB", "byte[]");
        }
    
        public static String getPojoType(String dataType) {
            String tmp = dataType.toUpperCase();
            StringTokenizer st = new StringTokenizer(tmp);
            return map.get(st.nextToken());
        }
    
        /**
         * 用于调试,懒得写相应的参数
         *
         * @return
         * @throws java.sql.SQLException
         * @throws ClassNotFoundException
         */
        public static Connection getConnection() throws ClassNotFoundException, SQLException {
            //return getConnection("10.83.2.134:1521", "BMS_Plus", "sa", "123", 1);
        	return getConnection("127.0.0.1:1521", "root", "BMS_Plus", "BMS_Plus", 3);
        }
    
    
        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:sqlserver://10.83.2.134:1433;DatabaseName=BMS_Plus"; //jdbc:jtds:sqlserver://" + ipport + ";databaseName="
                     //   + dbName;
                Class.forName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
            } 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;
        }
    
    
        public static String toLow0(String str) {
            char ch[] = str.toCharArray();
            for (int i = 0; i < ch.length; i++) {
                if (i == 0) {
                    ch[i] = Character.toLowerCase(ch[i]);
                }
                if ((i + 1) < ch.length) {
                    if (ch[i] == '_') {
                        ch[i + 1] = Character.toUpperCase(ch[i + 1]);
                    } else {
                        ch[i + 1] = Character.toLowerCase(ch[i + 1]);
                    }
                }
            }
            return new String(ch).replace("_", "");
        }
    
        public static String column2JavaAttribute(String str) {
            char ch[] = str.toCharArray();
            for (int i = 0; i < ch.length; i++) {
                if (i == 0) {
                    ch[i] = Character.toUpperCase(ch[i]);
                }
                if ((i + 1) < ch.length) {
                    if (ch[i] == '_') {
                        ch[i + 1] = Character.toUpperCase(ch[i + 1]);
                    } else {
                        ch[i + 1] = Character.toLowerCase(ch[i + 1]);
                    }
                }
            }
            return new String(ch).replace("_", "");
        }
    
        /**
         * 数据库表生成相应的java类，生成规则
         * 类名=      表名（第一个字母大写）
         * 属性名=   数据库列名
         * get/set方法 = 根据标准生成
         * 其中生成的基本类型均为包装类，例如Integer , Long , Boolean , String
         *
         * @param connection
         * @param tableName
         * @param dbType
         * @param path
         * @param isCreateFile
         * @return
         * @throws java.sql.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("@Entity\n" +
    //                "@Table(name = \""+tableName.toUpperCase()+"\", schema = DBProperties.SCHEMA_APP_PLACEHOLDER)");
    //        sb.append("public class " + tableName + " {");
            sb.append(LINE);
    //        sb.append("private static final long serialVersionUID = 1L;");
            sb.append("\r\n");
            for (int i = 1; i <= columnCount; i++) {
                sb.append(TAB);
                sb.append("@Column(name = \""+md.getColumnName(i).toUpperCase()+"\", length = "+md.getPrecision(i)+")");
                sb.append(LINE);
                sb.append(TAB);
                sb.append("private "
                        + getPojoType(md.getColumnTypeName(i)) + " "
                        + toLow0(md.getColumnName(i)) + ";");
                sb.append(LINE);
                sb.append("\r\n");
            }
    
            for (int i = 1; i <= columnCount; i++) {
                sb.append(TAB);
    
                String pojoType = getPojoType(md.getColumnTypeName(i));
                String columnName = md.getColumnName(i);
                columnName = column2JavaAttribute(columnName);
                String paramColumnName = toLow0(md.getColumnName(i));
                String getName = null;
                String setName = null;
                if (columnName.length() > 1) {
                    getName = "public " + pojoType + " get" + columnName + "() {";
                    setName = "public void set" + columnName + "("
                            + pojoType + " " + paramColumnName + ") {";
                } else {
                    getName = "public get" + columnName + "() {";
                    setName = "public set" + columnName + "(" + pojoType
                            + " " + paramColumnName + ") {";
                }
    
                sb.append(LINE).append(TAB).append(getName);
                sb.append(LINE).append(TAB).append(TAB);
                sb.append("return " + paramColumnName + ";");
                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." + paramColumnName + " = " + paramColumnName + ";");
                sb.append(LINE).append(TAB).append("}");
                sb.append(LINE);
    
            }
    //        sb.append("}");
    
            System.out.println(sb.toString());
    
            return null;
        }
    
        public static void main(String[] args) throws SQLException, ClassNotFoundException {
            Connection con = getConnection();
            table2pojo(con, "E_EXTRA_UNLADING_LUGGAGE", 3, "/BMSplus/src/main/java/com/xmcares/bmsplus/entity", true);
    
            Calendar calendar=Calendar.getInstance();
            calendar.setTime(new java.util.Date());
    //        System.out.println(calendar.get(Calendar.DAY_OF_MONTH));//今天的日期
    //        calendar.set(Calendar.DAY_OF_MONTH,calendar.get(Calendar.DAY_OF_MONTH)+1);//让日期加1
    //        System.out.println(calendar.get(Calendar.DATE));//加1之后的日期Top
            calendar.set(Calendar.DAY_OF_MONTH,calendar.get(Calendar.DAY_OF_MONTH)-1);//让日期加1
    //        System.out.println(calendar.get(Calendar.DATE));//加1之后的日期Top
    //        System.out.println(calendar.get(Calendar.MONTH)+1);//加1之后的日期Top
    //        System.out.println(calendar.get(Calendar.YEAR));//加1之后的日期Top
    //
    //        System.out.println("2016-06-30 15:23:43".substring(11,16));
    
        }
    
    
    }