feat: config current thread classloader and support fing mybatis configuaration by psimethod

Author: linyimin0812

src/main/java/io/github/linyimin/plugin/utils/MapperDomUtils.java

@@ -188,13 +188,49 @@ public static MybatisConfiguration findConfiguration(Project project, PsiMethod
 
         List<DomFileElement<MybatisConfiguration>> elements = DomService.getInstance().getFileElements(MybatisConfiguration.class, project, scope);
 
-        return elements.stream()
+        MybatisConfiguration mybatisConfiguration = elements.stream()
                 .filter(configuration -> {
                     assert module != null;
                     assert configuration.getModule() != null;
                     return StringUtils.equals(module.getName(), configuration.getModule().getName());
                 })
                 .map(DomFileElement::getRootElement)
                 .findFirst().orElse(null);
+
+        if (Objects.nonNull(mybatisConfiguration)) {
+            return mybatisConfiguration;
+        }
+
+
+        return findConfigurationByPsiMethod(project, elements, psiMethod);
+
+    }
+
+    private static MybatisConfiguration findConfigurationByPsiMethod(Project project, List<DomFileElement<MybatisConfiguration>> elements, PsiMethod psiMethod) {
+
+        if (Objects.isNull(psiMethod.getContainingClass())) {
+            return null;
+        }
+
+        List<Mapper> mappers = MapperDomUtils.findMappersByNamespace(project, psiMethod.getContainingClass().getQualifiedName());
+
+        if (CollectionUtils.isEmpty(mappers)) {
+            return null;
+        }
+
+        XmlTag xmlTag = mappers.get(0).getXmlTag();
+        if (Objects.isNull(xmlTag) || Objects.isNull(xmlTag.getContainingFile())) {
+            return null;
+        }
+
+        String mapperFileName = xmlTag.getContainingFile().getName();
+
+
+        return elements.stream()
+                .map(DomFileElement::getRootElement)
+                .filter(configuration -> {
+                    XmlTag tag = configuration.getXmlTag();
+                    return Objects.nonNull(tag) && StringUtils.contains(tag.getText(), mapperFileName);
+                }).findFirst().orElse(null);
     }
 }

src/main/java/io/github/linyimin/plugin/utils/MybatisSqlUtils.java

@@ -32,7 +32,12 @@ public class MybatisSqlUtils {
 
     public static String getSql(String mybatisConfiguration, String qualifiedMethod, String params, boolean isCheck) {
 
+        ClassLoader currentClassLoader = Thread.currentThread().getContextClassLoader();
+
         try {
+
+            Thread.currentThread().setContextClassLoader(MybatisPojoCompile.classLoader);
+
             InputStream in = IOUtils.toInputStream(mybatisConfiguration, Charset.defaultCharset());
             Resources.setDefaultClassLoader(MybatisPojoCompile.classLoader);
 
@@ -47,6 +52,8 @@ public static String getSql(String mybatisConfiguration, String qualifiedMethod,
                 Messages.showInfoMessage(e.getMessage(), Constant.APPLICATION_NAME);
             }
             throw e;
+        } finally {
+            Thread.currentThread().setContextClassLoader(currentClassLoader);
         }
     }