feat: compile project

Author: linyimin0812

build.gradle

@@ -12,11 +12,11 @@ repositories {
 }
 
 dependencies {
-    testImplementation 'org.junit.jupiter:junit-jupiter-api:5.6.0'
+    testImplementation 'org.junit.jupiter:junit-jupiter-api:5.8.2'
     testRuntimeOnly 'org.junit.jupiter:junit-jupiter-engine'
-    implementation 'org.mybatis:mybatis:3.5.6'
-    implementation 'com.alibaba:fastjson:1.2.78'
-    implementation 'mysql:mysql-connector-java:5.1.13'
+    compileOnly 'org.mybatis:mybatis:3.5.9'
+    implementation 'com.alibaba:fastjson:1.2.79'
+    implementation 'mysql:mysql-connector-java:8.0.25'
     implementation 'com.github.vertical-blank:sql-formatter:2.0.3'
 }
 

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

@@ -1,22 +1,12 @@
 package io.github.linyimin.plugin.compile;
 
-import com.intellij.compiler.impl.ModuleCompileScope;
+import com.intellij.compiler.impl.ProjectCompileScope;
 import com.intellij.openapi.compiler.CompilerManager;
-import com.intellij.openapi.module.Module;
-import com.intellij.openapi.module.ModuleUtilCore;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.roots.OrderEnumerator;
 import com.intellij.openapi.ui.Messages;
 import com.intellij.openapi.util.io.FileUtil;
-import com.intellij.openapi.vfs.VirtualFile;
-import com.intellij.psi.PsiClass;
-import com.intellij.psi.PsiMethod;
-import com.intellij.psi.util.ClassUtil;
 import com.intellij.util.lang.UrlClassLoader;
-import io.github.linyimin.plugin.service.MybatisSqlStateComponent;
-import io.github.linyimin.plugin.service.model.MybatisSqlConfiguration;
-import io.github.linyimin.plugin.utils.JavaUtils;
-import org.apache.commons.lang3.StringUtils;
 
 import java.io.File;
 import java.net.MalformedURLException;
@@ -37,26 +27,16 @@ public class MybatisPojoCompile {
 
     public static void compile(Project project) {
 
-        MybatisSqlConfiguration sqlConfig = project.getService(MybatisSqlStateComponent.class).getState();
-        assert sqlConfig != null;
-
-        String methodName = sqlConfig.getMethod();
-        String clazzName = StringUtils.substring(methodName,0, StringUtils.lastIndexOf(methodName, "."));
-
-        PsiClass psiClass = JavaUtils.findClazz(project, clazzName);
-        final Module module = ModuleUtilCore.findModuleForPsiElement(psiClass);
-        final VirtualFile virtualFile = psiClass.getContainingFile().getVirtualFile();
-        ModuleCompileScope scope = new ModuleCompileScope(project, new Module[]{module}, false);
-//        new FileSetCompileScope(Collections.singletonList(virtualFile), new Module[]{module})
+        ProjectCompileScope scope = new ProjectCompileScope(project);
 
         CompilerManager.getInstance(project).make(scope, null);
 
-        setClassLoader(module);
+        setClassLoader(project);
     }
 
-    public static void setClassLoader(Module module) {
+    public static void setClassLoader(Project project) {
         final List<URL> urls = new ArrayList<>();
-        final List<String> list = OrderEnumerator.orderEntries(module).recursively().runtimeOnly().getPathsList().getPathList();
+        final List<String> list = OrderEnumerator.orderEntries(project).recursively().runtimeOnly().getPathsList().getPathList();
         for (String path : list) {
             try {
                 urls.add(new File(FileUtil.toSystemIndependentName(path)).toURI().toURL());

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

@@ -32,13 +32,21 @@ public static String getSql(String mybatisConfiguration, String qualifiedMethod,
 
         InputStream in = IOUtils.toInputStream(mybatisConfiguration, Charset.defaultCharset());
 
-        Resources.setDefaultClassLoader(MybatisPojoCompile.classLoader);
-        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(in);
-        Configuration configuration = sqlSessionFactory.getConfiguration();
+        ClassLoader currentClassLoader = Thread.currentThread().getContextClassLoader();
+        try {
+            Thread.currentThread().setContextClassLoader(MybatisPojoCompile.classLoader);
+            Resources.setDefaultClassLoader(MybatisPojoCompile.classLoader);
+
+
+            SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(in);
+            Configuration configuration = sqlSessionFactory.getConfiguration();
 
-        BoundSql sql = getBoundSql(configuration, qualifiedMethod, params);
+            BoundSql sql = getBoundSql(configuration, qualifiedMethod, params);
 
-        return formatSql(configuration, sql);
+            return formatSql(configuration, sql);
+        } finally {
+            Thread.currentThread().setContextClassLoader(currentClassLoader);
+        }
     }
 
     private static String formatSql(Configuration configuration, BoundSql boundSql) {