diff --git a/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java b/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java index e1563c9..1e69879 100644 --- a/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java +++ b/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java @@ -32,6 +32,9 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator { // 处理表名称,替换前缀 String tableName = tableMetaData.getTableName(); String[] prefixes = dto.getTablePrefixes().split("[,,]"); + + tableMetaData.setCleanTableName(tableName); + for (String prefix : prefixes) { if (tableName.startsWith(prefix)) { String handlerTableName = tableName.replace(prefix, ""); @@ -52,9 +55,9 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator { @Override public void addContext(VelocityContext context) { // 当前的表名 - String handlerTableName = tableMetaData.getCleanTableName(); - handlerTableName = StringUtils.hasText(handlerTableName) ? handlerTableName : tableMetaData.getTableName(); - + String cleanTableName = tableMetaData.getCleanTableName(); + cleanTableName = StringUtils.hasText(cleanTableName) ? cleanTableName : tableMetaData.getTableName(); + // 表的注释内容 String comment = tableMetaData.getComment(); @@ -75,11 +78,11 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator { context.put("package", dto.getPackageName()); // 将类名称转成小驼峰 - String lowerCamelCase = MysqlTypeConvertUtil.convertToCamelCase(handlerTableName, false); + String lowerCamelCase = MysqlTypeConvertUtil.convertToCamelCase(cleanTableName, false); context.put("classLowercaseName", lowerCamelCase); // 将类名称转成大驼峰 - String upperCameCase = MysqlTypeConvertUtil.convertToCamelCase(handlerTableName, true); + String upperCameCase = MysqlTypeConvertUtil.convertToCamelCase(cleanTableName, true); context.put("classUppercaseName", upperCameCase); // 添加中划线 diff --git a/generator-code-server/src/main/resources/vms/server/controller/controller.java.vm b/generator-code-server/src/main/resources/vms/server/controller/controller.java.vm index d0867b7..e09e7ab 100644 --- a/generator-code-server/src/main/resources/vms/server/controller/controller.java.vm +++ b/generator-code-server/src/main/resources/vms/server/controller/controller.java.vm @@ -12,6 +12,9 @@ import org.springframework.web.bind.annotation.*; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import lombok.RequiredArgsConstructor; import ${package}.service.${classUppercaseName}Service; +import com.mall.common.domain.vo.result.PageResult; +import com.mall.common.domain.vo.result.Result; +import com.mall.common.domain.vo.result.ResultCodeEnum; import java.util.List; diff --git a/generator-code-server/src/main/resources/vms/server/domain/dto/dto.java.vm b/generator-code-server/src/main/resources/vms/server/domain/dto/dto.java.vm index 4de4590..10ac102 100644 --- a/generator-code-server/src/main/resources/vms/server/domain/dto/dto.java.vm +++ b/generator-code-server/src/main/resources/vms/server/domain/dto/dto.java.vm @@ -6,6 +6,10 @@ import lombok.Builder; import lombok.Data; import lombok.NoArgsConstructor; +import java.time.LocalDateTime; +import java.math.BigDecimal; +import java.util.Date; + @Data @AllArgsConstructor @NoArgsConstructor @@ -14,8 +18,8 @@ import lombok.NoArgsConstructor; public class ${classUppercaseName}Dto { #foreach($field in ${columnInfoList}) - @Schema(name = "${field.lowercaseName}", title = "${field.comment}") - private ${field.javaType} ${field.lowercaseName}; + @Schema(name = "${field.lowercaseName}", title = "${field.comment}") + private ${field.javaType} ${field.lowercaseName}; #end } \ No newline at end of file diff --git a/generator-code-server/src/main/resources/vms/server/domain/entity/entity.java.vm b/generator-code-server/src/main/resources/vms/server/domain/entity/entity.java.vm index 51f65a9..b0bc136 100644 --- a/generator-code-server/src/main/resources/vms/server/domain/entity/entity.java.vm +++ b/generator-code-server/src/main/resources/vms/server/domain/entity/entity.java.vm @@ -9,6 +9,10 @@ import lombok.Getter; import lombok.Setter; import lombok.experimental.Accessors; +import java.time.LocalDateTime; +import java.math.BigDecimal; +import java.util.Date; + @Getter @Setter @Accessors(chain = true) @@ -17,13 +21,11 @@ import lombok.experimental.Accessors; public class ${classUppercaseName} { #foreach($field in ${columnInfoList}) - -@Schema(name = "${field.lowercaseName}", title = "${field.comment}") - -#if($field.isPrimaryKey) -@TableId(type = IdType.ASSIGN_ID) -#end -private ${field.javaType} ${field.lowercaseName}; + @Schema(name = "${field.lowercaseName}", title = "${field.comment}") + #if($field.isPrimaryKey) + @TableId(type = IdType.ASSIGN_ID) + #end + private ${field.javaType} ${field.lowercaseName}; #end } \ No newline at end of file diff --git a/generator-code-server/src/main/resources/vms/server/domain/vo/vo.java.vm b/generator-code-server/src/main/resources/vms/server/domain/vo/vo.java.vm index fc1d185..c222aa0 100644 --- a/generator-code-server/src/main/resources/vms/server/domain/vo/vo.java.vm +++ b/generator-code-server/src/main/resources/vms/server/domain/vo/vo.java.vm @@ -1,21 +1,26 @@ package ${package}.domain.vo; +import java.util.Date; +import java.math.BigDecimal; + import io.swagger.v3.oas.annotations.media.Schema; import lombok.AllArgsConstructor; import lombok.Builder; import lombok.Data; import lombok.NoArgsConstructor; +import java.time.LocalDateTime; + @Data @AllArgsConstructor @NoArgsConstructor @Schema(name = "${classUppercaseName}VO对象", title = "${comment}", description = "${comment}的VO对象") public class ${classUppercaseName}Vo { -#foreach($field in ${columnInfoList}) + #foreach($field in ${columnInfoList}) @Schema(name = "${field.lowercaseName}", title = "${field.comment}") private ${field.javaType} ${field.lowercaseName}; -#end + #end } diff --git a/generator-code-server/src/main/resources/vms/server/service/impl/serviceImpl.java.vm b/generator-code-server/src/main/resources/vms/server/service/impl/serviceImpl.java.vm index ae1d9a3..f26607d 100644 --- a/generator-code-server/src/main/resources/vms/server/service/impl/serviceImpl.java.vm +++ b/generator-code-server/src/main/resources/vms/server/service/impl/serviceImpl.java.vm @@ -12,6 +12,9 @@ import ${package}.service.${classUppercaseName}Service; import ${package}.domain.dto.${classUppercaseName}Dto; import ${package}.domain.entity.${classUppercaseName}; import ${package}.domain.vo.${classUppercaseName}Vo; +import com.mall.common.domain.vo.result.PageResult; +import com.mall.common.domain.vo.result.Result; +import com.mall.common.domain.vo.result.ResultCodeEnum; import java.util.List; diff --git a/generator-code-server/src/main/resources/vms/server/service/service.java.vm b/generator-code-server/src/main/resources/vms/server/service/service.java.vm index 84a3999..03f2daf 100644 --- a/generator-code-server/src/main/resources/vms/server/service/service.java.vm +++ b/generator-code-server/src/main/resources/vms/server/service/service.java.vm @@ -6,6 +6,9 @@ import com.baomidou.mybatisplus.extension.service.IService; import ${package}.domain.dto.${classUppercaseName}Dto; import ${package}.domain.entity.${classUppercaseName}; import ${package}.domain.vo.${classUppercaseName}Vo; +import com.mall.common.domain.vo.result.PageResult; +import com.mall.common.domain.vo.result.Result; +import com.mall.common.domain.vo.result.ResultCodeEnum; import java.util.List;