SpringBoot使用hibernate-validator利用AOP实现统一参数校验

引入maven包

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
    <version>1.5.9.RELEASE</version>
</dependency>

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
    <version>2.0.1.RELEASE</version>
</dependency>

<dependency>
    <groupId>org.glassfish</groupId>
    <artifactId>javax.el</artifactId>
    <version>3.0.1-b08</version>
</dependency>

注入Validator的bean

package com.gwghk.starter.web.autoconfigure;

import lombok.extern.slf4j.Slf4j;
import org.hibernate.validator.HibernateValidator;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.validation.Validation;
import javax.validation.Validator;
import javax.validation.ValidatorFactory;

/**
 * @author chihiro.zhang
 */
@Slf4j
@Configuration
public class ValidatorConfiguration {

    @Bean
    public Validator getValidatorFactory(){
        ValidatorFactory validatorFactory = Validation.byProvider(HibernateValidator.class).configure().failFast(true).buildValidatorFactory();
        return validatorFactory.getValidator();
    }
}

添加参数校验的AOP

package com.gwghk.starter.web.validator;

import com.gwghk.common.framework.constant.ReturnCode;
import com.gwghk.common.framework.response.BaseResponse;
import com.gwghk.common.tools.utils.HttpResponseUtil;
import com.gwghk.common.tools.utils.JsonUtil;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletResponse;
import javax.validation.ConstraintViolation;
import javax.validation.Path;
import javax.validation.Validator;
import java.util.Set;

/**
 * @author chihiro.zhang
 */
@Slf4j
@Aspect
@Component
@AllArgsConstructor
public class ValidatorAspect {

    private Validator validator;

      @Pointcut("execution(* com.gwghk..*.web.*.*(..))")
      public void pointcut() {
      }

    /**
     * 入参校验
     * @param joinPoint
     * @throws Throwable
     */
    @Around("pointcut()")
    public Object execute(ProceedingJoinPoint joinPoint) throws Throwable {
        BaseResponse baseResponse = new BaseResponse();
        Object[] args = joinPoint.getArgs();
        for (Object arg : args){
            if(arg != null){
                Set<ConstraintViolation<Object>> constraintViolations = validator.validate(arg);
                if(constraintViolations.size() > 0){
                    baseResponse.setReturnCode(ReturnCode.CODE_1006);
                    for (ConstraintViolation<Object> constraintViolation : constraintViolations) {
                        Path property = constraintViolation.getPropertyPath();
                        String name = property.iterator().next().getName();
                        baseResponse.setMessage("[" + name + "]" + constraintViolation.getMessage());
                        break;
                    }
                    ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
                    HttpServletResponse response = attributes.getResponse();
                    String responseStr = JsonUtil.objectToJson(baseResponse);
                    log.error(responseStr);
                    HttpResponseUtil.setResponseJsonBody(response,responseStr);
                    return null;
                }
            }
        }
        return joinPoint.proceed();
    }
}

上面@Pointcut("execution(* com.gwghk..*.web.*.*(..))")为Controller切点位置
BaseResponse为一个统一返回对象，这个可以自定义

使用

使用是需要再Controller入参对象添加校验注解即可：

@Data
public class RegisteProxyParam {
    @NotEmpty
    private String appId;
}