feat: add index hit rule check

Author: banzhe

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

@@ -41,18 +41,6 @@ public String getSql() {
     }
 
     public void setSql(String sql) {
-        if (!GlobalConfig.isMybatisMode) {
-            this.sql = StringUtils.replace(sql, Constant.DOUBLE_CLICK_PROMPT, StringUtils.EMPTY);
-            return;
-        }
-        if (StringUtils.contains(sql, Constant.DOUBLE_CLICK_PROMPT)) {
-            this.sql = sql;
-        } else {
-            this.sql = Constant.DOUBLE_CLICK_PROMPT + sql;
-        }
-    }
-
-    public void setRawSql(String sql) {
         this.sql = sql;
     }
 

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

@@ -96,7 +96,7 @@ public class Constant {
     );
 
     public static final Map<CheckScopeEnum, List<Report>> DEFAULT_REPORT_MAP = new HashMap<CheckScopeEnum, List<Report>>() {{
-        put(CheckScopeEnum.index, DEFAULT_INDEX_REPORTS);
+        put(CheckScopeEnum.index_field, DEFAULT_INDEX_REPORTS);
         put(CheckScopeEnum.naming_convention, DEFAULT_TABLE_REPORTS);
         put(CheckScopeEnum.select, DEFAULT_SELECT_REPORTS);
         put(CheckScopeEnum.update, DEFAULT_SELECT_REPORTS);
@@ -107,8 +107,6 @@ public class Constant {
 
     public static final String INPUT_SQL_PROMPT = "Please input sql statement.";
 
-    public static final String DOUBLE_CLICK_PROMPT = "-- Double-Click the 'statement', 'result' and 'stress' tab to update the sql with params\n";
-
     public static final String ROOT_NAME = "Mybatis Sql";
 
 }

src/main/java/io/github/linyimin/plugin/sql/checker/Checker.java

@@ -1,5 +1,6 @@
 package io.github.linyimin.plugin.sql.checker;
 
+import com.intellij.openapi.project.Project;
 import io.github.linyimin.plugin.sql.checker.enums.CheckScopeEnum;
 import io.github.linyimin.plugin.sql.checker.rule.CheckRule;
 
@@ -20,11 +21,11 @@ public void registerRule(CheckRule rule) {
         this.rules.add(rule);
     }
 
-    public List<Report> check(String target) {
+    public List<Report> check(Project project, String target) {
         List<Report> reports = new ArrayList<>();
 
         for (CheckRule rule : rules) {
-            Report report = rule.check(target);
+            Report report = rule.check(project, target);
             if (report != null) {
                 reports.add(report);
             }

src/main/java/io/github/linyimin/plugin/sql/checker/CheckerHolder.java

@@ -51,4 +51,17 @@ public static Checker getChecker(CheckScopeEnum scope) {
         return checkers.get(scope);
     }
 
+    public static List<Checker> getCheckers(CheckScopeEnum... scopes) {
+
+        List<Checker> checkerList = new ArrayList<>();
+
+        for (CheckScopeEnum scope : scopes) {
+            if (checkers.containsKey(scope)) {
+                checkerList.add(checkers.get(scope));
+            }
+        }
+
+        return checkerList;
+    }
+
 }

…min/plugin/sql/checker/IndexChecker.java …lugin/sql/checker/IndexFieldChecker.javasrc/main/java/io/github/linyimin/plugin/sql/checker/IndexChecker.java renamed to src/main/java/io/github/linyimin/plugin/sql/checker/IndexFieldChecker.java src/main/java/io/github/linyimin/plugin/sql/checker/IndexChecker.java renamed to src/main/java/io/github/linyimin/plugin/sql/checker/IndexFieldChecker.java

@@ -6,9 +6,9 @@
  * @author banzhe
  * @date 2022/12/14 00:37
  **/
-public class IndexChecker extends Checker {
+public class IndexFieldChecker extends Checker {
     @Override
     CheckScopeEnum scope() {
-        return CheckScopeEnum.index;
+        return CheckScopeEnum.index_field;
     }
 }

src/main/java/io/github/linyimin/plugin/sql/checker/IndexHitChecker.java

@@ -0,0 +1,13 @@
+package io.github.linyimin.plugin.sql.checker;
+
+import io.github.linyimin.plugin.sql.checker.enums.CheckScopeEnum;
+
+/**
+ * @author banzhe
+ **/
+public class IndexHitChecker extends Checker {
+    @Override
+    CheckScopeEnum scope() {
+        return CheckScopeEnum.index_hit;
+    }
+}

src/main/java/io/github/linyimin/plugin/sql/checker/enums/CheckScopeEnum.java

@@ -15,9 +15,8 @@ public enum CheckScopeEnum {
     field,
     field_composition,
 
-    index,
-
-    mapper_xml,
+    index_field,
+    index_hit,
 
     none
 }

src/main/java/io/github/linyimin/plugin/sql/checker/enums/LevelEnum.java

@@ -7,5 +7,6 @@
 public enum LevelEnum {
     mandatory,
     recommended,
-    for_reference
+    for_reference,
+    error
 }

src/main/java/io/github/linyimin/plugin/sql/checker/rule/BigAndCompleteUpdateMethodRule.java

@@ -1,5 +1,6 @@
 package io.github.linyimin.plugin.sql.checker.rule;
 
+import com.intellij.openapi.project.Project;
 import io.github.linyimin.plugin.sql.checker.Report;
 import io.github.linyimin.plugin.sql.checker.enums.CheckScopeEnum;
 import io.github.linyimin.plugin.sql.checker.enums.LevelEnum;
@@ -16,7 +17,7 @@
  **/
 public class BigAndCompleteUpdateMethodRule implements CheckRule {
     @Override
-    public Report check(String target) {
+    public Report check(Project project, String target) {
 
         try {
             Update updateStatement = (Update) CCJSqlParserUtil.parse(target);

src/main/java/io/github/linyimin/plugin/sql/checker/rule/BooleanFieldNameRule.java

@@ -1,6 +1,7 @@
 package io.github.linyimin.plugin.sql.checker.rule;
 
 import com.alibaba.fastjson.JSONObject;
+import com.intellij.openapi.project.Project;
 import io.github.linyimin.plugin.mock.schema.TableField;
 import io.github.linyimin.plugin.sql.checker.Report;
 import io.github.linyimin.plugin.sql.checker.enums.CheckScopeEnum;
@@ -20,7 +21,7 @@ public class BooleanFieldNameRule implements CheckRule {
     private final Pattern START_WITH_IS_PREFIX = Pattern.compile("^is_.*");
 
     @Override
-    public Report check(String target) {
+    public Report check(Project project, String target) {
 
         TableField field = JSONObject.parseObject(target, TableField.class);