引入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;
}