package cn.autoform.fw;
|
|
import java.lang.reflect.Method;
|
import java.util.LinkedHashSet;
|
import java.util.Map;
|
import java.util.Set;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
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.stereotype.Component;
|
|
import cn.autoform.fw.exception.RestResult;
|
|
@Aspect
|
@Component
|
public class ControllerAspect {
|
|
/**
|
* 定义拦截规则：拦截cn.autoform.web.controller包下面的所有类中，有@RequestMapping注解的方法。
|
*/
|
@Pointcut("execution(* cn.autoform.web.controller..*(..)) and @annotation(org.springframework.web.bind.annotation.RequestMapping)")
|
public void controllerMethodPointcut() {
|
throw new UnsupportedOperationException();
|
}
|
|
/**
|
* 拦截器具体实现
|
*
|
* @param pjp
|
* @return JsonResult（被拦截方法的执行结果，或需要登录的错误提示。）
|
* @throws Throwable
|
*/
|
@Around("controllerMethodPointcut()")
|
public Object interceptor(ProceedingJoinPoint pjp) throws Throwable {
|
|
MethodSignature signature = (MethodSignature) pjp.getSignature();
|
Method method = signature.getMethod(); // 获取被拦截的方法
|
String methodName = method.getName(); // 获取被拦截的方法名
|
|
Logger logger = LoggerFactory.getLogger(pjp.getTarget().getClass());
|
|
Set<Object> allParams = new LinkedHashSet<>(); // 保存所有请求参数，用于输出到日志中
|
|
logger.info("请求开始，方法：{}", methodName);
|
|
Object[] args = pjp.getArgs();
|
logger.info("请求参数：");
|
int count = 0;
|
for (Object arg : args) {
|
count++;
|
logger.info("arg{}: {}", count, arg);
|
if (arg instanceof Map<?, ?>) {
|
// 提取方法中的MAP参数，用于记录进日志中
|
@SuppressWarnings("unchecked")
|
Map<String, Object> map = (Map<String, Object>) arg;
|
|
allParams.add(map);
|
} else if (arg instanceof HttpServletRequest) {
|
HttpServletRequest request = (HttpServletRequest) arg;
|
// 获取query string 或 posted form data参数
|
Map<String, String[]> paramMap = request.getParameterMap();
|
if (paramMap != null && paramMap.size() > 0) {
|
allParams.add(paramMap);
|
}
|
}
|
}
|
// 一切正常的情况下，继续执行被拦截的方法
|
|
Object result = pjp.proceed();
|
|
if (result instanceof RestResult) {
|
logger.info("请求结束，方法：{}", methodName);
|
}
|
|
return result;
|
}
|
}
|
