refactor: performance optimization, to avoid CPU occupancy

Author: linyimin0812

build.gradle

@@ -4,20 +4,20 @@ plugins {
 }
 
 group 'org.linyimin'
-version '1.0.2'
+version '1.0.3'
 
 repositories {
     maven { url 'https://maven.aliyun.com/repository/public/' }
     mavenCentral()
 }
 
 dependencies {
-    testImplementation 'org.junit.jupiter:junit-jupiter-api:5.8.2'
+    testImplementation 'org.junit.jupiter:junit-jupiter-api:5.9.0'
     testRuntimeOnly 'org.junit.jupiter:junit-jupiter-engine'
-    implementation 'org.mybatis:mybatis:3.3.1'
-    implementation 'com.alibaba:fastjson:1.2.79'
-    implementation 'mysql:mysql-connector-java:8.0.25'
+    implementation 'com.alibaba:fastjson:2.0.19'
+    implementation 'mysql:mysql-connector-java:8.0.30'
     implementation 'com.github.vertical-blank:sql-formatter:2.0.3'
+    implementation group: 'ognl', name: 'ognl', version: '3.0.4'
 }
 
 // See https://github.com/JetBrains/gradle-intellij-plugin/
@@ -28,6 +28,13 @@ intellij {
 }
 patchPluginXml {
     changeNotes = """
+        <h4>1.0.3</h4>
+        <ul>
+            <li>Refactoring, performance optimization, to avoid CPU occupancy</li>
+        </ul>
+        <ul>
+            <li>重构，性能优化避免CPU占满</li>
+        </ul>
         <h4>1.0.2</h4>
         <ul>
             <li>Suppot without mybatis configuration file</li>

src/main/java/io/github/linyimin/plugin/cache/MybatisXmlContentCache.java

@@ -3,15 +3,13 @@
 import com.google.common.collect.Lists;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.roots.ProjectRootManager;
-import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.PsiFileSystemItem;
 import com.intellij.psi.util.PsiUtil;
 import com.intellij.psi.xml.XmlAttribute;
 import com.intellij.psi.xml.XmlFile;
 import com.intellij.psi.xml.XmlTag;
 import io.github.linyimin.plugin.utils.MapperDomUtils;
-import org.apache.commons.lang3.StringUtils;
 
 import java.util.*;
 
@@ -23,56 +21,50 @@ public class MybatisXmlContentCache {
 
     private static final List<String> SUB_TAGS = Lists.newArrayList("insert", "update", "delete", "select");
 
-    private static final Map<Project, Map<String /* path */, String /* configuration */>> projectMybatisConfigurationMap = new HashMap<>();
-
     private static final Map<Project, Map<String /* namespace */, List<String> /* method name list */>> projectMybatisMapperMap = new HashMap<>();
 
     private static final Map<Project, Map<String /* namespace */, Set<XmlTag>>> projectMapperNamespaceMap = new HashMap<>();
 
     private static final Map<Project, Map<String /* method qualified name */, Set<XmlTag>>> projectMapperMethodMap = new HashMap<>();
 
-    private static final Map<Project, Map<String /* method qualified name */, String /* mapper xml string */>> projectMethodToMapperFilePath = new HashMap<>();
-
-
-    public static List<String> acquireConfigurations(Project project) {
-
-        addXmlCache(project);
-
-        Map<String, String> cacheMap = projectMybatisConfigurationMap.getOrDefault(project, Collections.emptyMap());
-
-        return new ArrayList<>(cacheMap.values());
-    }
 
     public static List<String> acquireByNamespace(Project project) {
 
-        addXmlCache(project);
-
         Set<String> namespaces = projectMybatisMapperMap.getOrDefault(project, new HashMap<>()).keySet();
-        return new ArrayList<>(namespaces);
-    }
 
-    public static String acquireMapperPathByMethodName(Project project, String methodName) {
+        if (!namespaces.isEmpty()) {
+            return new ArrayList<>(namespaces);
+        }
+
         addXmlCache(project);
 
-        return projectMethodToMapperFilePath.getOrDefault(project, new HashMap<>()).get(methodName);
+        return new ArrayList<>(projectMybatisMapperMap.getOrDefault(project, new HashMap<>()).keySet());
     }
 
     public static Set<XmlTag> acquireByNamespace(Project project, String namespace) {
 
-        addXmlCache(project);
-
         Map<String /* namespace */, Set<XmlTag>> cache = projectMapperNamespaceMap.getOrDefault(project, new HashMap<>());
 
-        return cache.getOrDefault(namespace, new HashSet<>());
+        if (cache.containsKey(namespace)) {
+            return cache.getOrDefault(namespace, new HashSet<>());
+        }
+
+        addXmlCache(project);
+
+       return projectMapperNamespaceMap.getOrDefault(project, new HashMap<>()).getOrDefault(namespace, new HashSet<>());
     }
 
     public static Set<XmlTag> acquireByMethodName(Project project, String methodQualifiedName) {
 
-        addXmlCache(project);
-
         Map<String /* namespace */, Set<XmlTag>> cache = projectMapperMethodMap.getOrDefault(project, new HashMap<>());
 
-        return cache.getOrDefault(methodQualifiedName, new HashSet<>());
+        if (cache.containsKey(methodQualifiedName)) {
+            return cache.get(methodQualifiedName);
+        }
+
+        addXmlCache(project);
+
+        return projectMapperMethodMap.getOrDefault(project, new HashMap<>()).getOrDefault(methodQualifiedName, new HashSet<>());
     }
 
     private static void addXmlCache(Project project) {
@@ -87,10 +79,6 @@ private static void addXmlCache(Project project) {
 
             PsiFile psiFile = item.getContainingFile();
 
-            if (MapperDomUtils.isMybatisConfigurationFile(psiFile)) {
-                addConfigurationCache(project, fileOrDir, psiFile);
-            }
-
             if (MapperDomUtils.isMybatisMapperFile(psiFile)) {
                 addMapperCache(project, psiFile);
             }
@@ -133,8 +121,6 @@ private static void addMapperCache(Project project, PsiFile psiFile) {
 
             String id = subAttribute.getValue();
 
-            addMethodToMapperCache(project, namespace, id, psiFile);
-
             addMethodXmlTagCache(project, namespace, id, subTag);
 
             addNamespaceCache(project, namespace, id);
@@ -143,25 +129,6 @@ private static void addMapperCache(Project project, PsiFile psiFile) {
 
     }
 
-    private static void addMethodToMapperCache(Project project, String namespace, String id, PsiFile psiFile) {
-        Map<String, String> methodCacheMap = projectMethodToMapperFilePath.getOrDefault(project, new HashMap<>());
-
-        String methodQualifiedName = namespace + "." + id;
-
-        String path = psiFile.getVirtualFile().getPath();
-
-        if (StringUtils.isBlank(path)) {
-            return;
-        }
-
-        path = path.substring(path.indexOf("resources/") + "resources/".length());
-
-        methodCacheMap.put(methodQualifiedName, path);
-
-        projectMethodToMapperFilePath.put(project, methodCacheMap);
-
-    }
-
     private static void addNamespaceXmlTagCache(Project project, String namespace, XmlTag xmlTag) {
 
         Map<String, Set<XmlTag>> namespaceCacheMap = projectMapperNamespaceMap.getOrDefault(project, new HashMap<>());
@@ -200,14 +167,4 @@ private static void addNamespaceCache(Project project, String namespace, String
 
         projectMybatisMapperMap.put(project, cacheMap);
     }
-
-    private static void addConfigurationCache(Project project, VirtualFile fileOrDir, PsiFile psiFile) {
-        Map<String, String> cacheMap = projectMybatisConfigurationMap.getOrDefault(project, new HashMap<>());
-
-        String path = fileOrDir.getPath();
-        String configuration = psiFile.getText();
-
-        cacheMap.put(path, configuration);
-        projectMybatisConfigurationMap.put(project, cacheMap);
-    }
 }

src/main/java/io/github/linyimin/plugin/compile/MybatisPojoCompile.java

@@ -0,0 +1,8 @@
+package io.github.linyimin.plugin.mybatis.mapping;
+
+/**
+ * @author Clinton Begin
+ */
+public enum SqlCommandType {
+    UNKNOWN, INSERT, UPDATE, DELETE, SELECT, FLUSH
+}

src/main/java/io/github/linyimin/plugin/compile/ProjectLoader.java

@@ -0,0 +1,9 @@
+package io.github.linyimin.plugin.mybatis.mapping;
+
+/**
+ * @author banzhe
+ * @date 2022/11/13 14:59
+ **/
+public interface SqlSource {
+    String getSql(Object parameterObject);
+}