Merge pull request #1118 from octoape/master

Add clientUpdateByPrimaryKeyMethodGenerated method

Author: jeffgbutler

core/mybatis-generator-core/src/main/java/org/mybatis/generator/api/CompositePlugin.java

@@ -1274,4 +1274,16 @@ public boolean shouldGenerate(IntrospectedTable introspectedTable) {
 
         return true;
     }
+
+    @Override
+    public boolean clientUpdateByPrimaryKeyMethodGenerated(Method method, Interface interfaze,
+            IntrospectedTable introspectedTable) {
+        for (Plugin plugin : plugins) {
+            if (!plugin.clientUpdateByPrimaryKeyMethodGenerated(method, interfaze, introspectedTable)) {
+                return false;
+            }
+        }
+
+        return true;
+    }
 }

core/mybatis-generator-core/src/main/java/org/mybatis/generator/api/Plugin.java

@@ -2044,4 +2044,30 @@ default boolean clientUpdateByPrimaryKeyMethodGenerated(KotlinFunction kotlinFun
     default boolean shouldGenerate(IntrospectedTable introspectedTable) {
         return true;
     }
+
+    /**
+     * The motivation for adding this method can be found in
+     * https://github.com/mybatis/generator/issues/1116
+     *
+     * This method is called when the updateByPrimaryKey method
+     * has been generated in the dynamic SQL runtime client interface.
+     *
+     * @param method
+     *            the generated updateByPrimaryKey method
+     * @param interfaze
+     *            the partially implemented client interface. You can add
+     *            additional imported classes to the interface if
+     *            necessary.
+     * @param introspectedTable
+     *            The class containing information about the table as
+     *            introspected from the database
+     * @return true if the method should be generated, false if the generated
+     *         method should be ignored. In the case of multiple plugins, the
+     *         first plugin returning false will disable the calling of further
+     *         plugins.
+     */
+    default boolean clientUpdateByPrimaryKeyMethodGenerated(Method method,
+            Interface interfaze, IntrospectedTable introspectedTable) {
+        return clientUpdateByPrimaryKeyWithBLOBsMethodGenerated(method, interfaze, introspectedTable);
+    }
 }

core/mybatis-generator-core/src/main/java/org/mybatis/generator/runtime/dynamic/sql/elements/UpdateByPrimaryKeyMethodGenerator.java

@@ -64,7 +64,7 @@ public MethodAndImports generateMethodAndImports() {
 
     @Override
     public boolean callPlugins(Method method, Interface interfaze) {
-        return context.getPlugins().clientUpdateByPrimaryKeyWithBLOBsMethodGenerated(method,
+        return context.getPlugins().clientUpdateByPrimaryKeyMethodGenerated(method,
                 interfaze, introspectedTable);
     }