mysql数据库字段生成

最新推荐文章于 2023-05-11 10:47:40 发布
lljliulljn
最新推荐文章于 2023-05-11 10:47:40 发布
阅读量486
点赞数
 
 import 
  static 
  java.lang.Character.toUpperCase; 
 
 import 
  java.sql.*; 
 
 import 
  java.util.*; 

    
 
 /** 
 
   
 * 
 
   
 * @author zhangjx 
 
   
 */ 
 
 public 
  class 
  ClassCreator { 

    
 
      
 public 
  static 
  void 
  main(String[] args)  
 throws 
  Exception { 
 
          
 create( 
 "org.om.model" 
 ,  
 "users" 
 ); 
 
      
 } 

    
 
      
 private 
  static 
  void 
  create(String pkg, String tablename)  
 throws 
  Exception { 
 
          
 com.mysql.jdbc.jdbc2.optional.MysqlConnectionPoolDataSource source =  
 new 
  com.mysql.jdbc.jdbc2.optional.MysqlConnectionPoolDataSource(); 
 
          
 source.setUrl( 
 "jdbc:mysql://localhost:3306/om?autoReconnect=true&amp;characterEncoding=utf8" 
 );  //数据库url 
 
          
 source.setUser( 
 "root" 
 );  
 //数据库账号 
 
          
 source.setPassword( 
 "admin" 
 );  
 //数据库密码 
 
          
 Connection conn = source.getConnection(); 
 
          
 DatabaseMetaData meta = conn.getMetaData(); 
 
          
 ResultSet rs = meta.getColumns( 
 null 
 ,  
 "%" 
 , tablename,  
 null 
 ); 
 
 //       ResultSetMetaData rsd = rs.getMetaData(); 
 
 //       for(int i =1 ; i<=rsd.getColumnCount();i++) { 
 
 //           System.out.println(rsd.getColumnName(i)); 
 
 //       }      
 
          
 StringBuilder sb =  
 new 
  StringBuilder(); 

    
 
          
 sb.append( 
 "package " 
  + pkg +  
 ";" 
  +  
 "\r\n\r\n" 
 ); 

    
 
          
 sb.append( 
 "import org.redkale.util.*;\r\n" 
 ); 
 
          
 sb.append( 
 "import javax.persistence.*;\r\n" 
 ); 
 
          
 sb.append( 
 "import org.redkale.convert.*;\r\n" 
 ); 
 
          
 sb.append( 
 "import " 
  + pkg.substring( 
 0 
 , pkg.lastIndexOf( 
 '.' 
 )) +  
 ".base.BaseEntity;\r\n" 
 ); 
 
          
 char 
 [] chs = tablename.toCharArray(); 
 
          
 chs[ 
 0 
 ] = toUpperCase(chs[ 
 0 
 ]); 

    
 
          
 String classname =  
 new 
  String(chs).replace( 
 "info" 
 ,  
 "Info" 
 ).replace( 
 "record" 
 ,  
 "Record" 
 ).replaceAll( 
 "his$" 
 ,  
 "His" 
 ); 
 
          
 sb.append( 
 "\r\n/**\r\n" 
 
              
 +  
 " *\r\n" 
 
              
 +  
 " * @author " 
  + System.getProperty( 
 "user.name" 
 ) +  
 "\r\n" 
 
              
 +  
 " */\r\n" 
 ); 
 
          
 if 
  (classname.contains( 
 "Info" 
 )) sb.append( 
 "@Cacheable\r\n" 
 ); 
 
          
 sb.append( 
 "public class " 
  + classname +  
 " extends BaseEntity {\r\n" 
 ); 
 
          
 boolean 
  idable =  
 false 
 ; 
 
          
 List<StringBuilder> list =  
 new 
  ArrayList<>(); 
 
          
 while 
  (rs.next()) { 
 
              
 boolean 
  incre = rs.getBoolean( 
 "IS_AUTOINCREMENT" 
 ); 
 
              
 String column = rs.getString( 
 "COLUMN_NAME" 
 ); 
 
              
 String type = rs.getString( 
 "TYPE_NAME" 
 ); 
 
              
 String remark = rs.getString( 
 "REMARKS" 
 ); 
 
              
 sb.append( 
 "\r\n" 
 ); 
 
              
 if 
  (!idable) { 
 
                  
 idable =  
 true 
 ; 
 
                  
 sb.append( 
 "    @Id\r\n" 
 ); 
 
                  
 if 
  (incre) sb.append( 
 "    @GeneratedValue\r\n" 
 ); 
 
              
 } 
 
              
 if 
  ( 
 "createtime" 
 .equals(column)) sb.append( 
 "    @Column(updatable = false)\r\n" 
 ); 
 
              
 String ctype =  
 "NULL" 
 ; 
 
              
 if 
  ( 
 "INT" 
 .equalsIgnoreCase(type)) { 
 
                  
 ctype =  
 "int" 
 ; 
 
              
 } 
 else 
  if 
 ( 
 "INT UNSIGNED" 
 .equalsIgnoreCase(type)){ 
 
                  
 ctype =  
 "int" 
 ; 
 
              
 }  
 else 
  if 
  ( 
 "BIGINT" 
 .equalsIgnoreCase(type)) { 
 
                  
 ctype =  
 "long" 
 ; 
 
              
 }  
 else 
  if 
  ( 
 "SMALLINT" 
 .equalsIgnoreCase(type)) { 
 
                  
 ctype =  
 "short" 
 ; 
 
              
 }  
 else 
  if 
  ( 
 "FLOAT" 
 .equalsIgnoreCase(type)) { 
 
                  
 ctype =  
 "float" 
 ; 
 
              
 }  
 else 
  if 
  ( 
 "DECIMAL" 
 .equalsIgnoreCase(type)) { 
 
                  
 ctype =  
 "float" 
 ; 
 
              
 }  
 else 
  if 
  ( 
 "VARCHAR" 
 .equalsIgnoreCase(type)) { 
 
                  
 ctype =  
 "String" 
 ; 
 
              
 }  
 else 
  if 
  (type.contains( 
 "TEXT" 
 )) { 
 
                  
 ctype =  
 "String" 
 ; 
 
              
 }  
 else 
  if 
  (type.contains( 
 "BLOB" 
 )) { 
 
                  
 ctype =  
 "byte[]" 
 ; 
 
              
 } 
 
              
 sb.append( 
 "    private " 
  + ctype +  
 " " 
  + column); 
 
              
 if 
  ( 
 "String" 
 .equals(ctype)) sb.append( 
 " = \"\"" 
 ); 
 
              
 sb.append( 
 "; //" 
  + remark +  
 "\r\n" 
 ); 

    
 
              
 char 
 [] chs2 = column.toCharArray(); 
 
              
 chs2[ 
 0 
 ] = toUpperCase(chs2[ 
 0 
 ]); 
 
              
 String sgname =  
 new 
  String(chs2); 

    
 
              
 StringBuilder setter =  
 new 
  StringBuilder(); 
 
              
 setter.append( 
 "\r\n    public void set" 
  + sgname +  
 "(" 
  + ctype +  
 " " 
  + column +  
 ") {\r\n" 
 ); 
 
              
 setter.append( 
 "        this." 
  + column +  
 " = " 
  + column +  
 ";\r\n" 
 ); 
 
              
 setter.append( 
 "    }\r\n" 
 ); 
 
              
 list.add(setter); 

    
 
              
 StringBuilder getter =  
 new 
  StringBuilder(); 
 
              
 getter.append( 
 "\r\n    public " 
  + ctype +  
 " get" 
  + sgname +  
 "() {\r\n" 
 ); 
 
              
 getter.append( 
 "        return this." 
  + column +  
 ";\r\n" 
 ); 
 
              
 getter.append( 
 "    }\r\n" 
 ); 
 
              
 list.add(getter); 
 
          
 } 
 
          
 for 
  (StringBuilder item : list) { 
 
              
 sb.append(item); 
 
          
 } 
 
          
 sb.append( 
 "}\r\n" 
 ); 
 
          
 System.out.println(sb); 
 
      
 } 
 
 } 