阅读数:80 评论数:0

一、目标

默认对全局请求参数进行验证，无需加注解@Validated。

二、具体实现

1、定制全局异常处理器GlobalExceptionHandler

/**
 * 全局异常处理
 *
 * @author seer
 * @version 1.0
 * @date 2022/6/9 15:40
 */
@RestControllerAdvice
@Slf4j
public class GlobalExceptionHandler {

    /**
     * 请求方式不支持
     */
    @ExceptionHandler(HttpRequestMethodNotSupportedException.class)
    public Result handleHttpRequestMethodNotSupported(HttpRequestMethodNotSupportedException e, HttpServletRequest request) {
        log.warn("请求方式不支持 {} {}", request.getServletPath(), e.getMethod());
        return Result.fail(e.getMessage());
    }

    /**
     * 参数验证错误
     */
    @ExceptionHandler(BindException.class)
    public Result handleBindException(BindException e, HttpServletRequest request) {
        String message = e.getAllErrors().get(0).getDefaultMessage();
        log.warn("请求参数错误 {} {}", request.getServletPath(), message);
        return Result.fail(Result.Type.REQUEST_PARAM_ERROR, message);
    }

    /**
     * 参数绑定异常
     */
    @ExceptionHandler(MissingServletRequestParameterException.class)
    public Result handleBindException(MissingServletRequestParameterException e, HttpServletRequest request) {
        String message = e.getMessage();
        log.warn("参数绑定异常 {} {}", request.getServletPath(), message);
        return Result.fail(Result.Type.REQUEST_PARAM_ERROR, e.getParameterName() + " 未提供");
    }

    /**
     * 参数验证错误
     */
    @ExceptionHandler(ConstraintViolationException.class)
    public Result handleBindException(ConstraintViolationException e, HttpServletRequest request) {
        String message = e.getMessage();
        log.warn("参数验证错误 {} {}", request.getServletPath(), message);
        return Result.fail(Result.Type.REQUEST_PARAM_ERROR, e.getMessage());
    }
}

2、自定义aop GlobalValidAspect

（1）抛出ConstraintViolationException异常

package com.sdhs.wash.admin.component;

import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.validation.beanvalidation.SpringValidatorAdapter;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
import java.util.Set;

/**
 * 请求参数验证
 * 基本数据类型，由@RequestParam注解，走框架
 * 对象由此校验，不区分是否添加 @RequestBody
 *
 * @author seer
 * @date 2022-06-28 22:06:04
 */
@Aspect
@Component
@Slf4j
public class GlobalValidAspect {

    @Autowired
    private SpringValidatorAdapter validator;

    /**
     * 前置
     *
     * @param joinPoint joinPoint
     */
    @Before("execution(public * com.sdhs.oilcoreserver.admin.controller.*.*(..))")
    public void atBefore(JoinPoint joinPoint) {
        for (Object arg : joinPoint.getArgs()) {
            Set<ConstraintViolation<Object>> violationSet = validator.validate(arg);
            if (!violationSet.isEmpty()) {
                throw new ConstraintViolationException(violationSet.iterator().next().getMessage(), null);
            }
            return;
        }
    }
}