[新增] 代码生成工具增生成generateColumnConsts时候,增加DB_XXX 记录字段名,用于pageHelper分页或排序

qiaopan · Jan 25, 2019 · 14327b0 · 14327b0
1 parent 61c21c1
commit 14327b0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/generator/src/main/java/tk/mybatis/mapper/generator/MapperPlugin.java b/generator/src/main/java/tk/mybatis/mapper/generator/MapperPlugin.java
@@ -190,6 +190,15 @@ private void processEntityClass(TopLevelClass topLevelClass, IntrospectedTable i
  field.setInitializationString("\"" + introspectedColumn.getJavaProperty() + "\"");
  context.getCommentGenerator().addClassComment(topLevelClass, introspectedTable);
  topLevelClass.addField(field);
+ //增加字段名常量,用于pageHelper
+ Field columnField = new Field();
+ columnField.setVisibility(JavaVisibility.PUBLIC);
+ columnField.setStatic(true);
+ columnField.setFinal(true);
+ columnField.setName("DB_" + introspectedColumn.getActualColumnName().toUpperCase()); //$NON-NLS-1$
+ columnField.setType(new FullyQualifiedJavaType(String.class.getName())); //$NON-NLS-1$
+ columnField.setInitializationString("\"" + introspectedColumn.getActualColumnName() + "\"");
+ topLevelClass.addField(columnField);
  }
  }
  }