通過實體類生成建表SQL語句實現方法

   
   
  
  
          
 
           
 
            
 
             
 
            
 
            
 
             
 
              import java.io.File; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              import java.io.FileOutputStream; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              import java.lang.reflect.Field; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              import java.util.ArrayList; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              import java.util.List; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              import org.hibernate.hql.internal.ast.SqlGenerator; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              import org.slf4j.Logger; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              import org.slf4j.LoggerFactory; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              public 
              class Test { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              private 
              static 
              final Logger logger = LoggerFactory.getLogger(SqlGenerator.class); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              public static void main(String[] args) { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              //實體類所在的package在磁盤上的絕對路徑 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               String packageName = 
              "d:/workspace/"; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              //生成sql的文件夾 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               String filePath = 
              "I:/create/"; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              //項目中實體類的路徑 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               String prefix = 
              "com.zh.entity."; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               String className = 
              ""; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               StringBuffer sqls = 
              new StringBuffer(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              //獲取包下的所有類名稱 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               List<String> list = getAllClasses(packageName); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              for (String str : list) { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               className = prefix + str.substring( 
              0, str.lastIndexOf( 
              ".")); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               String sql = generateSql(className, filePath); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               sqls.append(sql); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

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

               StringToSql(sqls.toString(), filePath + 
              "report.sql"); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              /** 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * 根據實體類生成建表語句 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @author 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @date 2019年1月14日 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @param className 全類名 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @param filePath 磁盤路徑 如 : d:/workspace/ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               */ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              public static String generateSql(String className,String filePath){ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              try { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               Class<?> clz = Class.forName(className); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               className = clz.getSimpleName(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               Field[] fields = clz.getDeclaredFields(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               StringBuffer column = 
              new StringBuffer(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               String varchar = 
              " varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci DEFAULT NULL,"; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              for (Field f : fields) { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               column.append( 
              " \n `"+f.getName()+ 
              "`").append(varchar); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               StringBuffer sql = 
              new StringBuffer(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               sql.append( 
              "\n DROP TABLE IF EXISTS `"+className+ 
              "`; ") 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               .append( 
              " \n CREATE TABLE `"+className+ 
              "` (") 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               .append( 
              " \n `id` int(11) NOT NULL AUTO_INCREMENT,") 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               .append( 
              " \n "+column) 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               .append( 
              " \n PRIMARY KEY (`id`) USING BTREE,") 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               .append( 
              "\n INDEX `id`(`id`) USING BTREE") 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               .append( 
              " \n ) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = utf8 COLLATE = utf8_general_ci;"); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              return sql.toString(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
              catch (ClassNotFoundException e) { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               logger.debug( 
              "該類未找到！"); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              return 
              null; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              /** 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * 獲取包下的所有類名稱,獲取的結果類似於 XXX.java 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @author 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @date 2019年1月14日 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @param packageName 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @return 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               */ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              public static List<String> getAllClasses(String packageName){ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               List<String> classList = 
              new ArrayList<String>(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               String className= 
              ""; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               File f = 
              new File(packageName); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              if(f.exists() && f.isDirectory()){ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               File[] files = f.listFiles(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              for (File file : files) { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               className = file.getName(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               classList.add(className); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              return classList; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
              else{ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               logger.debug( 
              "包路徑未找到！"); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              return 
              null; 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              /** 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * 將string 寫入sql文件 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @author 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @date 2019年1月14日 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @param str 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               * @param path 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
               */ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              public static void StringToSql(String str,String path){ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              byte[] sourceByte = str.getBytes(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              if( 
              null != sourceByte){ 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              try { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               File file = 
              new File(path); 
              //文件路徑（路徑+文件名） 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             
 
              if (!file.exists()) { 
              //文件不存在則創建文件，先創建目錄 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               File dir = 
              new File(file.getParent()); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               dir.mkdirs(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               file.createNewFile(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               FileOutputStream outStream = 
              new FileOutputStream(file); 
              //文件輸出流用於將數據寫入文件 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               outStream.write(sourceByte); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               outStream.flush(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               outStream.close(); 
              //關閉文件輸出流 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               System.out.println( 
              "生成成功"); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
              catch (Exception e) { 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               e.printStackTrace(); 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
           
 
            
 
             
 
            
 
            
 
             

               } 
             
 
            
 
          
  
  
 
 
         

生成的代碼截圖如下：

原文地址：https://blog.csdn.net/zgsdzczh/article/details/86487149