自定义校验注解实现ConstraintValidator

• @Constraint指定校验类。
•  除了自定义的message、require属性外，下面的groups和payload也是必须添加的。
• 注解校验类写在自定义注解中

@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = {QueryCondationValidator.FilterTypeCheck.class})
public @interface QueryCondationValidator {
    String message() default "params.error";

    Class[] groups() default {};

    Class[] payload() default {};

    class FilterTypeCheck implements ConstraintValidator {
        @Override
        public boolean isValid(Integer type, ConstraintValidatorContext constraintValidatorContext) {
            return type != null && (type ==0 || type ==1);
        }
    }
}

@Data
public class StoreUpdQueryDto {

    @NotNull(message = "taskId不可为空")
    Long taskId;

    @NotEmpty(message = "stepTypes")
    List stepTypes;

    @QueryCondationValidator(message = "queryCondition类型只能为0和1")
    Integer queryCondition;

}

• 在valid校验中，如果校验不通过，会产生BindException异常，这边进行全局异常捕获

@RestControllerAdvice
@Slf4j
public class ExceptionControllerAdvice {

    
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    @ExceptionHandler(BindException.class)
    public RestResponse bindExceptionHandler(BindException e) {
        List fieldErrors = e.getBindingResult().getFieldErrors();
        List msgList = fieldErrors.stream()
                .map(o -> o.getDefaultMessage())
                .collect(Collectors.toList());
        String messages = StringUtils.join(msgList.toArray(), ";");
        log.error("Validation格式校验异常: {}", messages);
        return ResponseWarp.error(messages);
    }

    
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    @ExceptionHandler(value = MethodArgumentNotValidException.class)
    public RestResponse handler(MethodArgumentNotValidException e) {
        BindingResult bindingResult = e.getBindingResult();
        ObjectError objectError = bindingResult.getAllErrors().stream().findFirst().get();
        String messages = objectError.getDefaultMessage();
        log.error("MethodArgumentNotValidException异常: {}", messages);
        return ResponseWarp.error(messages);
    }

    
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    @ExceptionHandler(value = ConstraintViolationException.class)
    public RestResponse handler(ConstraintViolationException e) {
        List msgList = new ArrayList<>();
        for (ConstraintViolation constraintViolation : e.getConstraintViolations()) {
            msgList.add(constraintViolation.getMessage());
        }
        String messages = StringUtils.join(msgList.toArray(), ";");
        log.error("entity格式校验异常: {}", messages);
        return ResponseWarp.error(messages);
    }

    @ExceptionHandler(Exception.class)
    public RestResponse methodArgumentNotValidExceptionHandler(Exception e) {
        log.error("ExceptionControllerAdvice Exception: ", e);
        return ResponseWarp.error("统一异常捕获，接口异常" + e.getMessage());
    }

}