From cf449ce1393da6663aa034abdf0ffd50efaf5de9 Mon Sep 17 00:00:00 2001
From: hewei <hewei@itfsw.com>
Date: Fri, 28 Jul 2017 15:06:02 +0800
Subject: [PATCH] =?UTF-8?q?ModelBuilderPlugin=20=E5=A2=9E=E5=8A=A0xxxKey?=
 =?UTF-8?q?=E5=AF=B9=E5=BA=94builder?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../generator/plugins/ModelBuilderPlugin.java   | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/itfsw/mybatis/generator/plugins/ModelBuilderPlugin.java b/src/main/java/com/itfsw/mybatis/generator/plugins/ModelBuilderPlugin.java
index ba06379..597216a 100644
--- a/src/main/java/com/itfsw/mybatis/generator/plugins/ModelBuilderPlugin.java
+++ b/src/main/java/com/itfsw/mybatis/generator/plugins/ModelBuilderPlugin.java
@@ -51,7 +51,7 @@ public class ModelBuilderPlugin extends BasePlugin {
         List<IntrospectedColumn> columns = introspectedTable.getRules().generateRecordWithBLOBsClass() ? introspectedTable.getNonBLOBColumns() : introspectedTable.getAllColumns();
         InnerClass innerClass = this.generateModelBuilder(topLevelClass, introspectedTable, columns, true);
         topLevelClass.addInnerClass(innerClass);
-        return true;
+        return super.modelBaseRecordClassGenerated(topLevelClass, introspectedTable);
     }
 
     /**
@@ -65,7 +65,20 @@ public class ModelBuilderPlugin extends BasePlugin {
     public boolean modelRecordWithBLOBsClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
         InnerClass innerClass = this.generateModelBuilder(topLevelClass, introspectedTable, introspectedTable.getAllColumns(), false);
         topLevelClass.addInnerClass(innerClass);
-        return true;
+        return super.modelRecordWithBLOBsClassGenerated(topLevelClass, introspectedTable);
+    }
+
+    /**
+     * 具体执行顺序 http://www.mybatis.org/generator/reference/pluggingIn.html
+     * @param topLevelClass
+     * @param introspectedTable
+     * @return
+     */
+    @Override
+    public boolean modelPrimaryKeyClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
+        InnerClass innerClass = this.generateModelBuilder(topLevelClass, introspectedTable, introspectedTable.getPrimaryKeyColumns(), false);
+        topLevelClass.addInnerClass(innerClass);
+        return super.modelPrimaryKeyClassGenerated(topLevelClass, introspectedTable);
     }
 
     /**