|
@ -24,6 +24,7 @@ Guns采用APACHE LICENSE 2.0开源协议,您在使用过程中,需要注意
|
|
|
*/
|
|
|
package cn.stylefeng.guns.core.validation.flag;
|
|
|
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
import cn.stylefeng.guns.core.enums.YesOrNotEnum;
|
|
|
|
|
|
import javax.validation.ConstraintValidator;
|
|
@ -37,8 +38,27 @@ import javax.validation.ConstraintValidatorContext;
|
|
|
*/
|
|
|
public class FlagValueValidator implements ConstraintValidator<FlagValue, String> {
|
|
|
|
|
|
private Boolean required;
|
|
|
|
|
|
@Override
|
|
|
public void initialize(FlagValue constraintAnnotation) {
|
|
|
this.required = constraintAnnotation.required();
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public boolean isValid(String flagValue, ConstraintValidatorContext context) {
|
|
|
return YesOrNotEnum.Y.getCode().equals(flagValue) || YesOrNotEnum.N.getCode().equals(flagValue);
|
|
|
|
|
|
// 如果是必填的
|
|
|
if (required) {
|
|
|
return YesOrNotEnum.Y.getCode().equals(flagValue) || YesOrNotEnum.N.getCode().equals(flagValue);
|
|
|
} else {
|
|
|
|
|
|
//如果不是必填,可以为空
|
|
|
if (StrUtil.isEmpty(flagValue)) {
|
|
|
return true;
|
|
|
} else {
|
|
|
return YesOrNotEnum.Y.getCode().equals(flagValue) || YesOrNotEnum.N.getCode().equals(flagValue);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|