feat: support generate sql from xml and execute the sql

Author: linyimin0812

build.gradle

@@ -15,6 +15,9 @@ dependencies {
     testImplementation 'org.junit.jupiter:junit-jupiter-api:5.6.0'
     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'
+    implementation 'com.github.vertical-blank:sql-formatter:2.0.3'
 }
 
 // See https://github.com/JetBrains/gradle-intellij-plugin/

gradlew

@@ -80,6 +80,7 @@ case "`uname`" in
     ;;
 esac
 
+
 CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
 
 

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

@@ -0,0 +1,70 @@
+package io.github.linyimin.plugin.compile;
+
+import com.intellij.compiler.impl.ModuleCompileScope;
+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;
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.List;
+
+
+/**
+ * @author yiminlin
+ * @date 2022/02/07 1:49 上午
+ **/
+public class MybatisPojoCompile {
+
+    // TODO: 需要优化成只编译Mybatis需要的类
+
+    public static ClassLoader classLoader;
+
+    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})
+
+        CompilerManager.getInstance(project).make(scope, null);
+
+        setClassLoader(module);
+    }
+
+    public static void setClassLoader(Module module) {
+        final List<URL> urls = new ArrayList<>();
+        final List<String> list = OrderEnumerator.orderEntries(module).recursively().runtimeOnly().getPathsList().getPathList();
+        for (String path : list) {
+            try {
+                urls.add(new File(FileUtil.toSystemIndependentName(path)).toURI().toURL());
+            }
+            catch (MalformedURLException e) {
+                Messages.showErrorDialog(e.getMessage(), "MalformedURLException");
+            }
+        }
+        classLoader = UrlClassLoader.build().urls(urls).parent(ClassLoader.getSystemClassLoader()).get();
+    }
+}

src/main/java/io/github/linyimin/plugin/description/MapperConfigurationDescription.java

@@ -0,0 +1,25 @@
+package io.github.linyimin.plugin.description;
+
+import com.intellij.openapi.module.Module;
+import com.intellij.psi.xml.XmlFile;
+import com.intellij.util.xml.DomFileDescription;
+import io.github.linyimin.plugin.dom.model.MybatisConfiguration;
+import io.github.linyimin.plugin.utils.MapperDomUtils;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+/**
+ * @author yiminlin
+ * @date 2022/02/05 2:45 上午
+ **/
+public class MapperConfigurationDescription extends DomFileDescription<MybatisConfiguration> {
+
+    public MapperConfigurationDescription() {
+        super(MybatisConfiguration.class, "configuration");
+    }
+
+    @Override
+    public boolean isMyFile(@NotNull XmlFile file, @Nullable Module module) {
+        return MapperDomUtils.isMybatisConfigurationFile(file);
+    }
+}

src/main/java/io/github/linyimin/plugin/dom/Constant.java

@@ -1,8 +1,16 @@
 package io.github.linyimin.plugin.dom;
 
+import com.alibaba.fastjson.JSON;
 import com.google.common.collect.Lists;
+import com.intellij.util.containers.ContainerUtil;
 
+import java.math.BigDecimal;
+import java.text.DateFormat;
+import java.text.SimpleDateFormat;
+import java.time.*;
+import java.util.Date;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author yiminlin
@@ -14,4 +22,41 @@ public class Constant {
     public static final List<String> MYBATIS_OPS = Lists.newArrayList(
             "insert", "update", "delete", "select"
     );
+
+    public static final String PARAM_ANNOTATION = "org.apache.ibatis.annotations.Param";
+
+    public static final String APPLICATION_NAME = "mybatis-sql-viewer";
+
+    public static final String DATABASE_URL_TEMPLATE = "jdbc:mysql://%s:%s/%s";
+
+    private static final String PATTERN = "yyyy-MM-dd HH:mm:ss";
+    private static final DateFormat DF = new SimpleDateFormat(PATTERN);
+    public static Map<String, Object> normalTypes = new ContainerUtil.ImmutableMapBuilder<String, Object>()
+            .put("byte", 0)
+            .put("java.lang.Byte", 0)
+            .put("short", 0)
+            .put("java.lang.Short", 0)
+            .put("int", 0)
+            .put("java.lang.Integer", 0)
+            .put("long", 0)
+            .put("java.lang.Long", 0)
+            .put("float", 0.0)
+            .put("java.lang.Float", 0.0)
+            .put("double", 0.0)
+            .put("java.lang.Double", 0.0)
+            .put("boolean", true)
+            .put("java.lang.Boolean", Boolean.TRUE)
+            .put("java.math.BigDecimal", new BigDecimal(0))
+            .put("char", "a")
+            .put("java.lang.Character", "a")
+            .put("java.lang.String", "")
+            .put("java.util.Date", DF.format(new Date()))
+            .put("java.sql.Date", DF.format(new Date()))
+            .put("java.sql.Timestamp", System.currentTimeMillis())
+            .put("java.time.LocalDate", LocalDate.now(ZoneId.of(JSON.defaultTimeZone.getID())).toString())
+            .put("java.time.LocalTime", LocalTime.now(ZoneId.of(JSON.defaultTimeZone.getID())).toString())
+            .put("java.time.LocalDateTime", LocalDateTime.now(ZoneId.of(JSON.defaultTimeZone.getID())).toString())
+            .put("java.time.YearMonth", YearMonth.now(ZoneId.of(JSON.defaultTimeZone.getID())).toString())
+            .build();
+
 }

src/main/java/io/github/linyimin/plugin/dom/model/MybatisConfiguration.java

@@ -0,0 +1,14 @@
+package io.github.linyimin.plugin.dom.model;
+
+import com.intellij.util.xml.DomElement;
+import com.intellij.util.xml.SubTagsList;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.List;
+
+/**
+ * @author yiminlin
+ * @date 2022/02/05 2:44 上午
+ **/
+public interface MybatisConfiguration extends DomElement {
+}

src/main/java/io/github/linyimin/plugin/message/ConfigChangeNotifier.java

@@ -0,0 +1,20 @@
+package io.github.linyimin.plugin.message;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.util.messages.Topic;
+
+/**
+ * @author yiminlin
+ * @date 2022/02/02 2:47 上午
+ **/
+public interface ConfigChangeNotifier {
+
+    Topic<ConfigChangeNotifier> PARAM_CHANGE_TOPIC = Topic.create("param change", ConfigChangeNotifier.class);
+
+    /**
+     * @param project {@link Project}
+     */
+    default void configChanged(Project project) {
+
+    }
+}

src/main/java/io/github/linyimin/plugin/provider/MapperXmlProcessor.java

@@ -16,7 +16,6 @@
 import java.util.Collections;
 import java.util.List;
 import java.util.Objects;
-import java.util.stream.Collectors;
 
 /**
  * @author yiminlin

src/main/java/io/github/linyimin/plugin/provider/generate/FunctionTooltip.java

@@ -1,6 +1,5 @@
 package io.github.linyimin.plugin.provider.generate;
 
-import com.intellij.openapi.wm.ToolWindowManager;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiIdentifier;
 import com.intellij.psi.PsiMethod;

src/main/java/io/github/linyimin/plugin/provider/generate/MapperXmlGenerateSqlLineMakerProvider.java

@@ -5,6 +5,8 @@
 import com.intellij.openapi.editor.markup.GutterIconRenderer;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.xml.XmlTag;
+import com.intellij.psi.xml.XmlToken;
+import com.intellij.psi.xml.XmlTokenType;
 import io.github.linyimin.plugin.dom.Constant;
 import io.github.linyimin.plugin.utils.IconUtils;
 import io.github.linyimin.plugin.utils.MapperDomUtils;
@@ -18,22 +20,29 @@ public class MapperXmlGenerateSqlLineMakerProvider implements LineMarkerProvider
     @Override
     public LineMarkerInfo<?> getLineMarkerInfo(@NotNull PsiElement element) {
 
+        // 避免LineMarker is supposed to be registered for leaf elements only, but got: XmlTag:select Warning
+        if (!(element instanceof XmlToken) || ((XmlToken)element).getTokenType() != XmlTokenType.XML_START_TAG_START) {
+            return null;
+        }
+
+        PsiElement tag = element.getParent();
+
         // 只处理Mapper接口中的内容
-        if (!MapperDomUtils.isElementWithinMapperXml(element)) {
+        if (!MapperDomUtils.isElementWithinMapperXml(tag)) {
             return null;
         }
 
-        XmlTag xmlTag = (XmlTag) element;
+        XmlTag xmlTag = (XmlTag) tag;
 
         if (!Constant.MYBATIS_OPS.contains(xmlTag.getName())) {
             return null;
         }
 
-        FunctionTooltip tooltip = new FunctionTooltip(element);
+        FunctionTooltip tooltip = new FunctionTooltip(tag);
 
         return new LineMarkerInfo<>(
-                xmlTag.getNavigationElement(),
-                xmlTag.getNavigationElement().getTextRange(),
+                element,
+                element.getTextRange(),
                 IconUtils.GENERATE_ICON,
                 tooltip,
                 new SqlGenerateNavigationHandler(),