From d2c745d3fdce8a45c8403bbf0301d58fe05b3b35 Mon Sep 17 00:00:00 2001 From: 疯狂的狮子li <15040126243@163.com> Date: 星期五, 14 五月 2021 15:12:40 +0800 Subject: [PATCH] add 代码生成器 增加校验注解 --- ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm | 30 ++++++++++++++++++++++++------ 1 files changed, 24 insertions(+), 6 deletions(-) diff --git a/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm b/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm index 93e2882..89550fe 100644 --- a/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm +++ b/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm @@ -1,15 +1,20 @@ package ${packageName}.bo; -import com.baomidou.mybatisplus.annotation.TableField; import io.swagger.annotations.ApiModel; import io.swagger.annotations.ApiModelProperty; import lombok.Data; +import lombok.EqualsAndHashCode; + import java.util.Date; -import java.util.Map; -import java.util.HashMap; +import javax.validation.constraints.*; #foreach ($import in $importList) import ${import}; +#end +#if($table.crud || $table.sub) +import com.ruoyi.common.core.domain.BaseEntity; +#elseif($table.tree) +import com.ruoyi.common.core.domain.TreeEntity; #end /** @@ -18,9 +23,16 @@ * @author ${author} * @date ${datetime} */ +#if($table.crud || $table.sub) +#set($Entity="BaseEntity") +#elseif($table.tree) +#set($Entity="TreeEntity") +#end + @Data +@EqualsAndHashCode(callSuper = true) @ApiModel("${functionName}鍒嗛〉鏌ヨ瀵硅薄") -public class ${ClassName}QueryBo { +public class ${ClassName}QueryBo extends ${Entity} { /** 鍒嗛〉澶у皬 */ @ApiModelProperty("鍒嗛〉澶у皬") @@ -39,11 +51,17 @@ #foreach ($column in $columns) #if(!$table.isSuperColumn($column.javaField) && $column.query) /** $column.columnComment */ +#if($column.javaType == 'Date') + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") +#end @ApiModelProperty("$column.columnComment") +#if($column.javaType == 'String') + @NotBlank(message = "$column.columnComment涓嶈兘涓虹┖") +#else + @NotNull(message = "$column.columnComment涓嶈兘涓虹┖") +#end private $column.javaType $column.javaField; #end #end - @TableField(exist = false) - private Map<String, Object> params = new HashMap<>(); } -- Gitblit v1.9.3