hhyykk/RuoYi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

代码生成实体根据模板区分不同父字段

Browse Source
This commit is contained in:
RuoYi
2019-10-15 12:45:55 +08:00
parent 9f5067fbad
commit 69cc7a4111
3 changed files with 22 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
ruoyi-generator/src/main/java/com/ruoyi/generator/domain/GenTable.java
View File

@ -252,4 +252,18 @@ public class GenTable extends BaseEntity
{
return tplCategory != null && StringUtils.equals(GenConstants.TPL_CRUD, tplCategory);
}
public boolean isSuperColumn(String javaField)
{
return isSuperColumn(this.tplCategory, javaField);
}
public static boolean isSuperColumn(String tplCategory, String javaField)
{
if (isTree(tplCategory))
{
StringUtils.equalsAnyIgnoreCase(javaField, GenConstants.TREE_ENTITY);
}
return StringUtils.equalsAnyIgnoreCase(javaField, GenConstants.BASE_ENTITY);
}
}

4
ruoyi-generator/src/main/resources/vm/java/domain.java.vm
View File

@ -28,7 +28,7 @@ public class ${ClassName} extends ${Entity}
private static final long serialVersionUID = 1L;
#foreach ($column in $columns)
#if(!$column.superColumn)
#if(!$table.isSuperColumn($column.javaField))
/** $column.columnComment */
#if($column.list)
#set($parentheseIndex=$column.columnComment.indexOf("（"))
@ -50,7 +50,7 @@ public class ${ClassName} extends ${Entity}
#end
#end
#foreach ($column in $columns)
#if(!$column.superColumn)
#if(!$table.isSuperColumn($column.javaField))
#set($AttrName=$column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})
public void set${AttrName}($column.javaType $column.javaField)
{