feat: support RowBounds

Author: banzhe

src/main/java/io/github/linyimin/plugin/component/SqlParamGenerateComponent.java

@@ -74,18 +74,21 @@ public static ProcessResult<MybatisSqlConfiguration> generate(PsiElement psiElem
             configuration.setPsiElement(psiElement);
             configuration.setMethod(statementId);
             configuration.setParams("{}");
+            configuration.setRowBounds(false);
 
             return ProcessResult.success(configuration);
 
         }
 
         String params = generateMethodParam(psiMethod, parser);
+        boolean isRowBounds = isRowBounds(psiMethod);
 
         if (cache) {
             sqlConfig.setPsiElement(psiElement);
             sqlConfig.setMethod(statementId);
             sqlConfig.setParams(params);
             sqlConfig.setUpdateSql(true);
+            sqlConfig.setRowBounds(isRowBounds);
             if (parser instanceof RandomPOJO2JSONParser) {
                 sqlConfig.setDefaultParams(false);
             }
@@ -100,10 +103,19 @@ public static ProcessResult<MybatisSqlConfiguration> generate(PsiElement psiElem
         configuration.setPsiElement(psiElement);
         configuration.setMethod(statementId);
         configuration.setParams(params);
+        configuration.setRowBounds(isRowBounds);
 
         return ProcessResult.success(configuration);
     }
 
+    private static boolean isRowBounds(PsiMethod method) {
+        List<ParamNameType> paramNameTypes = getMethodBodyParamList(method);
+        return paramNameTypes.stream().anyMatch(paramNameType -> {
+           String classQualifier = paramNameType.psiType.getCanonicalText();
+           return StringUtils.equals(classQualifier, "org.apache.ibatis.session.RowBounds");
+        });
+    }
+
     public static ProcessResult<String> generateSql(Project project, String methodQualifiedName, String params, boolean cache) {
 
         try {
@@ -200,7 +212,14 @@ private static ProcessResult<String> getSqlFromXml(Project project, String quali
                 return ProcessResult.fail(String.format("Oops! There is not %s in mapper file!!!", qualifiedMethod));
             }
 
-            return ProcessResult.success(sqlSourceMap.get(qualifiedMethod).getSql(params));
+            String sql = sqlSourceMap.get(qualifiedMethod).getSql(params);
+
+            MybatisSqlConfiguration sqlConfig = project.getService(MybatisSqlStateComponent.class).getConfiguration();
+            if (sqlConfig.isRowBounds()) {
+                sql = String.format("%s\nLIMIT 0, 10", sql);
+            }
+
+            return ProcessResult.success(sql);
         } catch (Throwable t) {
             StringWriter sw = new StringWriter();
             t.printStackTrace(new PrintWriter(sw));

src/main/java/io/github/linyimin/plugin/configuration/model/MybatisSqlConfiguration.java

@@ -19,6 +19,7 @@ public class MybatisSqlConfiguration {
     private String result;
     private boolean defaultParams;
     private boolean updateSql = true;
+    private boolean rowBounds = false;
 
     public String getMethod() {
         return method;
@@ -85,4 +86,12 @@ public boolean isUpdateSql() {
     public void setUpdateSql(boolean updateSql) {
         this.updateSql = updateSql;
     }
+
+    public boolean isRowBounds() {
+        return rowBounds;
+    }
+
+    public void setRowBounds(boolean rowBounds) {
+        this.rowBounds = rowBounds;
+    }
 }

src/main/java/io/github/linyimin/plugin/mybatis/scripting/tags/DynamicContext.java

@@ -94,7 +94,7 @@ public int getUniqueNumber() {
     static class ContextMap extends HashMap<String, Object> {
 
         private static final long serialVersionUID = 2977601501966151582L;
-        private final JSONObject parameterJSONObject;
+        private JSONObject parameterJSONObject;
         private final Object parameterObject;
 
         public ContextMap(Object parameterObject) {
@@ -118,6 +118,7 @@ public ContextMap(Object parameterObject) {
 
             if (JSONObject.isValidObject(parameterObjectStr)) {
                 this.parameterJSONObject = JSONObject.parseObject(parameterObjectStr);
+                this.processRowBounds();
             } else {
                 this.parameterJSONObject = null;
             }
@@ -150,6 +151,30 @@ public Object get(Object key) {
 
             return null;
         }
+
+        private void processRowBounds() {
+            if (this.parameterJSONObject.size() != 2) {
+                return;
+            }
+            for (Map.Entry<String, Object> entry : this.parameterJSONObject.entrySet()) {
+                if (!(entry.getValue() instanceof JSONObject)) {
+                    return;
+                }
+                JSONObject json = (JSONObject) entry.getValue();
+                if (json.size() == 2 && json.containsKey("limit") && json.containsKey("offset")) {
+                    this.parameterJSONObject.remove(entry.getKey());
+                }
+            }
+
+            if (this.parameterJSONObject.size() != 1) {
+                return;
+            }
+
+            Object obj = new ArrayList<>(this.parameterJSONObject.values()).get(0);
+            if (obj instanceof JSONObject) {
+                parameterJSONObject = (JSONObject) obj;
+            }
+        }
     }
 
     static class ContextAccessor implements PropertyAccessor {

src/main/java/io/github/linyimin/plugin/mybatis/xml/XMLIncludeTransformer.java

@@ -23,9 +23,6 @@ public XMLIncludeTransformer(Map<String, XNode> sqlFragments) {
     }
 
     public void applyIncludes(Node source) {
-//        Properties variablesContext = new Properties();
-//        Properties configurationVariables = configuration.getVariables();
-//        Optional.ofNullable(configurationVariables).ifPresent(variablesContext::putAll);
         applyIncludes(source, new Properties(), false);
     }
 

src/main/java/io/github/linyimin/plugin/ui/tree/TreeListener.java

@@ -24,7 +24,6 @@
 import io.github.linyimin.plugin.sql.result.SelectResult;
 import io.github.linyimin.plugin.ui.MouseCursorAdapter;
 import io.github.linyimin.plugin.ui.MybatisSqlScannerPanel;
-import org.apache.commons.lang3.StringUtils;
 import org.jetbrains.annotations.NotNull;
 
 import javax.swing.*;
@@ -38,7 +37,6 @@
 
 /**
  * @author banzhe
- * @date 2023/01/02 01:03
  **/
 public class TreeListener extends MouseAdapter {