GroupSequenceProvider动态校验入参

Posted 2023-01-31 好大的月亮

需求场景

平时在controller的入参校验中大多都是单个参数校验
一般都是直接用javax的注解 + javax.validation.Valid注解一把梭。

不过也有场景是需要动态校验。
假设入参的年龄和地区相关。
上海地区的年龄范围需要在10-20
浙江地区的年龄范围需要在10-30

这个时候要么就是直接硬编码了，但是前人肯定遇到过这种问题。

org.hibernate.validator.group.GroupSequenceProvider注解就能帮我们稍微实现的漂亮一点。

如果当前项目中没有对应依赖的话先引入依赖

<dependency>
	<groupId>org.hibernate.validator</groupId>
	<artifactId>hibernate-validator</artifactId>
	<version>6.2.0.Final</version>
</dependency>

代码Demo

自定义编码实现动态校验

package com.xxx.xxx.valid.group;

/**
 * 根据动态条件来决定是否判断 NotNull
 */
public interface NotNullByConditionGroup 





package com.xxx.xxx.xxx;

import com.xxx.xxx.valid.group.NotNullByConditionGroup;
import com.xxx.xxx.valid.ProductAccessoriesSkuQueryProvider;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import org.hibernate.validator.group.GroupSequenceProvider;

import javax.validation.constraints.NotNull;
import java.io.Serializable;


@GroupSequenceProvider(ProductAccessoriesSkuQueryProvider.class)
@Data
public class ProductAccessoriesSkuQueryListDTO implements Serializable 

    private static final long serialVersionUID = 8297881653432975830L;

    @ApiModelProperty(value = "产品id")
    @NotNull(message = "产品id不能为空", groups = NotNullByConditionGroup.class)
    private Long productId;

    @ApiModelProperty(value = "颜色id,详情见数据字典 color")
    @NotNull(message = "颜色id不能为空", groups = NotNullByConditionGroup.class)
    private Long colorId;

    @ApiModelProperty(value = "尺寸id,详情见数据字典 sales_size")
    @NotNull(message = "尺寸id不能为空", groups = NotNullByConditionGroup.class)
    private Long sizeId;

    @ApiModelProperty(value = "产品skuId")
    private Long productSkuId;





package com.xxx.xxx.valid;

import com.xxx.xxx.xxx.ProductAccessoriesSkuQueryListDTO;
import com.xxx.xxx.xxx.group.NotNullByConditionGroup;
import org.hibernate.validator.spi.group.DefaultGroupSequenceProvider;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;


/**
 * 校验当 productSkuId 不存在的时候 sizeId + productId + colorId 不能为空
 */
public class ProductAccessoriesSkuQueryProvider implements DefaultGroupSequenceProvider<ProductAccessoriesSkuQueryListDTO> 

    @Override
    public List<Class<?>> getValidationGroups(ProductAccessoriesSkuQueryListDTO bean) 
        List<Class<?>> defaultGroupSequence = new ArrayList<>();
        defaultGroupSequence.add(ProductAccessoriesSkuQueryListDTO.class); // 这一步不能省,否则Default分组都不会执行了，会抛错的
        if(Objects.nonNull(bean))
            Long productSkuId = bean.getProductSkuId();

            if(Objects.isNull(productSkuId))
                defaultGroupSequence.add(NotNullByConditionGroup.class);
            

        

        return defaultGroupSequence;
    





@PostMapping("queryList")

public CommonResult<List<? extends xxx>> queryList(@RequestBody @Validated ProductAccessoriesSkuQueryListDTO queryList)
    return CommonResult.success();

@Validated指定分组校验

public class ValidatedGroup 
 
    public interface CREATE
 
    public interface DELET
 
    public interface UPDATE
 
    public interface QUERY




import lombok.Data;
 
import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;
 
@Data
public class ValidatedVO 
 
    //更新、删除时不能为空
    @NotNull(message = "id不能为空", groups = ValidatedGroup.UPDATE.class, ValidatedGroup.DELET.class)
    private Long id;
 
    //新增、更新时不能为空
    @NotBlank(message = "name不能为空", groups = ValidatedGroup.CREATE.class ,ValidatedGroup.UPDATE.class)
    private String name;
 
    //查询时不能为空
    @NotBlank(message = "queryParam不能为空", groups = ValidatedGroup.QUERY.class)
    private String queryParam;
 





public ValidatedVO bindValidate(@RequestBody @Validated(value= ValidatedGroup.DELET.class) ValidatedVO validatedVO/*, BindingResult result*/) 
    return validatedVO;