700字范文,内容丰富有趣,生活中的好帮手!
700字范文 > springboot 自定义注解校验

springboot 自定义注解校验

时间:2019-05-28 15:29:35

相关推荐

springboot 自定义注解校验

1.普通字段自定义注解校验 例如:校验传入值是否在指定的多个值中

1.1.创建注解接口类:

@Retention(RetentionPolicy.RUNTIME)@Target(value = {ElementType.FIELD})@Documented@Constraint(validatedBy = InValuesValidator.class)public @interface InValuesAnnotation {String values();String message() default "不在指定的值中";Class<?>[] groups() default {};Class<? extends Payload>[] payload() default {};}

1.2.创建具体的校验类

public class InValuesValidator implements ConstraintValidator<InValuesAnnotation, Object> {private InValuesAnnotation constraintAnnotation;@Overridepublic void initialize(InValuesAnnotation constraintAnnotation) {this.constraintAnnotation = constraintAnnotation;}@Overridepublic boolean isValid(Object value, ConstraintValidatorContext context) {// 如果没有传入不进行校验if (Func.isNull(value)) return Boolean.TRUE;List<String> values = Arrays.asList(this.constraintAnnotation.values().split(","));boolean result = values.contains(Func.toStr(value));// 如果校验失败就自定义返回消息if (!result) {context.disableDefaultConstraintViolation();context.buildConstraintViolationWithTemplate("不在指定的值" + this.constraintAnnotation.values() + "中").addConstraintViolation();}return result;}}

1.3.在bean中使用

// 状态,0-禁用,1-启用@InValuesAnnotation(values = "0,1")private Integer status;

2.对类自定义注解校验

2.1.注解接口类:

@Retention(RetentionPolicy.RUNTIME)@Target(value = {ElementType.TYPE})@Documented@Constraint(validatedBy = DIYFildValidator.class)public @interface DIYAnnotation {String message() default "!!!!!!No";Class<?>[] groups() default {};Class<? extends Payload>[] payload() default {};}

2.2.具体校验类:

public class DIYFildValidator implements ConstraintValidator<DIYAnnotation, TestValid> {@Overridepublic void initialize(DIYAnnotation constraintAnnotation) {}@Overridepublic boolean isValid(TestValid testValid, ConstraintValidatorContext context) {if(testValid.getInt1() == 0 && nonNull(testValid.getInt2())) {return diyErrMsg(context, "int1为0时,int2必须为null");}if (testValid.getInt1() == 1 && isNull(testValid.getInt2())) {return false;}return true;}// 自定义messageprivate boolean diyErrMsg(ConstraintValidatorContext context, String message) {context.disableDefaultConstraintViolation();context.buildConstraintViolationWithTemplate(message).addConstraintViolation();return false;}}

2.3.将自定义注解加在需要校验的类上:

@Data@DIYAnnotationpublic class TestValid {// 当int1为0时,int2必须为null;当int1为1时,int2必须不为null;private Integer int1;private Integer int2;}

以上就完成了自定义的类校验,别忘了在controller方法上加上@Valid

public int test(@RequestBody @Valid TestValid testValid)

捕获MethodArgumentNotValidException异常,获取错误信息,封装返回

@ExceptionHandler(value = {MethodArgumentNotValidException.class})public @ResponseBodyResponseObject handleValidException(MethodArgumentNotValidException ex, NativeWebRequest request) {BindingResult result = ex.getBindingResult();List<FieldError> fieldErrorList = result.getFieldErrors();List<String> filedErrorInfo = new ArrayList<String>();String message = null;HttpServletRequest requestHttp = request.getNativeRequest(HttpServletRequest.class);for (FieldError filedError : fieldErrorList) {filedErrorInfo.add(filedError.getField());message = filedError.getDefaultMessage();log.error("/data/controller 参数校验失败 路径:{},错误信息:{}", requestHttp.getServletPath(), message);break;}return ResponseObject.getResponseObject(BaseCodeType.PARAM_FAIL, message);}

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。