文章目录
- 1、定义注解
- 2、使用注解
- 3、其余校验实现思路2.0
- 4、其余校验实现思路3.0
写接口,Dto里很多字段要检验传参范围,自定义个注解来校验。
1、定义注解
注解定义代码:
import javax.validation.Constraint;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import javax.validation.Payload;
import java.lang.annotation.*;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
@Target(ElementType.FIELD)
@Constraint(validatedBy = ListValue.ValidIfInRange.class) //借助@Constraint注解实现自定义校验逻辑,validatedBy属性是数组类型,可同时定义多种校验逻辑
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface ListValue {
String message() default "The parameters are out of range, please check it."; //超出取值范围后的抛错信息
String[] valueList() default {}; //要检验的取值范围
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
class ValidIfInRange implements ConstraintValidator<ListValue, String> {
private final Set<String> set = new HashSet<>();
/**
* 初始化逻辑,把取值范围存入Collection集合
*/
@Override
public void initialize(ListValue constraintAnnotation) {
String[] values = constraintAnnotation.valueList();
set.addAll(Arrays.asList(values));
}
/**
* 校验逻辑,返回false即值不存在,代表超出范围
*/
@Override
public boolean isValid(String s, ConstraintValidatorContext constraintValidatorContext) {
return set.contains(s);
}
}
}
实现思路是使用JSR303校验框架的@Constraint注解,实现ConstraintValidator接口,定义初始化和参数校验逻辑。
<!--PS: JSR303校验框架依赖-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-validation</artifactId>
</dependency>
2、使用注解
@PostMapping("/test")
public void doSome3(@Validated @RequestBody Dto dto){
}
public class Dto {
@ListValue(valueList = {"createTime", "updateTime"}, message ="排序字段超出取值范围")
String orderField;
}
调用下:
3、其余校验实现思路2.0
借用 @JsonCreator 注解,反序列化前端传参成一个枚举对象时,进行校验。
@AllArgsConstructor
@Getter
public enum OrderFieldEnum {
CREATE_TIME("createTime","create_time"),
USER_NAME("userName","user_name");
private final String value;
private final String field;
private static final Map<String,OrderFieldEnum> map = new HashMap<>(3);
@JsonCreator
public static OrderFieldEnum unSerializer(String value){
//把以value为key,以枚举对象为value,存进map
if(map.isEmpty()){
for (OrderFieldEnum fieldEnum : OrderFieldEnum.values()) {
map.put(fieldEnum.value,fieldEnum);
}
}
//map中找不到就是超出范围
if(!map.containsKey(value)){
throw new RuntimeException("超出范围");
}
return map.get(value);
}
@JsonValue
public String getValue(){
return this.value;
}
}
此时Dto:
public class Dto {
OrderFieldEnum orderField;
}
4、其余校验实现思路3.0
这个就比较原始了,直接枚举类定义静态代码块完成取值范围初始化 + 一个静态方法完成校验:
这么实现的话,Service层就得调用方法校验下:
@Service
public Service implements IService{
@Override
public doSome(Dto dto){
FieldEnum.checkCodeExist(dto.getField);
//....
}
}
其余优秀帖子备份:https://juejin.cn/post/7009190724214194207