减少循环次数

Author: linpeilie

mapstruct-plus-processor/src/main/java/io/github/linpeilie/processor/AutoMapperProcessor.java

@@ -543,54 +543,23 @@ private void generateMapper() {
 
         mapperList.removeIf(metadata -> !metadata.isConvertGenerate());
 
-        // 兼容同模块下，同名不同包的场景
         mapperList.forEach(metadata -> {
-            String mapperName = metadata.mapperName();
-            // 同名类时，增加后缀
-            Integer index = AUTO_MAPPER_INDEX.getOrDefault(mapperName, 0);
-            if (index > 0) {
-                mapperName = mapperName + "__" + index;
-            }
-            AUTO_MAPPER_INDEX.put(metadata.mapperName(), ++index);
-            metadata.setMapperName(mapperName);
-        });
+            // 兼容同模块下，同名不同包的场景
+            this.mapperNameAddSuffix(metadata);
 
-        mapperList.forEach(metadata -> {
             if (metadata.isCycleAvoiding()) {
                 addAdapterMethod(metadata);
             } else {
                 typeRelationMapper(metadata);
             }
         });
 
-        // import dependency
         mapperList.forEach(metadata -> {
-            Set<TypeName> dependencies = metadata.getDependencies();
-            if (CollectionUtils.isNotEmpty(dependencies)) {
-                List<ClassName> dependencyMappers = dependencies.stream().map(dependency ->
-                    typeRelationMappers.get(dependency.toString())
-                ).filter(Objects::nonNull).flatMap(Collection::stream).collect(Collectors.toList());
-
-                if (CollectionUtils.isNotEmpty(dependencyMappers)) {
-                    metadata.addUseList(dependencyMappers);
-                }
-            }
-            // source
-            List<ClassName> sourceDependencies =
-                typeRelationMappers.get(metadata.getSourceClassName().reflectionName());
-
-            if (CollectionUtils.isNotEmpty(sourceDependencies)) {
-                metadata.addUseList(sourceDependencies);
-            }
-            // remove itself
-            if (CollectionUtils.isNotEmpty(metadata.getUsesClassNameList())) {
-                metadata.getUsesClassNameList()
-                    .removeIf(use -> use.reflectionName().equals(metadata.mapperClass().reflectionName()));
-            }
+            this.relationDependencies(metadata);
+            this.usesRemoveItself(metadata);
+            this.writeAutoMapperClassFile(metadata);
         });
 
-        mapperList.forEach(this::writeAutoMapperClassFile);
-
         adapterMapperGenerator.write(processingEnv,
             methodMap.values(),
             AutoMapperProperties.getAdapterClassName(),
@@ -616,6 +585,45 @@ private void generateMapper() {
         }
     }
 
+    private void mapperNameAddSuffix(AutoMapperMetadata metadata) {
+        String mapperName = metadata.mapperName();
+        // 同名类时，增加后缀
+        Integer index = AUTO_MAPPER_INDEX.getOrDefault(mapperName, 0);
+        if (index > 0) {
+            mapperName = mapperName + "__" + index;
+        }
+        AUTO_MAPPER_INDEX.put(metadata.mapperName(), ++index);
+        metadata.setMapperName(mapperName);
+    }
+
+    private void relationDependencies(AutoMapperMetadata metadata) {
+        Set<TypeName> dependencies = metadata.getDependencies();
+        if (CollectionUtils.isNotEmpty(dependencies)) {
+            List<ClassName> dependencyMappers = dependencies.stream().map(dependency ->
+                typeRelationMappers.get(dependency.toString())
+            ).filter(Objects::nonNull).flatMap(Collection::stream).collect(Collectors.toList());
+
+            if (CollectionUtils.isNotEmpty(dependencyMappers)) {
+                metadata.addUseList(dependencyMappers);
+            }
+        }
+        // source
+        List<ClassName> sourceDependencies =
+            typeRelationMappers.get(metadata.getSourceClassName().reflectionName());
+
+        if (CollectionUtils.isNotEmpty(sourceDependencies)) {
+            metadata.addUseList(sourceDependencies);
+        }
+    }
+
+    private void usesRemoveItself(AutoMapperMetadata metadata) {
+        // remove itself
+        if (CollectionUtils.isNotEmpty(metadata.getUsesClassNameList())) {
+            metadata.getUsesClassNameList()
+                .removeIf(use -> use.reflectionName().equals(metadata.mapperClass().reflectionName()));
+        }
+    }
+
     private Set<TypeName> listDependencies(TypeElement autoMapperEle) {
         Set<TypeName> set = new HashSet<>();