springboot~ConstraintValidatorContext驗證兩(liang)個字(zi)段內容相同
場景
我在開發修改密(mi)碼(ma)功能,通過(guo)原密(mi)碼(ma)和新(xin)密(mi)碼(ma)及確(que)認新(xin)密(mi)碼(ma),希望通過(guo)ConstraintValidator這(zhe)個方式(shi)來校驗新(xin)密(mi)碼(ma)和確(que)認新(xin)密(mi)碼(ma),規則是(shi)這(zhe)兩個密(mi)碼(ma)需要是(shi)相同的。
參考文檔
實現
定義Matches注解
@Constraint(validatedBy = SameContentMatchesValidator.class)
@Target({ ElementType.FIELD })
@Retention(RetentionPolicy.RUNTIME)
public @interface SameContentMatches {
String message() default "內容不一致";
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對象,實現驗證的代碼邏輯
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類中,無法獲取到當前對象,除非把SameContentMatches注解作用到當前類上面,而非字段上面。
- 這個問題應該主是無法解決的,因為你攔截的是字段,在這個ConstraintValidatorContext處理的都是和當前字段有關的信息
應用到類上,代碼調整,問題解決
@Constraint(validatedBy = SameContentMatchesValidator.class)
@Target({ ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
public @interface SameContentMatches {
String message() default "內容不一致";
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
/**
* 源字段名
* @return
*/
String sourceField();
/**
* 目標字段名
* @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注(zhu)解出現了弱編碼,這塊(kuai)需要再進行優化。