Alawn/mybatis-generator-plugin
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ModelBuilderPlugin 增加xxxKey对应builder

Browse Source
This commit is contained in:
hewei
2017-07-28 15:06:02 +08:00
parent 20225dae8f
commit cf449ce139
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/itfsw/mybatis/generator/plugins/ModelBuilderPlugin.java
+15 -2
View File
@@ -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);
}
/**