SpringBoot自定义校验注解

2023-11-16 14:22:22

自定义校验注解

1. 编写自定义注解

JAVA校验注解的规范，必须包含String message() Class<?>[] groups() Class<? extends Payload>[] payload()

@ListValue的作用是只能接收在数组的值

import static java.lang.annotation.ElementType.*;
import static java.lang.annotation.ElementType.TYPE_USE;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

@Documented
@Constraint(validatedBy = { ListValueConstrainValidator.class}) //可以指定多个不同的校验器，适配不同的数据类型
@Target({ METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE })
@Retention(RUNTIME)
public @interface ListValue {

    // 在ValidationMessages.properties文件中的字段
    String message() default "{com.atguigu.common.valid.ListValue.message}";

    Class<?>[] groups() default { };

    Class<? extends Payload>[] payload() default { };

    int[] vals() default {};
}

通过@Constraint 与自己编写的校验器相关联，可以指定多个不同的校验器，适配不同的数据类型

2. 编写自定义校验器

实现 ConstraintValidator<ListValue,Integer>接口（Ctrl+Alt+B 查看接口的全部实现类）

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.HashSet;
import java.util.Set;

/**
 * @author 廖家银
 * @version 1.0
 * @date 2022/2/4 17:16
 */
public class ListValueConstrainValidator implements ConstraintValidator<ListValue,Integer> {

    private Set<Integer> set =new HashSet<>();
    //初始化方法
    @Override
    public void initialize(ListValue constraintAnnotation) {
        int[] vals = constraintAnnotation.vals();
        for (int val : vals) {
            set.add(val);
        }
    }

    // 判断是否校验成功

    /**
     *
     * @param value 需要校验的值
     * @param context
     * @return
     */

    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
        return set.contains(value);
    }
}

3. 使用自定义校验注解

3.1 在实体类的属性上添加@ValueList

@ListValue(vals={0,1}, groups = {AddGroup.class, UpdateGroup.class})
private Integer showStatus;

groups 选择校验的组， AddGroupt和UpdateGroup是自定义的空接口

3.2 校验Controller方法的参数

使用我们自定义的注解来校验Controller类接收的实体类的值

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

import com.atguigu.common.valid.AddGroup;
import com.atguigu.common.valid.UpdateGroup;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.BindingResult;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import com.atguigu.gulimall.product.entity.BrandEntity;
import com.atguigu.gulimall.product.service.BrandService;
import com.atguigu.common.utils.PageUtils;
import com.atguigu.common.utils.R;

import javax.validation.Valid;


/**
 * 品牌
 *
 * @author liaojiaying
 * @email 2254536570@qq.com
 * @date 2022-01-19 00:28:30
 */
@RestController
@RequestMapping("product/brand")
public class BrandController {
    @Autowired
    private BrandService brandService;

    @RequestMapping("/save")
    // @RequiresPermissions("product:brand:save")
    public R save(@Validated({AddGroup.class}) @RequestBody BrandEntity brand){
      brandService.save(brand);

        return R.ok();
    }
}

@Valid不需要指定使用校验的组，全部校验注解都生效

@Validated在后面指定要使用校验的组，如果在实体类中的属性的校验注解不带groups参数的话，则不会使用这个校验注解

码农公寓