SelectSelectivePlugin插件重构

src/main/java/com/itfsw/mybatis/generator/plugins/BatchInsertPlugin.java

@@ -129,7 +129,7 @@ public class BatchInsertPlugin extends BasePlugin {
         XmlElementGeneratorTools.useGeneratedKeys(batchInsertEle, introspectedTable);
 
         batchInsertEle.addElement(new TextElement("insert into " + introspectedTable.getFullyQualifiedTableNameAtRuntime()));
-        for (Element element : XmlElementGeneratorTools.generateKeys(ListUtilities.removeIdentityAndGeneratedAlwaysColumns(introspectedTable.getAllColumns()))) {
+        for (Element element : XmlElementGeneratorTools.generateKeys(ListUtilities.removeIdentityAndGeneratedAlwaysColumns(introspectedTable.getAllColumns()), true)) {
             batchInsertEle.addElement(element);
         }
 

src/main/java/com/itfsw/mybatis/generator/plugins/SelectOneByExamplePlugin.java

@@ -132,9 +132,13 @@ public class SelectOneByExamplePlugin extends BasePlugin {
 
         // 只查询一条
         selectOneElement.addElement(new TextElement("limit 1"));
-        // 添加到根节点
-        FormatTools.addElementWithBestPosition(document.getRootElement(), selectOneElement);
-        logger.debug("itfsw(查询单条数据插件):" + introspectedTable.getMyBatis3XmlMapperFileName() + "增加selectOneByExample方法。");
+
+        // hook
+        if (PluginTools.getHook(ISelectOneByExamplePluginHook.class).sqlMapSelectOneByExampleWithoutBLOBsElementGenerated(document, selectOneElement, introspectedTable)) {
+            // 添加到根节点
+            FormatTools.addElementWithBestPosition(document.getRootElement(), selectOneElement);
+            logger.debug("itfsw(查询单条数据插件):" + introspectedTable.getMyBatis3XmlMapperFileName() + "增加selectOneByExample方法。");
+        }
 
         // ------------------------------------ selectOneByExampleWithBLOBs ----------------------------------
         // !!! 注意这里的行为不以有没有生成Model 的 WithBLOBs类为基准
@@ -179,9 +183,12 @@ public class SelectOneByExamplePlugin extends BasePlugin {
             // 只查询一条
             selectOneWithBLOBsElement.addElement(new TextElement("limit 1"));
 
-            // 添加到根节点
-            FormatTools.addElementWithBestPosition(document.getRootElement(), selectOneWithBLOBsElement);
-            logger.debug("itfsw(查询单条数据插件):" + introspectedTable.getMyBatis3XmlMapperFileName() + "增加selectOneByExampleWithBLOBs方法。");
+            // hook
+            if (PluginTools.getHook(ISelectOneByExamplePluginHook.class).sqlMapSelectOneByExampleWithBLOBsElementGenerated(document, selectOneWithBLOBsElement, introspectedTable)) {
+                // 添加到根节点
+                FormatTools.addElementWithBestPosition(document.getRootElement(), selectOneWithBLOBsElement);
+                logger.debug("itfsw(查询单条数据插件):" + introspectedTable.getMyBatis3XmlMapperFileName() + "增加selectOneByExampleWithBLOBs方法。");
+            }
         }
 
 

src/main/java/com/itfsw/mybatis/generator/plugins/utils/XmlElementGeneratorTools.java

@@ -134,28 +134,17 @@ public class XmlElementGeneratorTools {
      * @return
      */
     public static List<Element> generateKeys(List<IntrospectedColumn> columns) {
-        return generateKeys(columns, null);
+        return generateKeys(columns, false);
     }
 
     /**
      * 生成keys Ele
      * @param columns
-     * @param prefix
-     * @return
-     */
-    public static List<Element> generateKeys(List<IntrospectedColumn> columns, String prefix) {
-        return generateKeys(columns, prefix, true);
-    }
-
-    /**
-     * 生成keys Ele
-     * @param columns
-     * @param prefix
      * @param bracket
      * @return
      */
-    public static List<Element> generateKeys(List<IntrospectedColumn> columns, String prefix, boolean bracket) {
-        return generateCommColumns(columns, prefix, bracket, 1);
+    public static List<Element> generateKeys(List<IntrospectedColumn> columns, boolean bracket) {
+        return generateCommColumns(columns, null, bracket, 1);
     }
 
     /**

src/main/java/com/itfsw/mybatis/generator/plugins/utils/hook/HookAggregator.java

@@ -25,6 +25,7 @@ import org.mybatis.generator.api.dom.java.InnerClass;
 import org.mybatis.generator.api.dom.java.Interface;
 import org.mybatis.generator.api.dom.java.Method;
 import org.mybatis.generator.api.dom.java.TopLevelClass;
+import org.mybatis.generator.api.dom.xml.Document;
 import org.mybatis.generator.api.dom.xml.Element;
 import org.mybatis.generator.api.dom.xml.XmlElement;
 import org.mybatis.generator.config.Context;
@@ -210,7 +211,6 @@ public class HookAggregator implements IUpsertPluginHook, IModelBuilderPluginHoo
 
     // ============================================= ISelectOneByExamplePluginHook ==============================================
 
-
     @Override
     public boolean clientSelectOneByExampleWithBLOBsMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) {
         for (ISelectOneByExamplePluginHook plugin : this.getPlugins(ISelectOneByExamplePluginHook.class)) {
@@ -230,4 +230,24 @@ public class HookAggregator implements IUpsertPluginHook, IModelBuilderPluginHoo
         }
         return true;
     }
+
+    @Override
+    public boolean sqlMapSelectOneByExampleWithoutBLOBsElementGenerated(Document document, XmlElement element, IntrospectedTable introspectedTable) {
+        for (ISelectOneByExamplePluginHook plugin : this.getPlugins(ISelectOneByExamplePluginHook.class)) {
+            if (!plugin.sqlMapSelectOneByExampleWithoutBLOBsElementGenerated(document, element, introspectedTable)) {
+                return false;
+            }
+        }
+        return true;
+    }
+
+    @Override
+    public boolean sqlMapSelectOneByExampleWithBLOBsElementGenerated(Document document, XmlElement element, IntrospectedTable introspectedTable) {
+        for (ISelectOneByExamplePluginHook plugin : this.getPlugins(ISelectOneByExamplePluginHook.class)) {
+            if (!plugin.sqlMapSelectOneByExampleWithBLOBsElementGenerated(document, element, introspectedTable)) {
+                return false;
+            }
+        }
+        return true;
+    }
 }

src/main/java/com/itfsw/mybatis/generator/plugins/utils/hook/ISelectOneByExamplePluginHook.java

@@ -19,6 +19,8 @@ package com.itfsw.mybatis.generator.plugins.utils.hook;
 import org.mybatis.generator.api.IntrospectedTable;
 import org.mybatis.generator.api.dom.java.Interface;
 import org.mybatis.generator.api.dom.java.Method;
+import org.mybatis.generator.api.dom.xml.Document;
+import org.mybatis.generator.api.dom.xml.XmlElement;
 
 /**
  * ---------------------------------------------------------------------------
@@ -48,4 +50,21 @@ public interface ISelectOneByExamplePluginHook {
      */
     boolean clientSelectOneByExampleWithoutBLOBsMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable);
 
+    /**
+     * selectOneByExample 方法sqlMap实现
+     * @param document
+     * @param element
+     * @param introspectedTable
+     * @return
+     */
+    boolean sqlMapSelectOneByExampleWithoutBLOBsElementGenerated(Document document, XmlElement element, IntrospectedTable introspectedTable);
+
+    /**
+     * selectOneByExampleWithBLOBs 方法sqlMap实现
+     * @param document
+     * @param element
+     * @param introspectedTable
+     * @return
+     */
+    boolean sqlMapSelectOneByExampleWithBLOBsElementGenerated(Document document, XmlElement element, IntrospectedTable introspectedTable);
 }