Springboot——@valid 做字段校验和自定义注解
迪丽瓦拉
2024-05-30 02:26:23
0

文章目录

  • 前言
  • 注意实现
  • 测试环境
  • 验证自带的注解
  • 自定义valid注解
    • 自定义注解和处理类
    • 创建参数接收类,并增加字段注解
    • 接口中使用
  • 自测环节
    • 正常测试
    • 异常测试
  • 自定义全局异常监听
  • 扩展
    • 递归参数下valid不识别的坑

前言

再项目开发中,针对前端传递的参数信息,有些接口中需要写大量的if判断,导致代码臃肿,不够优雅。

此时,可以使用@Valid实现基本的字段校验。

注意实现

  • springboot 2.3之前 ,直接进行开发即可,无需引用额外的依赖
    集成在spring-boot-starter-web中。
  • springboot 2.3之后 需要额外引入spring-boot-starter-validation依赖信息

测试环境

springboot 2.1.4

如果你的springboot版本高于 2.3,需要额外引入下列依赖:

org.springframework.bootspring-boot-starter-validation

验证自带的注解

验证自带的注解,以及实现原理,可以移步到我的另一篇博客中,本篇博客不做过多的阐述。

做一个优雅的接口

自定义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 = 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[] 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);}
}

重启项目,再次异常测试:
在这里插入图片描述

扩展

递归参数下valid不识别的坑

递归参数的意思就是接收对象是一个类,假设是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[] 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;
}

重启项目,继续按照上面的传参,观察响应信息。
在这里插入图片描述

相关内容