admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 提取 lombok 配置到模板。

Browse Source
This commit is contained in:
Suomm 2023-06-11 10:25:22 +08:00
parent 6e08986d39
commit e4a042d760
2 changed files with 17 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java
View File

@ -128,18 +128,6 @@ public class Table {
EntityConfig entityConfig = globalConfig.getEntityConfig();
//开启 lombok
if (entityConfig.isWithLombok()) {
//import lombok.AllArgsConstructor;
//import lombok.Builder;
//import lombok.Data;
//import lombok.NoArgsConstructor;
imports.add("lombok.AllArgsConstructor");
imports.add("lombok.Builder");
imports.add("lombok.Data");
imports.add("lombok.NoArgsConstructor");
}
if (entityConfig.getSupperClass() != null) {
imports.add(entityConfig.getSupperClass().getName());
}
@ -277,16 +265,6 @@ public class Table {
*/
public String buildTableAnnotation() {
StringBuilder tableAnnotation = new StringBuilder();
if (globalConfig.getEntityConfig().isWithLombok()) {
//@Data
//@Builder
//@NoArgsConstructor
//@AllArgsConstructor
tableAnnotation.append("@Data\n");
tableAnnotation.append("@Builder\n");
tableAnnotation.append("@NoArgsConstructor\n");
tableAnnotation.append("@AllArgsConstructor\n");
}
tableAnnotation.append("@Table(value = \"").append(name).append("\"");

21
mybatis-flex-codegen/src/main/resources/templates/enjoy/entity.tpl
View File

@ -1,8 +1,15 @@
#set(withLombok = entityConfig.isWithLombok())
package #(packageConfig.entityPackage);
#for(importClass:table.buildImports())
#for(importClass : table.buildImports())
import #(importClass);
#end
#if(withLombok)
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
#end
/**
* #(table.getComment()) 实体类。
@ -10,15 +17,21 @@ import #(importClass);
* @author #(javadocConfig.getAuthor())
* @since #(javadocConfig.getSince())
*/
#if(withLombok)
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
#end
#(table.buildTableAnnotation())
public class #(table.buildEntityClassName())#(table.buildExtends())#(table.buildImplements()) {
#for(column: table.columns)
#for(column : table.columns)
#(column.buildComment())
#(column.buildAnnotations())private #(column.propertySimpleType) #(column.property);
#end
#if(!entityConfig.isWithLombok())
#if(!withLombok)
#for(column: table.columns)
public #(column.propertySimpleType) #(column.getterMethod())() {
return #(column.property);
@ -29,5 +42,5 @@ public class #(table.buildEntityClassName())#(table.buildExtends())#(table.build
}
#end
#end
#end
}