为了账号安全,请及时绑定邮箱和手机立即绑定

自己简单写的JDBCUtils,可以根据当前数据库下面的表生成java实体类。

标签:
Java

自己简单写的JDBCUtils,可以根据当前数据库下面的表生成java实体类,代码萌新,请多多包涵。

初始化配置:

复制代码

  //驱动程序名//不固定,根据驱动
    static String driver = "com.mysql.jdbc.Driver";    // URL指向要访问的数据库名******,8.0jar包新增时区。
    static String url = "jdbc:mysql://localhost/你的数据库名?serverTimezone=GMT%2B8";    // MySQL配置时的用户名
    static String user = "你的帐号";    // Java连接MySQL配置时的密码******
    static String password = "你的密码";    
    private static String[] colnames; // 列名数组

    private static String[] colTypes; // 列名类型数组
    
    private static String[] upperColnames; //驼峰命名的字段
    
    private static List<StringBuffer> listArr=new ArrayList<StringBuffer>(); //存储最终的数据
    
    static {        try {
            Class.forName(driver);
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
    }

复制代码

打开与关闭数据库连接:

复制代码

public static Connection mySQLOpen() {
        Connection con = null;        try {
            con = DriverManager.getConnection(url, user, password);
            System.out.println("succeed to connection mysql!");
        } catch (SQLException e) {
            e.printStackTrace();
        }        return con;
    }    public static void mySQLClose(ResultSet rs, Statement st, Connection con) {           try {               try {                   if (rs != null) {
                       rs.close();
                   }
               } finally {                   try {                       if (st != null) {
                           st.close();
                       }
                   } finally {                       if (con != null)
                           con.close();
                   }
               }
           } catch (SQLException e) {
               e.printStackTrace();
           }
       }

复制代码

获得当前数据库下的所有表名:

复制代码

 /**
     * 获得当前数据库生成数据库下所有的表名
     * @author one
     * 
     * */
    public static List<String> getTableFromNowConnectDB(){
        Connection conn=null;
        DatabaseMetaData dbmd = null;
        List<String> list = null;        try {
            conn=mySQLOpen();
            dbmd=(DatabaseMetaData) conn.getMetaData();            //conn.getCatalog():获得当前目录
            ResultSet rs = dbmd.getTables(conn.getCatalog(), "%", "%", new String[] { "TABLE" });            if (rs != null) {
                list = new ArrayList<String>();
            }            while(rs.next()){                //System.out.println(rs.getString("TABLE_NAME"));
                list.add(rs.getString("TABLE_NAME"));
            }
        } catch (Exception e) {
            e.printStackTrace();
        }        return list;
    }

复制代码

生成每个表名的所有字段:

复制代码

   /**
     * 生成所有的字段
     * 
     * @param sb,把stringbuffer 对象传进来继续append
     * @return 
     */
    private static void processAllField(StringBuffer sb) {        for (int i = 0; i < colnames.length; i++) {
             sb.append("\tprivate " + colTypes[i] + " " + colnames[i] + ";\r\n");
        }     
    }

复制代码

生成每个表名的set(),get()访问器:

复制代码

 /**
     * 生成所有的set(),get()
     * 
     * @param sb,把stringbuffer 对象传进来继续append     */
    private static void processAllMethod(StringBuffer sb) {        for (int i = 0; i < colnames.length; i++) {
            sb.append("\tpublic void set" + upperColnames[i] + "(" + colTypes[i] + " " + colnames[i] + "){\r\n");
            sb.append("\t\tthis." + colnames[i] + " = " + colnames[i] + ";\r\n");
            sb.append("\t}\r\n");

            sb.append("\tpublic " + colTypes[i] + " get" + upperColnames[i] + "(){\r\n");
            sb.append("\t\treturn this." + colnames[i] + ";\r\n");
            sb.append("\t}\r\n");
        }
    }

复制代码

生成每个表名的构造函数:

复制代码

  /**
     * @author one
     * 该方法用于生成构造函数
     * @param sb,把stringbuffer 对象传进来继续append
     * @see 默认生成一个无参数构造和带全部参数构造
     * 
     * public UserInfo(){}
        public UserInfo(int uid, String uname, String usex) {
        super();
        this.uid = uid;
        this.uname = uname;
        this.usex = usex;
    }
     * */
    private static void processConstructor(StringBuffer sb,String tableName) {
        sb.append("\tpublic "+tableName+"(){}\r\n");
        sb.append("\tpublic "+tableName+"(");
        String link="";        for (int i = 0; i < colnames.length; i++) {
            sb.append(link + colTypes[i] + " " + colnames[i]);
            link=",";
        }
        sb.append("){\r\n");        for (int i = 0; i < colnames.length; i++) {
            sb.append("\t\tthis."+colnames[i]+"="+colnames[i]+";\r\n");
        }
        sb.append("\t}\r\n");
    }

复制代码

数据类型转换的方法:

复制代码

 /**
     * 该方法用于类型转换
     * @param dbType:传入的数据类型
     * @author one
     * */
    private static String sqlTypeToJava(String dbType) {
        dbType = dbType.toUpperCase();        switch (dbType) {        case "VARCHAR":        case "VARCHAR2":        case "CHAR":            return "String";        case "NUMBER":        case "DECIMAL":            return "double";        case "INT":        case "SMALLINT":        case "INTEGER":            return "int";        case "BIGINT":            return "int";        case "DATETIME":        case "TIMESTAMP":        case "DATE":            return "Date";        default:            return "Object";
        }
    }

复制代码

创建文件夹,写入项目:

复制代码

 /**
     * 创建java 文件 将生成的属性 get/set 方法 保存到 文件中 markerBean
     * 
     * @author one
     * @packageName fanshe
     * @param className
     *            类名称
     * @param content
     *            类内容 包括属性 getset 方法     */
    public static void createFloder(String className, String content, String packageName) {
        String folder = System.getProperty("user.dir") + "/src/" + packageName + "/";

        File file = new File(folder);        if (!file.exists()) {
            file.mkdirs();
        }
        String fileName = folder + className + ".java";        try {
            File newdao = new File(fileName);
            FileWriter fw = new FileWriter(newdao);
            fw.write("package\t" + packageName.replace("/", ".") + ";\r\n");
            fw.write(content);
            fw.flush();
            fw.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

复制代码

生成实体类:

复制代码

 /**
     * 该方法生成Entity
     * @param tablename:表名集合
     * @param packname:要生成的路径
     * @author one
     * */
    public static List<StringBuffer> createEntity(List<String> tableNames,String packname) throws Exception{
        Connection conn=null;
        PreparedStatement pstmt=null;
        ResultSetMetaData rsmd = null;        //每个表的表名
        String eveTableName="";
        conn=mySQLOpen(); 
        //System.out.println("当前数据库下的表的总数:"+tableNames.size());
        for (int i = 0; i < tableNames.size(); i++) {            //用StringBuffer的形式先输出测试一下
            StringBuffer sb=new StringBuffer("");            //表名首字母转大写
            eveTableName=tableNames.get(i).substring(0, 1).toUpperCase()+tableNames.get(i).substring(1,tableNames.get(i).length());
            sb.append("public class "+eveTableName+" {\n\n");            //查询语句
            String sql="select * from "+eveTableName;
            
            pstmt=conn.prepareStatement(sql);            //获得表的元数据
            rsmd=pstmt.getMetaData();             // 每个表共有多少列
            int size = rsmd.getColumnCount();            //把字段放在集合里面
            colnames = new String[size];
            colTypes = new String[size];
            upperColnames = new String[size];            
            for (int j = 0; j < size; j++) {
                 String columnName = rsmd.getColumnName(j+1);//每列的字段名
                 colnames[j]=columnName;
                 String colunmType=sqlTypeToJava(rsmd.getColumnTypeName(j+1));//每列的类型
                 colTypes[j]=colunmType;                 //接下来做驼峰命名的字段
                 String upperColumnNam= rsmd.getColumnName(j+1).substring(0,1).toUpperCase()+
                         rsmd.getColumnName(j+1).substring(1,rsmd.getColumnName(j+1).length());
                 upperColnames[j]=upperColumnNam;
            }
            processAllField(sb); //生成字段
            processAllMethod(sb); //生成set,get方法
            processConstructor(sb,eveTableName); //生成构造函数 
            //添加最后一个括号
            sb.append("}");            //创建文件夹,sb.toString(),把数据扔进去            createFloder(eveTableName,sb.toString(),packname);
            listArr.add(sb);
        }        //关闭连接
        mySQLClose(null,pstmt,conn);        return listArr;
    }

复制代码

展示数据的方法:

复制代码

 /**
     * 展示
     * */   
    public static void goCreate(){        try {
            List<StringBuffer> result=createEntity(getTableFromNowConnectDB(),"vo2");            for (int i = 0; i < result.size(); i++) {
                System.out.println("============第"+(i+1)+"个表的实体类生成============");
                System.out.println(result.get(i));
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

复制代码

效果如下:

https://img1.sycdn.imooc.com//5b4f319900015a8407100497.jpg

文件夹创建效果:

https://img1.sycdn.imooc.com//5b4f31a20001f52201820067.jpg


原文


点击查看更多内容
TA 点赞

若觉得本文不错,就分享一下吧!

评论

作者其他优质文章

正在加载中
  • 推荐
  • 评论
  • 收藏
  • 共同学习,写下你的评论
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦
今天注册有机会得

100积分直接送

付费专栏免费学

大额优惠券免费领

立即参与 放弃机会
意见反馈 帮助中心 APP下载
官方微信

举报

0/150
提交
取消