com.murong.val.config.Aspects Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of verify Show documentation
Show all versions of verify Show documentation
verify project for Spring Boot
package com.murong.val.config;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.murong.val.exception.BusinessException;
import com.murong.val.util.StringUtils;
import com.murong.val.verify.*;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.bind.annotation.RequestBody;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.math.BigDecimal;
import java.util.Arrays;
import java.util.Map;
/**
* aspect for verify
*
* @author murong
* 2019/2/13
*/
@Aspect
@Configuration
public class Aspects {
private static final int parameterCode = -900;
private static final int bodyCode = -600;
@Pointcut("@within(org.springframework.stereotype.Controller) || @within(org.springframework.web.bind.annotation.RestController)")
public void controller() {
}
/**
* @param joinPoint
* @throws InvocationTargetException
* @throws IllegalAccessException
*/
@Before("controller()")
public void valNewParam(JoinPoint joinPoint) throws Throwable {
final Object[] args = joinPoint.getArgs();
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
Method method = signature.getMethod();
Parameter[] parameters = method.getParameters();
method_param:
for (int i = 0; i < parameters.length; i++) {
Parameter parameter = parameters[i];
Object arg = args[i];
ValParam valparam = parameter.getAnnotation(ValParam.class);
if (valparam != null) {
verfyParam(valparam, arg, parameter);
}
// 如果该controller 方法上的参数有包含 @RequestBody,那么就做参数校验
boolean valp = parameter.isAnnotationPresent(RequestBody.class);
if (valp) {
verfyBody(arg, parameter);
}
}
}
/**
* 校验body体
*
* @param arg
* @param parameter
* @throws InvocationTargetException
* @throws IllegalAccessException
*/
public void verfyBody(Object arg, Parameter parameter) throws Throwable {
final Class clazz = parameter.getType();
final Field[] fields = clazz.getDeclaredFields();
final Map map = (JSONObject) JSON.toJSON(arg);
// 接受实体,所有标注有@ValNullable 注解的字段做校验
class_field:
for (int j = 0; j < fields.length; j++) {
final Field field = fields[j];
final Object value = map.get(field.getName());
// 非空
boolean valNull = field.isAnnotationPresent(ValNotNull.class);
if (valNull) {
if (value == null) {
ValNotNull valParam = field
.getAnnotationsByType(ValNotNull.class)[0];
String msg = valParam.msg();
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "is null";
}
throw new BusinessException(bodyCode, msg);
}
}
// 正则
boolean valExp = field.isAnnotationPresent(ValExpression.class);
if (valExp) {
if (value != null) {
ValExpression valParam = field
.getAnnotationsByType(ValExpression.class)[0];
final boolean matches = value.toString()
.matches(valParam.value());
if (!matches) {
String msg = valParam.msg();
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "malformed ";
}
throw new BusinessException(bodyCode, msg);
}
}
}
// 最小
boolean valMin = field.isAnnotationPresent(ValMin.class);
if (valMin) {
if (value != null) {
ValMin valParam = field
.getAnnotationsByType(ValMin.class)[0];
String fieldValue = valParam.value();
BigDecimal min = new BigDecimal(fieldValue);
BigDecimal decimal = new BigDecimal(value.toString());
int i1 = decimal.compareTo(min);
String msg = valParam.msg();
if (valParam.contains() && i1 < 0) {
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "can not less than "
+ fieldValue;
}
throw new BusinessException(bodyCode, msg);
}
if (!valParam.contains() && i1 <= 0) {
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "can not less than or equal to "
+ fieldValue;
}
throw new BusinessException(bodyCode, msg);
}
}
}
// 最大
boolean valMax = field.isAnnotationPresent(ValMax.class);
if (valMax) {
if (value != null) {
ValMax valParam = field
.getAnnotationsByType(ValMax.class)[0];
String fieldValue = valParam.value();
BigDecimal max = new BigDecimal(fieldValue);
BigDecimal decimal = new BigDecimal(value.toString());
int i1 = decimal.compareTo(max);
String msg = valParam.msg();
if (valParam.contains() && i1 > 0) {
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "can not greater to "
+ fieldValue;
}
throw new BusinessException(bodyCode, msg);
}
if (!valParam.contains() && i1 >= 0) {
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "can not greater or equal to "
+ fieldValue;
}
throw new BusinessException(bodyCode, msg);
}
}
}
// in
boolean valIn = field.isAnnotationPresent(ValIn.class);
if (valIn) {
if (value != null) {
ValIn valParam = field
.getAnnotationsByType(ValIn.class)[0];
final boolean contains = Arrays.asList(valParam.value())
.contains(value.toString());
if (!contains) {
String msg = valParam.msg();
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "error with ranges";
}
throw new BusinessException(bodyCode, msg);
}
}
}
// valNoneBlank
boolean valNoneBlank = field
.isAnnotationPresent(ValNotBlank.class);
if (valNoneBlank) {
if (value != null) {
ValNotBlank valParam = field
.getAnnotationsByType(ValNotBlank.class)[0];
if (value.toString().trim().length() == 0) {
String msg = valParam.msg();
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "is blank";
}
throw new BusinessException(bodyCode, msg);
}
}
}
// valStringLen
boolean valStringLen = field
.isAnnotationPresent(ValStringLen.class);
if (valStringLen) {
if (value != null) {
ValStringLen valParam = field
.getAnnotationsByType(ValStringLen.class)[0];
int min = valParam.min();
int max = valParam.max();
int length = value.toString().length();
if (length < min || length > max) {
String msg = valParam.msg();
if (StringUtils.isBlank(valParam.msg())) {
msg = field.getName() + ":" + "error with string length ranges";
}
throw new BusinessException(bodyCode, msg);
}
}
}
}
final Method[] methods = clazz.getDeclaredMethods();
for (int j = 0; j < methods.length; j++) {
Method emethod = methods[j];
ValMethod annotation = emethod.getAnnotation(ValMethod.class);
if (annotation != null) {
try {
emethod.invoke(arg);
} catch (InvocationTargetException e) {
throw e.getTargetException();
}
}
}
}
/**
* 校验参数
*
* @param valparam
* @param arg
* @param parameter
*/
public void verfyParam(ValParam valparam, Object arg, Parameter parameter) {
String name = parameter.getName();
boolean isNullVap = valparam.nullAble();
if (!isNullVap && arg == null) {
throw new BusinessException(parameterCode, name + ": can not be null");
}
if (arg == null) {
return;
}
// 正则
if (!StringUtils.isBlank(valparam.expression())) {
final boolean matches = arg.toString().matches(valparam.expression());
if (!matches) {
String msg = valparam.msg();
if (StringUtils.isBlank(valparam.msg())) {
msg = name + ":" + " malformed ";
}
throw new BusinessException(parameterCode, msg);
}
}
// 最小
if (!StringUtils.isBlank(valparam.min())) {
BigDecimal min = new BigDecimal(valparam.min());
BigDecimal decimal = new BigDecimal(arg.toString());
int i1 = decimal.compareTo(min);
String msg = valparam.msg();
if (valparam.minContains() && i1 < 0) {
if (StringUtils.isBlank(msg)) {
msg = name + ":" + "can not less than " + valparam.min();
}
throw new BusinessException(parameterCode, msg);
}
if (!valparam.minContains() && i1 <= 0) {
if (StringUtils.isBlank(msg)) {
msg = name + ":" + "can not less than or equal to " + valparam.min();
}
throw new BusinessException(parameterCode, msg);
}
}
// 最大
if (!StringUtils.isBlank(valparam.max())) {
BigDecimal max = new BigDecimal(valparam.max());
BigDecimal decimal = new BigDecimal(arg.toString());
int i1 = decimal.compareTo(max);
String msg = valparam.msg();
if (valparam.maxContains() && i1 > 0) {
if (StringUtils.isBlank(msg)) {
msg = name + ":" + "can not greater to " + valparam.max();
}
throw new BusinessException(parameterCode, msg);
}
if (!valparam.maxContains() && i1 >= 0) {
if (StringUtils.isBlank(msg)) {
msg = name + ":" + "can not greater or equal to " + valparam.max();
}
throw new BusinessException(parameterCode, msg);
}
}
// stringLenMin
if (!StringUtils.isBlank(valparam.strLenMin())) {
int min = Integer.parseInt(valparam.strLenMin());
if (arg.toString().length() < min) {
String msg = valparam.msg();
if (StringUtils.isBlank(valparam.msg())) {
msg = name + ":" + "length can not less than " + min;
}
throw new BusinessException(parameterCode, msg);
}
}
// stringLenMax
if (!StringUtils.isBlank(valparam.strLenMax())) {
int max = Integer.parseInt(valparam.strLenMax());
if (arg.toString().length() > max) {
String msg = valparam.msg();
if (StringUtils.isBlank(valparam.msg())) {
msg = name + ":" + "length can not greater to " + max;
}
throw new BusinessException(parameterCode, msg);
}
}
// blankAble
if (!valparam.blankAble()) {
if (arg.toString().trim().length() == 0) {
String msg = valparam.msg();
if (StringUtils.isBlank(valparam.msg())) {
msg = name + ":" + "can not be blank";
}
throw new BusinessException(parameterCode, msg);
}
}
}
}