如果新增一个数据,直接在前端页面新增,由于前端代码中有设置数据不能为空,所以不会传入空值。但是不通过前端页面新增一个数据时,比如使用swagger,直接访问后端时,当某个值为空时,可能会被传进数据库,这就会造成一些问题。
org.springframework.boot
spring-boot-starter-validation
2.3.2.RELEASE
(2)在controller层的方法前加上注解@Validated开启数据校验
(3)如果每个方法要校验的参数不同,可以使用分组校验。
实体类上:
每个分组都要创建一个对应的接口:
controller层开启分组校验:
@Validated注解里面支持多个分组。
@Valid注解不支持分组校验
1.自定义注解:
import com.seckill.validator.IsMobileValidator;
import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.*;
@Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER, ElementType.TYPE_USE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(
validatedBy = {IsMobileValidator.class}
)
public @interface IsMobile {
// 默认为true
boolean required() default true;
String message() default "手机号码格式错误";
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
}
2.定义号码的校验类
import org.thymeleaf.util.StringUtils;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* Description:手机号码校验
* Date: 2022/6/28 15:28
**/
public class PhoneNumberValidator {
// 正则表达式
private static final Pattern mobile_pattern=Pattern.compile("^1(3[0-9]|5[0-3,5-9]|7[0-3,5-8]|8[0-9])\\d{8}$");
public static boolean isMobile(String mobile){
if(StringUtils.isEmpty(mobile)){
return false;
}
Matcher matcher = mobile_pattern.matcher(mobile);
return matcher.matches();
}
}
3.自定义校验规则
import com.seckill.annotations.IsMobile;
import com.seckill.utils.PhoneNumberValidator;
import org.thymeleaf.util.StringUtils;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
/**
* Description:自定义校验规则
* Date: 2022/6/28 16:59
**/
public class IsMobileValidator implements ConstraintValidator<IsMobile,String> {
private boolean required=false;
@Override
public void initialize(IsMobile constraintAnnotation) {
//先获取到填的值 true/false
required=constraintAnnotation.required();
}
@Override
public boolean isValid(String s, ConstraintValidatorContext constraintValidatorContext) {
// 判断是否为必填
if(required){ //必填
return PhoneNumberValidator.isMobile(s);
}else { //非必填
if(StringUtils.isEmpty(s)){ //非必填时填的值为空时
return true;
}else{ //非必填时填的值不为空时
return PhoneNumberValidator.isMobile(s);
}
}
}
}
4.捕获数据校验抛出的异常:
/**
* Description:全局异常处理类
* Date: 2022/6/28 17:35
**/
@RestControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler(Exception.class)
public ResponseBean ExceptionHandler(Exception e){
if(e instanceof GlobalException){
GlobalException ex= (GlobalException) e;
return ResponseBean.error(ex.getResponseBeanEnum());
// 处理参数校验抛出的异常 BindException
}else if(e instanceof BindException){
BindException ex= (BindException) e;
ResponseBean res=ResponseBean.error(ResponseBeanEnum.BINDING_ERROR);
res.setMessage("参数校验异常:"+ex.getAllErrors().get(0).getDefaultMessage());
return res;
}
return ResponseBean.error(ResponseBeanEnum.ERROR);
}
}
/** 选手联系电话 */
@IsMobile(message = "联系电话格式不正确")
@NotNull(message = "手机号不能为空")
@Excel(name = "选手联系电话")
private String phoneNumber;
然后在Controller类上或者方法传入的参数前加@Validated或@Valid注解来开启参数校验。
controller层:
加上@Validated
@PostMapping
public AjaxResult addInfo(@RequestBody @Validated TeamInfoDto teamInfoDto)
{
return toAjax(comTeamService.insert(teamInfoDto));
}
实体类中:
在要校验的集合属性上加@Valid这个注解,否则它只会校验这个集合中元素是否为空,不会校验集合中各个元素
@Data
public class TeamInfoDto {
@NotEmpty(message = "选手信息不能为空")
@Valid
private List<ComUser> user;
}
对集合中元素数据的限制设置:
在ComUser实体类中:
在想要校验的属性上加上对应注解
public class ComUser extends BaseEntity
{
private static final long serialVersionUID = 1L;
private Long id;
/** 选手姓名 */
@NotNull(message = "姓名不能为空")
private String name;
/** 选手联系电话 */
@NotNull(message = "手机号不能为空")
@IsMobile(message = "手机格式不正确")
private String phoneNumber;
/** 邮箱 */
@NotEmpty(message = "邮箱不能为空")
private String email;
}