springboot添加request和response输出

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 
 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在 Spring Boot 中添加拦截器,需要完成以下步骤: 1. 创建一个拦截器类,并实现 HandlerInterceptor 接口中的三个方法:preHandle、postHandle 和 afterCompletion。 ```java @Component public class MyInterceptor implements HandlerInterceptor { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { // 在请求处理之前进行调用(Controller 方法调用之前) System.out.println("preHandle方法执行"); return true; // 如果返回 false,则中断请求 } @Override public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception { // 在请求处理之后进行调用,但是在视图被渲染之前(Controller 方法调用之后) System.out.println("postHandle方法执行"); } @Override public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { // 在整个请求结束之后被调用,也就是在 DispatcherServlet 渲染了对应的视图之后执行(主要用于资源清理工作) System.out.println("afterCompletion方法执行"); } } ``` 2. 创建一个配置类,并在其中添加拦截器。 ```java @Configuration public class MyInterceptorConfig implements WebMvcConfigurer { @Autowired private MyInterceptor myInterceptor; @Override public void addInterceptors(InterceptorRegistry registry) { // 添加拦截器 registry.addInterceptor(myInterceptor) .addPathPatterns("/**"); // 拦截所有请求 } } ``` 3. 测试拦截器是否生效。启动应用程序并发送一个请求,查看控制台输出是否包含 preHandle 方法执行。如果输出包含 preHandle 方法执行,则表示拦截器已经生效。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值