再项目开发中,针对前端传递的参数信息,有些接口中需要写大量的if
判断,导致代码臃肿,不够优雅。
此时,可以使用@Valid
实现基本的字段校验。
spring-boot-starter-web
中。spring-boot-starter-validation
依赖信息springboot 2.1.4
如果你的springboot版本高于 2.3
,需要额外引入下列依赖:
org.springframework.boot spring-boot-starter-validation
验证自带的注解,以及实现原理,可以移步到我的另一篇博客中,本篇博客不做过多的阐述。
做一个优雅的接口
官方提供的一些常用的注解,有时候并不能适合所有的开发需求。此时可以采取自定义valid
的方式,实现其应有的功能。
创建一个自定义的注解
检查排序号是否输入,以及是否满足要求。
import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Constraint(validatedBy = POrderParse.class) // 注解对应的处理类
@Target({ElementType.METHOD, ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
public @interface POrder {// 默认提示语句String message() default "排序号不允许为空,且只允许是1到20的数字!";// 默认校验正则表达式String regexp() default "^([1-9])|([1]\\d)|20$";Class>[] groups() default { };Class extends Payload>[] payload() default { };}
定义注解后,还需要定义其指定的处理类
,如下所示:
import org.hibernate.validator.internal.engine.constraintvalidation.ConstraintValidatorContextImpl;
import org.springframework.stereotype.Component;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;@Component
public class POrderParse implements ConstraintValidator {@Overridepublic void initialize(POrder constraintAnnotation) {System.out.println("my para order validator init");}@Overridepublic boolean isValid(Object value, ConstraintValidatorContext context) {// 校验逻辑ConstraintValidatorContextImpl con = (ConstraintValidatorContextImpl) context;// 获取注解中的属性值Map maps = con.getConstraintDescriptor().getAttributes();// 获取设置的或者默认的正则表达式String regexp = (String) maps.get("regexp");// 获取数据值String param = String.valueOf(value);// 正则判断Pattern regexpVo = Pattern.compile(regexp);Matcher matcher = regexpVo.matcher(param);return matcher.matches();}
}
import cn.xj.bi.volid.MyPhone;
import cn.xj.bi.volid.POrder;
import lombok.Data;@Data
public class User {@POrderprivate Integer order;
}
创建一个测试接口,进行应用测试。
需要使用到
@Valid
注解标识
@RestController
@RequestMapping("/test1")
@Api(tags = "测试")
public class TestController {@PostMapping("/demo4")@ApiOperation(value = "demo4")public CommonResult test4(@RequestBody @Valid User user){return CommonResult.success("6666");}
)
启动项目,进入swagger进行请求测试。
传递满足正则要求
的值,查看返回结果信息。
{"order": 10
}
参数中传递一个不满足正则
表达式的值,观察返回信息。
{"order": 0
}
每次返回这样的报错信息不够直观,此时可以自定义全局异常监听,如下所示:
import cn.xj.bi.vo.CommonResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import java.util.Objects;@Slf4j
@RestControllerAdvice
public class GlobalExceptionHandler {@ExceptionHandler(MethodArgumentNotValidException.class)public CommonResult handleScopeFiledException(MethodArgumentNotValidException e) {log.error("字段合法性校验异常:[{}]", e.getMessage());// getFieldError() 和 getDefaultMessage() 的区别return CommonResult.error( Objects.requireNonNull(e.getBindingResult().getFieldError()).getDefaultMessage(), null);}
}
重启项目,再次异常测试:
递归参数的意思就是接收对象是一个类,假设是DataScope
,但是在这个接收类中,还有一个List
这个参数变量,并且User
中依旧还含有需要valid
校验的字段属性。
再自定义一个valid注解,如下所示:
import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Constraint(validatedBy = {MyPhoneValidtor.class})
@Target({ElementType.METHOD, ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
public @interface MyPhone {String message();Class>[] groups() default {};Class extends Payload>[] payload() default {};}
valid注解具体处理类:
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;public class MyPhoneValidtor implements ConstraintValidator {@Overridepublic boolean isValid(Object o, ConstraintValidatorContext constraintValidatorContext) {System.out.println("校验");// 故意返回false,触发异常return false;}
}
然后再请求参数接收对象中,定义如下格式:
import cn.xj.bi.volid.POrder;
import lombok.Data;@Data
public class User {@POrderprivate Integer order;private Address address;
}
import cn.xj.bi.volid.MyPhone;
import lombok.Data;
import java.io.Serializable;@Data
public class Address implements Serializable {@MyPhone(message = "这只是一个测试")private String phoneNum;
}
重启项目,传递正常的 order
值,观察Address 类中的 phoneNum
是否触发valid校验。
{"order": 10,"address":{"phoneNum":""}
}
发现并未触发对应的valid校验。
解决方式很简单,没有触发说明注解无效,接口中定义@Valid User
是对user对象进行valid处理,但对象类型的并不在列,只需要在对象类型的变量上增加@Valid
注解即可。
import cn.xj.bi.volid.POrder;
import lombok.Data;import javax.validation.Valid;@Data
public class User {@POrderprivate Integer order;@Valid // 迭代validprivate Address address;
}
重启项目,继续按照上面的传参,观察响应信息。
上一篇:ddnsto穿透服务变成开机自启
下一篇:数字信号处理知识点