springboot~ConstraintValidatorContext驗證兩個字段內(nèi)容相同
場景
我在開發(fā)修改密碼功能,通過原密碼和新密碼及確認(rèn)新密碼,希望通過ConstraintValidator這個方式來校驗新密碼和確認(rèn)新密碼,規(guī)則是這兩個密碼需要是相同的。
參考文檔
- https://github.com/micronaut-projects/micronaut-core/issues/3243
- https://stackoverflow.com/questions/37750656/how-to-access-a-field-which-is-described-in-annotation-property
- https://discourse.hibernate.org/t/how-can-i-retrieve-current-validation-contexts-groups-in-a-validator/414/4
實現(xiàn)
定義Matches注解
@Constraint(validatedBy = SameContentMatchesValidator.class)
@Target({ ElementType.FIELD })
@Retention(RetentionPolicy.RUNTIME)
public @interface SameContentMatches {
String message() default "內(nèi)容不一致";
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
String field(); // 新增屬性,指定要比較的字段
}
定義DTO對象
@Data
public class UserModifyPasswordDTO implements UserDTO {
@NotNull
private String userName;
@NotNull
private String password;
private String newPassword;
@SameContentMatches(field = "newPassword")
private String confirmPassword;
}
定義MatchesValidator對象,實現(xiàn)驗證的代碼邏輯
public class SameContentMatchesValidator implements ConstraintValidator<SameContentMatches, String> {
private String field;
@Override
public void initialize(SameContentMatches constraintAnnotation) {
this.field = constraintAnnotation.field();
}
@Override
public boolean isValid(String object, final ConstraintValidatorContext context) {
return true;
}
}
遇到的問題
- 在MatchesValidator類中,無法獲取到當(dāng)前對象,除非把SameContentMatches注解作用到當(dāng)前類上面,而非字段上面。
- 這個問題應(yīng)該主是無法解決的,因為你攔截的是字段,在這個ConstraintValidatorContext處理的都是和當(dāng)前字段有關(guān)的信息
應(yīng)用到類上,代碼調(diào)整,問題解決
@Constraint(validatedBy = SameContentMatchesValidator.class)
@Target({ ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
public @interface SameContentMatches {
String message() default "內(nèi)容不一致";
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
/**
* 源字段名
* @return
*/
String sourceField();
/**
* 目標(biāo)字段名
* @return
*/
String destinationField();
}
public class SameContentMatchesValidator implements ConstraintValidator<SameContentMatches, Object> {
private String sourceField;
private String destinationField;
@Override
public void initialize(SameContentMatches constraintAnnotation) {
this.sourceField = constraintAnnotation.sourceField();
this.destinationField = constraintAnnotation.destinationField();
}
@Override
public boolean isValid(Object o, final ConstraintValidatorContext context) {
final Object sourceFieldVal = BeanUtil.getProperty(o, this.sourceField);
final Object destinationFieldVal = BeanUtil.getProperty(o, this.destinationField);
return sourceFieldVal.equals(destinationFieldVal);
}
}
@Data
@SameContentMatches(sourceField = "confirmPassword", destinationField = "newPassword")
public class UserModifyPasswordDTO implements UserDTO {
@NotNull
private String userName;
@NotNull
private String password;
private String newPassword;
private String confirmPassword;
}
上面的代碼SameContentMatches注解出現(xiàn)了弱編碼,這塊需要再進(jìn)行優(yōu)化。
浙公網(wǎng)安備 33010602011771號