业务需求:在controller中对必填参数和参数类型做合法性校验
1、在项目父工程添加spring-boot-starter-validation的依赖
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-validation</artifactId>
</dependency>
2、在引入的validation依赖包下有很多这样的校验注解,直接使用注解定义校验规则即可。
3、现在对下面业务中的文章发布功能做参数校验
package com.heima.wemedia.controller.v1;import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.wemedia.model.po.WmNewsMaterial;
import com.heima.wemedia.service.WmNewsService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;@RestController
@RequestMapping("/api/v1/news")
public class WmNewsController {@AutowiredWmNewsService wmNewsService;/*文章发布*/@PostMapping("/submit")public ResponseResult addNews(@RequestBody WmNewsDto dto){return wmNewsService.addNews(dto);}}
4、可以看到上面接口使用WmNewsDto模型对象接收参数,所以进入WmNewsDto类,在属性上添加校验规则。
下面我使用了NotEmpty注解验证非空
package com.heima.model.common.wemedia.model.dto;import lombok.Data;import javax.validation.constraints.NotEmpty;
import java.util.Date;
import java.util.List;@Data
public class WmNewsDto {private Integer id;/*** 标题*/@NotEmpty(message = "文章标题不能为空")private String title;/*** 频道id*/private Integer channelId;/*** 标签*/@NotEmpty(message = "标签不能为空")private String labels;/*** 发布时间*/private Date publishTime;/*** 文章内容*/@NotEmpty(message = "文章内容不能为空")private String content;/*** 文章封面类型 0 无图 1 单图 3 多图 -1 自动*/private Short type;/*** 提交时间*/private Date submitedTime; /*** 状态 提交为1 草稿为0*/private Short status;/*** 封面图片列表 多张图以逗号隔开*/private List<String> images;
}
5、还需要再controller中添加@Validated 注解开启参数校验
/*文章发布*/@PostMapping("/submit")public ResponseResult addNews(@RequestBody @Validated WmNewsDto dto){return wmNewsService.addNews(dto);}
6、如果校验出错Spring会抛出MethodArgumentNotValidException异常,我们需要在统一异常处理器中捕获异常,解析出异常信息。
代码如下
@ResponseBody
@ExceptionHandler(MethodArgumentNotValidException.class)
@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
public RestErrorResponse methodArgumentNotValidException(MethodArgumentNotValidException e) {BindingResult bindingResult = e.getBindingResult();List<String> msgList = new ArrayList<>();//将错误信息放在msgListbindingResult.getFieldErrors().stream().forEach(item->msgList.add(item.getDefaultMessage()));//拼接错误信息String msg = StringUtils.join(msgList, ",");log.error("【系统异常】{}",msg);return new RestErrorResponse(msg);
}