快捷導(dǎo)航

Springboot中的Validation參數(shù)校驗詳解

更新時間：2023年10月08日 09:47:35 作者：__如風(fēng)__

這篇文章主要介紹了Springboot中的Validation參數(shù)校驗詳解,Springboot參數(shù)校驗是一種常用的驗證機制,在傳遞參數(shù)時進行校驗,以確保參數(shù)的有效性和正確性,該機制可以幫助開發(fā)者在代碼實現(xiàn)前就避免一些常見的錯誤,需要的朋友可以參考下

Springboot 參數(shù)校驗

參考文檔

Springboot項目中，引入依賴

<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-validation</artifactId>
</dependency>

基本的使用網(wǎng)上有很多教程，就不介紹了。

關(guān)于使用

校驗@RequestParam、@PathVariable等，只需在controller類上加@Validated注解
校驗JavaBean，在controlle上注解@Validated且在參數(shù)前加@Validated

下面以該類為例

@Data
public class TestParam {
    /**
     * insert時要求為null
     */
    @Null(groups = Insert.class)
    /**
     * update時要求不為空
     */
    @NotEmpty(groups = Update.class)
    private List<@NotNull Long> ids;
    @EnumValue(value = Gender.class)
    private String gender;
    @Valid
    private List<@NotNull Pet> pets;
    /**
     * 默認(rèn)分組為default，若不繼承
     * Validated指定分組后，非指定分組將失效
     */
    public interface Insert extends Default {
    }
    public interface Update extends Default {
    }
}
@Data
public class Pet {
    @Size(min = 1, max = 5)
    private String name;
}

分組校驗

分組校驗有點類似Jackson序列化的JsonView，不同的情況下采取不同的策略，例如，注冊時賬戶名必須在數(shù)據(jù)庫中唯一，而登錄時賬戶必須在數(shù)據(jù)庫中存在，這個時候就需要為校驗設(shè)置分組了。情況一

@PostMapping("/insert")
public TestParam insert(@RequestBody @Validated(TestParam.Insert.class) TestParam param) {
	return param;
}

情況二

@PostMapping("/update")
public TestParam update(@RequestBody @Validated(TestParam.Update.class) TestParam param) {
        return param;
}

自定義校驗

如下為自定義枚舉值校驗，用于解決屬性定義不是enum類型，但想約束其值為枚舉類型，例如上面的gender，假如定義為Gender，那么默認(rèn)只能MALE或者FEMALE，想要男、女也通過校驗，可以使用該注解

@Target({FIELD, PARAMETER})
@Retention(RUNTIME)
@Documented
@Constraint(validatedBy = {EnumValidator.class})
public @interface EnumValue {
    String message() default "錯誤的枚舉值";
    EnumValidateBy by() default EnumValidateBy.TOSTRING;
    Class<? extends Enum<?>> targetEnum();
    Class<?>[] groups() default {};
    Class<? extends Payload>[] payload() default {};
}
public enum EnumValidateBy {
    /**
     * 使用枚舉類的哪個值校驗
     * 如MAEL("男"), FEMALE("女")
     * NAME 值必須為MAEL或者FEMALE
     * TOSTRING 重寫toString方法返回男或者女， 值必須為男或者女
     * BOTH 以上值均可
     */
    BOTH, NAME, TOSTRING
}

實現(xiàn)ConstraintValidator接口校驗

public class EnumValidator implements ConstraintValidator<EnumValue, String> {
    private final Set<String> set = new HashSet<>(8);
    @Override
    public void initialize(EnumValue constraintAnnotation) {
        Class<? extends Enum<?>> enumCls = constraintAnnotation.targetEnum();
        EnumValidateBy by = constraintAnnotation.by();
        Enum<?>[] constants = enumCls.getEnumConstants();
        if (by == EnumValidateBy.NAME) {
            set.addAll(Arrays.stream(constants).map(Enum::name).collect(Collectors.toSet()));
        } else if (by == EnumValidateBy.TOSTRING) {
            set.addAll(Arrays.stream(constants).map(Enum::toString).collect(Collectors.toSet()));
        } else {
            set.addAll(Arrays.stream(constants).map(Enum::name).collect(Collectors.toSet()));
            set.addAll(Arrays.stream(constants).map(Enum::toString).collect(Collectors.toSet()));
        }
    }
    @Override
    public boolean isValid(String value, ConstraintValidatorContext context) {
        return set.contains(value);
    }
}

枚舉.png

null不在set中，校驗不通過

集合元素校驗

基本類型

直接在泛型參數(shù)上加上需要的注解即可

泛型基本類型.png

JavaBean類型

如僅對整個JavaBean進行校驗，例如@NotNull注解，直接將注解寫在泛型參數(shù)即可
如需對集合中每個JavaBean的屬性進行校驗，還需加上@Valid注解，如上面的pets

泛型JavaBean.png

快速失敗模式

hibernate官網(wǎng)的配置如下：

ValidatorFactory validatorFactory = Validation.byProvider( HibernateValidator.class )
        .configure()
        .addProperty( "hibernate.validator.fail_fast", "true" )
        .buildValidatorFactory();
Validator validator = validatorFactory.getValidator();

只要往容器中注入Validator對象就行了。因為spring自動配置好了，但沒有開啟快速失敗模式，最后，參照Springboot自動配置的ValidationAutoConfiguration，只添加了快速失敗部分，其余未做改動。

@Configuration
public class ValidationConfig {
    @Bean
    public Validator validatorFactory(ApplicationContext context) {
        LocalValidatorFactoryBean factoryBean = new LocalValidatorFactoryBean() {
            @Override
            protected void postProcessConfiguration(javax.validation.Configuration<?> configuration) {
                configuration.addProperty("hibernate.validator.fail_fast", "true");
            }
        };
        MessageInterpolatorFactory interpolatorFactory = new MessageInterpolatorFactory(context);
        factoryBean.setMessageInterpolator(interpolatorFactory.getObject());
        return factoryBean;
    }
}

總結(jié)

Springboot 參數(shù)校驗是一種有效的驗證機制，可以避免常見的錯誤，并提高程序的健壯性和穩(wěn)定性。它的實現(xiàn)方式基于Java Bean驗證框架，通過添加注解的方式，對屬性值進行校驗。開發(fā)者只需要在程序中加入相關(guān)注解即可完成參數(shù)的校驗，從而提高了程序的開發(fā)效率和穩(wěn)定性。