소스 검색

修改实体生成模板

= 8 년 전
부모
커밋
ae8bdd5f4a
2개의 변경된 파일6개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 0
      src/main/java/com/baomidou/mybatisplus/generator/config/po/TableInfo.java
  2. 2 1
      src/main/resources/templates/entity.java.vm

+ 4 - 0
src/main/java/com/baomidou/mybatisplus/generator/config/po/TableInfo.java

@@ -177,6 +177,10 @@ public class TableInfo {
                     // 普通字段
                     pkgSet.add("com.baomidou.mybatisplus.annotations.TableField");
                 }
+                if (null != field.getFill()) {
+                    // 填充字段
+                    pkgSet.add("com.baomidou.mybatisplus.enums.FieldFill");
+                }
             }
             if (!pkgSet.isEmpty()) {
                 this.importPackages = new ArrayList<>(Arrays.asList(pkgSet.toArray(new String[]{})));

+ 2 - 1
src/main/resources/templates/entity.java.vm

@@ -4,7 +4,6 @@ package ${package.Entity};
 import ${pkg};
 #end
 #if(${entityLombokModel})
-import com.baomidou.mybatisplus.enums.FieldFill;
 
 import lombok.Data;
 import lombok.experimental.Accessors;
@@ -107,6 +106,7 @@ public class ${entity} implements Serializable {
 	}
 
 #end
+#if(!${entityLombokModel})
 	@Override
 	public String toString() {
 		return "${entity}{" +
@@ -119,4 +119,5 @@ public class ${entity} implements Serializable {
 #end
 			"}";
 	}
+#end
 }