diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java index a05fdf5f..847c035b 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java @@ -154,6 +154,16 @@ public class Table { return imports.stream().sorted(Comparator.naturalOrder()).collect(Collectors.toList()); } + public String buildRemarks(){ + if (StringUtil.isBlank(remarks)){ + return ""; + }else { + StringBuilder sb = new StringBuilder("/**\n") + .append(" * ").append(remarks).append("\n") + .append(" */"); + return sb.toString(); + } + } /** * 构建 entity 的 Class 名称 diff --git a/mybatis-flex-codegen/src/main/resources/templates/enjoy/entity.tpl b/mybatis-flex-codegen/src/main/resources/templates/enjoy/entity.tpl index 2816dd43..ef2cd733 100644 --- a/mybatis-flex-codegen/src/main/resources/templates/enjoy/entity.tpl +++ b/mybatis-flex-codegen/src/main/resources/templates/enjoy/entity.tpl @@ -4,7 +4,7 @@ package #(globalConfig.entityPackage); import #(importClass); #end - +#(table.buildRemarks()) #(table.buildTableAnnotation()) public class #(table.buildEntityClassName())#(table.buildExtends())#(table.buildImplements()) { #for(column: table.columns)