Skip to content

M
mybatis-generator-plugin
Commit e0bee6b8 authored by hewei's avatar hewei
Browse files
Options

一些代码精简

parent 6a153b23
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 33 deletions
src/main/java/com/itfsw/mybatis/generator/plugins/ModelBuilderPlugin.java
View file @ e0bee6b8
...@@ -39,7 +39,6 @@ public class ModelBuilderPlugin extends BasePlugin { ...@@ -39,7 +39,6 @@ public class ModelBuilderPlugin extends BasePlugin {
/** /**
* Model Methods 生成 * Model Methods 生成
* 具体执行顺序 http://www.mybatis.org/generator/reference/pluggingIn.html * 具体执行顺序 http://www.mybatis.org/generator/reference/pluggingIn.html
*
* @param topLevelClass * @param topLevelClass
* @param introspectedTable * @param introspectedTable
* @return * @return
...@@ -53,7 +52,7 @@ public class ModelBuilderPlugin extends BasePlugin { ...@@ -53,7 +52,7 @@ public class ModelBuilderPlugin extends BasePlugin {
innerClass.setVisibility(JavaVisibility.PUBLIC); innerClass.setVisibility(JavaVisibility.PUBLIC);
innerClass.setStatic(true); innerClass.setStatic(true);
commentGenerator.addClassComment(innerClass, introspectedTable); commentGenerator.addClassComment(innerClass, introspectedTable);
logger.debug("itfsw(数据Model链式构建插件):"+topLevelClass.getType().getShortName()+"增加内部Builder类。"); logger.debug("itfsw(数据Model链式构建插件):" + topLevelClass.getType().getShortName() + "增加内部Builder类。");
// 构建内部obj变量 // 构建内部obj变量
Field f = JavaElementGeneratorTools.generateField("obj", JavaVisibility.PRIVATE, topLevelClass.getType(), null); Field f = JavaElementGeneratorTools.generateField("obj", JavaVisibility.PRIVATE, topLevelClass.getType(), null);
...@@ -67,35 +66,12 @@ public class ModelBuilderPlugin extends BasePlugin { ...@@ -67,35 +66,12 @@ public class ModelBuilderPlugin extends BasePlugin {
constructor.addBodyLine(new StringBuilder("this.obj = new ").append(topLevelClass.getType().getShortName()).append("();").toString()); constructor.addBodyLine(new StringBuilder("this.obj = new ").append(topLevelClass.getType().getShortName()).append("();").toString());
commentGenerator.addGeneralMethodComment(constructor, introspectedTable); commentGenerator.addGeneralMethodComment(constructor, introspectedTable);
innerClass.addMethod(constructor); innerClass.addMethod(constructor);
logger.debug("itfsw(数据Model链式构建插件):"+topLevelClass.getType().getShortName()+".Builder增加的构造方法。"); logger.debug("itfsw(数据Model链式构建插件):" + topLevelClass.getType().getShortName() + ".Builder增加的构造方法。");
// ！！可能Model存在复合主键情况，字段要加上这些
if (topLevelClass.getSuperClass() != null && topLevelClass.getSuperClass().compareTo(new FullyQualifiedJavaType(introspectedTable.getPrimaryKeyType())) == 0){
for (IntrospectedColumn introspectedColumn : introspectedTable.getPrimaryKeyColumns()) {
Field field = JavaBeansUtil.getJavaBeansField(introspectedColumn, context, introspectedTable);
Method setterMethod = JavaBeansUtil.getJavaBeansSetter(introspectedColumn, context, introspectedTable);
Method method = JavaElementGeneratorTools.generateMethod( for (IntrospectedColumn introspectedColumn : introspectedTable.getAllColumns()) {
field.getName(), Field field = JavaBeansUtil.getJavaBeansField(introspectedColumn, context, introspectedTable);
JavaVisibility.PUBLIC, Method setterMethod = JavaBeansUtil.getJavaBeansSetter(introspectedColumn, context, introspectedTable);
innerClass.getType(),
new Parameter(field.getType(), field.getName())
);
commentGenerator.addSetterComment(method, introspectedTable, introspectedColumn);
method = JavaElementGeneratorTools.generateMethodBody(
method,
"obj." + setterMethod.getName() + "(" + field.getName() + ");",
"return this;"
);
innerClass.addMethod(method);
logger.debug("itfsw(数据Model链式构建插件):"+topLevelClass.getType().getShortName()+".Builder增加"+method.getName()+"方法(复合主键)。");
}
}
// 根据Model属性生成链式赋值方法
for (Field field : fields) {
if (field.isStatic())
continue;
Method method = JavaElementGeneratorTools.generateMethod( Method method = JavaElementGeneratorTools.generateMethod(
field.getName(), field.getName(),
...@@ -103,14 +79,14 @@ public class ModelBuilderPlugin extends BasePlugin { ...@@ -103,14 +79,14 @@ public class ModelBuilderPlugin extends BasePlugin {
innerClass.getType(), innerClass.getType(),
new Parameter(field.getType(), field.getName()) new Parameter(field.getType(), field.getName())
); );
commentGenerator.addGeneralMethodComment(method, introspectedTable); commentGenerator.addSetterComment(method, introspectedTable, introspectedColumn);
method = JavaElementGeneratorTools.generateMethodBody( method = JavaElementGeneratorTools.generateMethodBody(
method, method,
"obj." + field.getName() + " = " + field.getName() + ";", "obj." + setterMethod.getName() + "(" + field.getName() + ");",
"return this;" "return this;"
); );
innerClass.addMethod(method); innerClass.addMethod(method);
logger.debug("itfsw(数据Model链式构建插件):"+topLevelClass.getType().getShortName()+".Builder增加"+method.getName()+"方法。"); logger.debug("itfsw(数据Model链式构建插件):" + topLevelClass.getType().getShortName() + ".Builder增加" + method.getName() + "方法(复合主键)。");
} }
Method build = JavaElementGeneratorTools.generateMethod( Method build = JavaElementGeneratorTools.generateMethod(
...@@ -121,7 +97,7 @@ public class ModelBuilderPlugin extends BasePlugin { ...@@ -121,7 +97,7 @@ public class ModelBuilderPlugin extends BasePlugin {
build.addBodyLine("return this.obj;"); build.addBodyLine("return this.obj;");
commentGenerator.addGeneralMethodComment(build, introspectedTable); commentGenerator.addGeneralMethodComment(build, introspectedTable);
innerClass.addMethod(build); innerClass.addMethod(build);
logger.debug("itfsw(数据Model链式构建插件):"+topLevelClass.getType().getShortName()+".Builder增加build方法。"); logger.debug("itfsw(数据Model链式构建插件):" + topLevelClass.getType().getShortName() + ".Builder增加build方法。");
topLevelClass.addInnerClass(innerClass); topLevelClass.addInnerClass(innerClass);
return true; return true;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment