优化生成代码执行性能 github issues/219

mybatis-plus-generate/src/main/java/com/baomidou/mybatisplus/generator/config/builder/ConfigBuilder.java

@@ -362,15 +362,18 @@ public class ConfigBuilder {
     }
 
     /**
+     * <p>
      * 检查是否有
-     *     {@link com.baomidou.mybatisplus.annotations.TableId}
-     *  {@link com.baomidou.mybatisplus.annotations.TableField}
-     *  注解
+     * {@link com.baomidou.mybatisplus.annotations.TableId}
+     * {@link com.baomidou.mybatisplus.annotations.TableField}
+     * 注解
+     * </p>
+     *
      * @param config
      * @param tableInfo
      * @param fieldPrefix
      */
-    private void checkTableIdTableFieldAnnotation(StrategyConfig config, TableInfo tableInfo, String[] fieldPrefix){
+    private void checkTableIdTableFieldAnnotation(StrategyConfig config, TableInfo tableInfo, String[] fieldPrefix) {
         boolean importTableFieldAnnotaion = false;
         boolean importTableIdAnnotaion = false;
         if (config.isEntityTableFieldAnnotationEnable()) {
@@ -396,8 +399,8 @@ public class ConfigBuilder {
         if (importTableIdAnnotaion) {
             tableInfo.getImportPackages().add(com.baomidou.mybatisplus.annotations.TableId.class.getCanonicalName());
         }
-        if(globalConfig.getIdType()!=null){
-            if(!importTableIdAnnotaion){
+        if (globalConfig.getIdType() != null) {
+            if (!importTableIdAnnotaion) {
                 tableInfo.getImportPackages().add(com.baomidou.mybatisplus.annotations.TableId.class.getCanonicalName());
             }
             tableInfo.getImportPackages().add(com.baomidou.mybatisplus.enums.IdType.class.getCanonicalName());
@@ -465,7 +468,7 @@ public class ConfigBuilder {
                     if (isInclude) {
                         for (String includeTab : config.getInclude()) {
                             if (includeTab.equalsIgnoreCase(tableName)) {
-                                includeTableList.add(tableInfo);
+                                includeTableList.add(this.convertTableFields(tableInfo, strategy));
                             } else {
                                 notExistTables.add(includeTab);
                             }
@@ -479,7 +482,7 @@ public class ConfigBuilder {
                             }
                         }
                     }
-                    tableList.add(this.convertTableFields(tableInfo, strategy));
+                    tableList.add(tableInfo);
                 } else {
                     System.err.println("当前数据库为空！！！");
                 }
@@ -501,6 +504,12 @@ public class ConfigBuilder {
             if (!isInclude && !isExclude) {
                 includeTableList = tableList;
             }
+            /**
+             * 性能优化，只处理需执行表字段 github issues/219
+             */
+            for (TableInfo ti : includeTableList) {
+                this.convertTableFields(ti, strategy);
+            }
         } catch (SQLException e) {
             e.printStackTrace();
         } finally {
@@ -692,7 +701,9 @@ public class ConfigBuilder {
     }
 
     /**
+     * <p>
      * 获取当前的SQL类型
+     * </p>
      *
      * @return DB类型
      */

mybatis-plus-support/src/main/java/com/baomidou/mybatisplus/plugins/SqlParserHandler.java

@@ -48,7 +48,8 @@ public abstract class SqlParserHandler {
             }
             // SQL 解析
             if (CollectionUtils.isNotEmpty(this.sqlParserList)) {
-                int flag = 0;// 标记是否修改过 SQL
+                // 标记是否修改过 SQL
+                int flag = 0;
                 String originalSql = (String) metaObject.getValue(PluginUtils.DELEGATE_BOUNDSQL_SQL);
                 for (ISqlParser sqlParser : this.sqlParserList) {
                     SqlInfo sqlInfo = sqlParser.optimizeSql(metaObject, originalSql);