自动生成Spring Boot + lombok + MySql结构的数据库映射entity

生成代码路径、基本信息及数据库配置

    private String packageOutPath = "main.java.com.demo.code.entity";//指定实体生成所在包的路径
    private String packagePath = "com.demo.code.entity";//实体类包头
    private String authorName = "Zhang Wei";//作者名字
    private String tablename = "";//表名  
    private String[] colnames; // 列名数组  
    private String[] colTypes; //列名类型数组  
    private int[] colSizes; //列名大小数组  
    private boolean f_util = false; // 是否需要导入包java.util.*  
    private boolean f_sql = false; // 是否需要导入包java.sql.*  

    //数据库连接  
    private static final String URL ="jdbc:mysql://localhost:3306/demo?useUnicode=true&characterEncoding=utf-8&useSSL=false";
    private static final String NAME = "root";  
    private static final String PASS = "password";
    private static final String DRIVER ="com.mysql.jdbc.Driver";  

自动生成实体类完整代码

package com.demo.code;

import java.io.File;  
import java.io.FileWriter;  
import java.io.IOException;  
import java.io.PrintWriter;  
import java.sql.Connection;  
import java.sql.DatabaseMetaData;  
import java.sql.DriverManager;  
import java.sql.PreparedStatement;  
import java.sql.ResultSet;  
import java.sql.ResultSetMetaData;  
import java.sql.SQLException;  
import java.util.ArrayList;
import java.util.List;

/** 
 * 从数据库表反射出实体类,自动生成实体类 
 * @author Zhang Wei
 * 
 */
public class GenEntityMysql {
      
    private String packageOutPath = "main.java.com.demo.code.entity";//指定实体生成所在包的路径
    private String packagePath = "com.demo.code.entity";//实体类包头
    private String authorName = "Zhang Wei";//作者名字
    private String tablename = "";//表名  
    private String[] colnames; // 列名数组  
    private String[] colTypes; //列名类型数组  
    private int[] colSizes; //列名大小数组  
    private boolean f_util = false; // 是否需要导入包java.util.*  
    private boolean f_sql = false; // 是否需要导入包java.sql.*  
      
    //数据库连接  
    private static final String URL ="jdbc:mysql://localhost:3306/demo?useUnicode=true&characterEncoding=utf-8&useSSL=false";
    private static final String NAME = "root";  
    private static final String PASS = "password";
    private static final String DRIVER ="com.mysql.jdbc.Driver";  
  
    /* 
     * 构造函数 
     */  
    public GenEntityMysql(){  
        //获取所有的表名
        List<String> list=getTableName();  
        
        for(int p=0;p<list.size();p++){  
            tablename = list.get(p);  
            //创建连接  
            Connection con;  
            //查要生成实体类的表  
            String sql = "select * from " + tablename;  
            PreparedStatement pStemt = null;  
            try {  
                try {  
                    Class.forName(DRIVER);  
                } catch (ClassNotFoundException e1) {  
                    e1.printStackTrace();  
                }  
                con = DriverManager.getConnection(URL,NAME,PASS);  
                pStemt = con.prepareStatement(sql);  
                ResultSetMetaData rsmd = pStemt.getMetaData();  
                int 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("datetime")){  
                        f_util = true;  
                    }  
                    if(colTypes[i].equalsIgnoreCase("image") || colTypes[i].equalsIgnoreCase("text")){  
                        f_sql = true;  
                    }  
                    colSizes[i] = rsmd.getColumnDisplaySize(i + 1);  
                }  
                  
                String content = parse(colnames,colTypes,colSizes);  
                  
                try {  
                    File directory = new File("");  
                    String outputPath = directory.getAbsolutePath()+ "/src/"+this.packageOutPath.replace(".", "/")+"/"+initcap(tablename) + ".java";  
                    FileWriter fw = new FileWriter(outputPath);  
                    PrintWriter pw = new PrintWriter(fw);  
                    pw.println(content);  
                    pw.flush();  
                    pw.close();  
                } catch (IOException e) {  
                    e.printStackTrace();  
                }  
                  
            } catch (SQLException e) {  
                e.printStackTrace();  
            } finally{   
            }  
        }  
        System.out.println("生成完毕!");  
    }  
    /** 
      * Java方法 得到当前数据库下所有的表名
      */  
    private List<String> getTableName() {  
         List<String> list=new ArrayList<String>();  
         try {    
             DatabaseMetaData meta = DriverManager.getConnection(URL,NAME,PASS).getMetaData();    
             ResultSet rs = meta.getTables(null, null, null,new String[]{"TABLE"});  
             while (rs.next()) {  
                 list.add(rs.getString(3));    
             }    
           }catch(Exception e){    
               e.printStackTrace();    
           }  
        return list;  
     }  
    /** 
     * 功能:生成实体类主体代码 
     * @param colnames 
     * @param colTypes 
     * @param colSizes 
     * @return 
     */  
    private String parse(String[] colnames, String[] colTypes, int[] colSizes) {  
        StringBuffer sb = new StringBuffer();  
        //包路径
        sb.append("package " + this.packagePath + ";\r\n");  
        sb.append("\r\n");  


        sb.append("import java.io.Serializable;\r\n");
        //如果存在时间类型的字段,加入Date引用
        if(f_util){
            sb.append("import java.util.Date;\r\n\r\n");
        }
        
        //引用
        sb.append("import javax.persistence.*;\r\n");
        sb.append("import lombok.AllArgsConstructor;\r\n");
        sb.append("import lombok.Builder;\r\n");
        sb.append("import lombok.Data;\r\n");
        sb.append("import lombok.NoArgsConstructor;\r\n");
        sb.append("import lombok.ToString;\r\n");
        
        //注解
        sb.append("@Data\r\n");  
        sb.append("@Builder\r\n");  
        sb.append("@NoArgsConstructor\r\n");  
        sb.append("@AllArgsConstructor\r\n");  
        sb.append("@ToString\r\n");  
        sb.append("@Entity\r\n");  
        sb.append("@Table(name = \"" + tablename + "\" )\r\n");
        
        //实体部分  
        sb.append("public class " + initcap(tablename) + " implements Serializable {\r\n\r\n");
        //序列化
        sb.append("\tprivate static final long serialVersionUID = 1L;\r\n\r\n");
        processAllAttrs(sb);//属性  
        sb.append("}\r\n");  
//        System.out.println(sb.toString());
        return sb.toString();  
    }  
      
    /** 
     * 功能:生成所有属性 
     * @param sb 
     */  
    private void processAllAttrs(StringBuffer sb) {  
        for (int i = 0; i < colnames.length; i++) {  
            //字段下划线分割处理
            String rStr = "";
            String[] strs = colnames[i].split("_");
            if(strs.length > 1){
                for (int j = 0; j < strs.length; j++) {
                    
                    char[] chc = strs[j].toCharArray();  
                    if(j > 0){
                        if(chc[0] >= 'a' && chc[0] <= 'z'){  
                            chc[0] = (char)(chc[0] - 32);  
                        }
                    }
                    rStr += new String(chc);
                }
            }else{
                rStr = colnames[i];
            }
            //id列单独增加注解标识
            if(rStr.equals("id")){
                sb.append("\t@Id\r\n\t@GeneratedValue(strategy = GenerationType.IDENTITY)\r\n");
            }
            //组装属性
            sb.append("\tprivate " + sqlType2JavaType(colTypes[i]) + " " + rStr + ";\r\n\r\n");  
        }  
        sb.append("\r\n");  
    }
      
    /** 
     * 功能:将输入字符串的首字母改成大写 
     * @param str 
     * @return 
     */  
    private String initcap(String str) {  
          
        char[] ch = str.toCharArray();  
        if(ch[0] >= 'a' && ch[0] <= 'z'){  
            ch[0] = (char)(ch[0] - 32);  
        }  
        //特殊需要对_进行分割组装
        String rStr = "";
        String[] strs = str.split("_");
        for (int i = 0; i < strs.length; i++) {
            char[] chc = strs[i].toCharArray();  
            if(chc[0] >= 'a' && chc[0] <= 'z'){  
                chc[0] = (char)(chc[0] - 32);  
            }
            strs[i] = new String(chc);
            rStr += strs[i];
        }
        
        return new String(rStr);  
    }  
  
    /** 
     * 功能:获得列的数据类型 
     * @param sqlType 
     * @return 
     */  
    private String sqlType2JavaType(String sqlType) {  
          
        if(sqlType.equalsIgnoreCase("bit")){  
            return "boolean";  
        }else if(sqlType.equalsIgnoreCase("tinyint")){  
            return "byte";  
        }else if(sqlType.equalsIgnoreCase("smallint")){  
            return "short";  
        }else if(sqlType.equalsIgnoreCase("int")||sqlType.equalsIgnoreCase("INT UNSIGNED")){  
            //INT UNSIGNED无符号整形  
            return "Integer";  
        }else if(sqlType.equalsIgnoreCase("bigint")){  
            return "long";  
        }else if(sqlType.equalsIgnoreCase("float")){  
            return "float";  
        }else if(sqlType.equalsIgnoreCase("decimal") || sqlType.equalsIgnoreCase("numeric")   
                || sqlType.equalsIgnoreCase("real") || sqlType.equalsIgnoreCase("money")   
                || sqlType.equalsIgnoreCase("smallmoney")){  
            return "double";  
        }else if(sqlType.equalsIgnoreCase("varchar") || sqlType.equalsIgnoreCase("char")   
                || sqlType.equalsIgnoreCase("nvarchar") || sqlType.equalsIgnoreCase("nchar")   
                || sqlType.equalsIgnoreCase("text")){  
            return "String";  
        }else if(sqlType.equalsIgnoreCase("datetime")){  
            return "Date";  
        }else if(sqlType.equalsIgnoreCase("image")){  
            return "Blod";  
        }         
        return null;  
    }  
    
    public static void main(String[] args) {
        new GenEntityMysql();
    }
    
}

pom.xml依赖包

<!-- MySql -->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
        <!-- lombok -->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.16.18</version>
        </dependency>
        <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-lang3</artifactId>
            <version>3.0</version>
        </dependency>
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 230,825评论 6 546
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 99,814评论 3 429
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 178,980评论 0 384
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 64,064评论 1 319
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 72,779评论 6 414
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 56,109评论 1 330
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 44,099评论 3 450
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 43,287评论 0 291
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 49,799评论 1 338
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 41,515评论 3 361
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 43,750评论 1 375
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 39,221评论 5 365
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 44,933评论 3 351
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 35,327评论 0 28
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 36,667评论 1 296
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 52,492评论 3 400
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 48,703评论 2 380

推荐阅读更多精彩内容