spring AOP实现@Around输出请求参数和返回参数

目录

• @Around输出请求参数和返回参数
• 先把我的打印日志代码贴出来
• 测试
• spring AOP中Around切面处理参数
• 解决的办法
• 具体的代码

@Around输出请求参数和返回参数

spring 的AOP是通过cglib动态代理和jdk的动态代理实现的。

先把我的打印日志代码贴出来

package com.zhd.exploit.api.config;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import javax.servlet.http.HttpServletResponseWrapper;
import org.aspectj.lang.JoinPoint;
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.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.mvc.method.annotation.ExtendedServletRequestDataBinder;
import com.alibaba.fastjson.JSONObject;
@Aspect
@Component
@Order(1)
public class ControllerLogInterceptor {
	private static final Logger log = LoggerFactory.getLogger(ControllerLogInterceptor.class);
    //创建Pointcut表示式，表示所有controller请求
	@Pointcut("execution(* com..*.controller..*(..))")
	private void controllerAspect() {
	}// 请求method前打印内容
	@Around(value = "controllerAspect()")
	public void around(ProceedingJoinPoint pjp) throws Throwable {
	//通过uuid关联请求参数和返回参数
		String uuid = UUID.randomUUID().toString().replaceAll("-", "");
		methodBefore(pjp, uuid);
		try {
			Object proceed = pjp.proceed();
			methodAfterReturing(proceed, uuid);
		} catch (Exception e) {
			log.error("[{}]Response异常内容:{}", uuid, e);
			throw e;
		}
	}
	public void methodBefore(JoinPoint joinPoint, String uuid) {
		// 打印请求内容
		try {
			// 下面两个数组中，参数值和参数名的个数和位置是一一对应的。
			Object[] objs = joinPoint.getArgs();
			String[] argNames = ((MethodSignature) joinPoint.getSignature()).getParameterNames(); // 参数名
			Map<String, Object> paramMap = new HashMap<String, Object>();
			for (int i = 0; i < objs.length; i++) {
				if (!(objs[i] instanceof ExtendedServletRequestDataBinder) && !(objs[i] instanceof HttpServletResponseWrapper)) {
					paramMap.put(argNames[i], objs[i]);
				}
			}
			if (paramMap.size() > 0) {
				log.info("\n[{}]方法:{}\n参数:{}", uuid, joinPoint.getSignature(), JSONObject.toJSONString(paramMap));
			}
		} catch (Exception e) {
			log.error("[{}]AOP methodBefore:", uuid, e);
		}
	}
	public void methodAfterReturing(Object o, String uuid) {
		try {
			if (o != null)
				log.info("[{}]Response内容:{}", uuid, JSONObject.toJSON(o));
		} catch (Exception e) {
			log.error("[{}]AOP methodAfterReturing:", uuid, e);
		}
	}
}

测试

请求参数类型1

    @RequestMapping(value = "/test0", method = RequestMethod.GET, produces = { MediaType.APPLICATION_JSON_VALUE })
    public Result test0(@RequestParam String name, @RequestParam String password) {
        System.out.println("test0 OK");
        return new Result("1", "mock a Result");
    }

打印日志：

[fe7155a3089b4dd7896b759a933cf958]方法:Result com.zhd.exploit.api.controller.TestController.test0(String,String)
参数:{"password":"123","name":"zhang"}

请求参数类型2

    @RequestMapping(value = "/test1", method = RequestMethod.GET, produces = { MediaType.APPLICATION_JSON_VALUE })
    public Result test1(PayDTO payDTO) {
        System.out.println("test1 OK");
        return new Result("1", "mock a Result");
    }

打印日志：

[a2f7d19dea834c54a45b480bd4e8c3cd]方法:Result com.zhd.exploit.api.controller.TestController.test1(PayDTO)
参数:{"payDTO":{"appmount":"10","paytype":"1"}}

请求参数类型3

    @RequestMapping(value = "/test2", method = RequestMethod.POST, produces = { MediaType.APPLICATION_JSON_VALUE })
    public Result test2(@RequestBody PayDTO payDTO) {
        System.out.println("test2 OK");
        return new Result("2", "mock a Result");
    }

打印日志：

[cd6a3d9d05244eee95bbf3c607d038cc]方法:Result com.zhd.exploit.api.controller.TestController.test2(PayDTO)
参数:{"payDTO":{"appmount":"10","paytype":"1"}}

spring AOP中Around切面处理参数

最近遇到一个场景，在业务流程处理中，很多的方法都需要对传入的参数对象做公共的处理【比如：添加编辑人信息】，而且这些传入对象都继承自一个父类，同时需要用到HttpServletRequest。

解决的办法

使用自定义annotation+aop来实现预处理 具体的处理流程是

1、自定义一个annotation用于标记需要处理的地方

2、创建切面类，在pointcut时对annotation进行拦截，在@Around环绕通知里面获取@annotation对应的当前对象，获取当前对象参数，并修改参数内容，然后proceed一下，继续执行

具体的代码

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Operate {
}

@Aspect
@Component
public class OperateInterceptor { 
    @Pointcut("@annotation(com.yili.web.entity.Operate)")
    public void interceptor() {
    }
 
    @Resource
    private SqlObjectUtil sqlObjectUtil;
 
    @Around(value = "interceptor()")
    public Object check(ProceedingJoinPoint pjp) throws Throwable {
        System.out.println("进入interceptor");
        Signature signature = pjp.getSignature();
        if (!(signature instanceof MethodSignature)) {
            throw new IllegalArgumentException("该注解只适用于方法");
        }
        Object[] objects = pjp.getArgs();
        RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) attributes;
        HttpServletRequest request = servletRequestAttributes.getRequest();
        String loginToken = getLoginToken(request);        
        for (int i=0;i<objects.length;i++){
            if (SqlObject.class.isAssignableFrom(objects[i].getClass()))) {
                sqlObjectUtil.setOperatorInfo(loginToken,(SqlObject)objects[i]);
            }
        }
        return pjp.proceed(objects);
    }

以上为个人经验，希望能给大家一个参考，也希望大家多多支持我们。