package com.cyc.aop;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.apache.http.client.utils.URLEncodedUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.http.HttpMethod;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;
import java.io.Serializable;
import java.nio.charset.Charset;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.Map;
/**
* @Description:打印交互消息内容,方便检查问题
* @Package: com.cyc
* @Author: Chen.Yongcun
* @Version: 1.0
* <p>Copyright: Copyright (c) 2019</p>
*/
@Aspect
@Component
@Slf4j
public class ControllerAspect {
private static ThreadLocal<Long> startTime = new ThreadLocal<>();
private static ThreadLocal<String> key = new ThreadLocal<>();
private ObjectMapper objectMapper = new ObjectMapper();
/*@Bean
public ObjectMapper objectMapper() {
return new ObjectMapper().disable(SerializationFeature.FAIL_ON_EMPTY_BEANS);
}
@Autowired
private ObjectMapper objectMapper;
*/
@Pointcut("execution(public * com.cyc.controller..*(..))")
public void printParamAndResult() {
}
@Before("printParamAndResult()")
public void doBefore(JoinPoint joinPoint) {
startTime.set(System.currentTimeMillis());
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = attributes.getRequest();
// 获取请求头
Enumeration<String> enumeration = request.getHeaderNames();
StringBuffer headers = new StringBuffer();
while (enumeration.hasMoreElements()) {
String name = enumeration.nextElement();
String value = request.getHeader(name);
headers.append(name + ":" + value).append(",");
}
String uri = request.getRequestURI();
String method = request.getMethod();
StringBuffer params = new StringBuffer();
if (HttpMethod.GET.toString().equals(method)) {// get请求
String queryString = request.getQueryString();
if (StringUtils.isNotBlank(queryString)) {
params.append(URLEncodedUtils.parse(queryString, Charset.forName("utf-8")));
}
} else {
Map<String, String[]> parameterMap = request.getParameterMap();
if(parameterMap!=null && !parameterMap.isEmpty()){
for(Map.Entry<String, String[]> entry : parameterMap.entrySet()){
params.append(entry.getKey() + ":" + Arrays.toString(entry.getValue())).append(",");
}
}
/*Object[] paramsArray = joinPoint.getArgs();
if (paramsArray != null && paramsArray.length > 0) {
for (int i = 0; i < paramsArray.length; i++) {
if (paramsArray[i] instanceof Serializable) {
params.append(paramsArray[i].toString()).append(",");
} else {
//使用json系列化 反射等等方法 反系列化会影响请求性能建议重写tostring方法实现系列化接口
try {
if(paramsArray[i]==null || (paramsArray[i] instanceof String && "null".equalsIgnoreCase((String)paramsArray[i]))){
continue;
}
objectMapper.setVisibility(PropertyAccessor.FIELD, JsonAutoDetect.Visibility.ANY);
String param = objectMapper.writeValueAsString(paramsArray[i]);
if (StringUtils.isNotBlank(param))
params.append(param).append(",");
} catch (JsonProcessingException e) {
log.error("doBefore obj to json exception obj={},msg={}", paramsArray[i], e.getMessage(), e);
}
}
}
}*/
}
key.set(uri);
log.info("request params: > uri={},method={},params={},headers={}", uri, method, params, headers);
}
@AfterReturning(returning = "obj", pointcut = "printParamAndResult()")
public void doAfterReturning(Object obj) {
long costTime = System.currentTimeMillis() - startTime.get();
String uri = key.get();
startTime.remove();
key.remove();
String result = null;
if (obj instanceof Serializable) {
result = obj.toString();
} else {
if (obj != null) {
try {
result = objectMapper.writeValueAsString(obj);
} catch (JsonProcessingException e) {
log.error("doAfterReturing obj to json exception obj={},msg={}", obj, e);
}
}
}
log.info("response result:< uri={},result={},costTime={}毫秒", uri, result, costTime);
}
}
建议使用apache-common重写toString方法,避免增加字段时需要更新toString方法
toString方法
return ToStringBuilder.reflectionToString(this, ToStringStyle.SHORT_PREFIX_STYLE);
org.apache.commons.lang.builder.ReflectionToStringBuilder是commons-lang里的一个类
参考: https://blog.csdn.net/u010180738/article/details/80490704