自定义validator分组检验

自定义校验

import com.chensino.core.api.dto.UserLoginDTO;
import com.chensino.core.api.validate.group.PhoneLogin;
import com.chensino.core.api.validate.group.UserNameLogin;
import org.springframework.validation.Errors;
import org.springframework.validation.ValidationUtils;
import org.springframework.validation.Validator;

import java.util.Objects;

/**
 * @Description TODO
 * @Date 2023/5/4 上午9:23
 * @Created by chenxk
 */
public class UserLoginDTOValidator implements Validator {
    @Override
    public boolean supports(Class<?> clazz) {
        return Objects.equals(UserLoginDTO.class, clazz);
    }

    @Override
    public void validate(Object target, Errors errors) {
    }

    /**
     * 分组校验
     * @param target
     * @param errors
     * @param group
     */
    public void validate(Object target, Errors errors,Class<?> group) {
        UserLoginDTO userLoginDTO = (UserLoginDTO) target;
        //PhoneLogin 手机号验证码登录校验
        if (Objects.equals(group, PhoneLogin.class)) {
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "phone", "手机号不能为空");
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "phoneCode", "验证码不能为空");
            if (userLoginDTO.getPhoneCode()!=null && userLoginDTO.getPhoneCode().length() !=6) {
                errors.rejectValue("phoneCode","phoneCode.length.require.six","验证码长度为6");
            }
            if (userLoginDTO.getPhone()!=null && userLoginDTO.getPhone().length() !=11) {
                errors.rejectValue("phone","phone.length.require.six","手机号长度为11");
            }
        }
        //UserNameLogin 用户名密码登录校验
        if (Objects.equals(group, UserNameLogin.class)) {
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "username", "field.required","用户名不能为空");
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "password", "field.required","密码不能为空");
        }
    }
}

调用

自定义validator，可以在任意位置使用，不用非要在controller层。

@Component
@RequiredArgsConstructor
@Slf4j
public class LoginByPhoneStrategy implements LoginStrategy {

    //注入自定义的validator
    private final UserLoginDTOValidator userLoginDTOValidator;
    /**
     * 手机号验证码登录
     *
     * @param userLoginDTO
     * @return
     */
    @Override
    public LoginUserVO login(UserLoginDTO userLoginDTO) {
        //调用，仅校验PhoneLogin分组
        validate(userLoginDTOValidator, PhoneLogin.class,userLoginDTO);
        //....

        return  null;
    }

        /**
     * 校验参数
     *
     * @param userLoginDTOValidator
     * @param clazz
     * @param userLoginDTO
     */
    private void validate(UserLoginDTOValidator userLoginDTOValidator, Class<?> clazz, UserLoginDTO userLoginDTO) {
        BindException errors = new BindException(userLoginDTO, "userLoginDTO");
        userLoginDTOValidator.validate(userLoginDTO, errors, clazz);
        // 判断校验是否成功
        if (errors.hasErrors()) {
            throw new ValidationException(errors);
        }
    }
}