如何记录来自@RestController 的任何传出响应？

Question

我正在使用CommonsRequestLoggingFilter 在我的@RestController 上记录任何传入请求。

@RestController
public class MyController {
    @PostMapping("/")
    public MyRsp ping() {
        ...
        return myRsp;
    }
}

用户将发送POST JSON 请求，这些请求已经使用：

@Bean
public CommonsRequestLoggingFilter requestLoggingFilter() {
    CommonsRequestLoggingFilter filter = new CommonsRequestLoggingFilter();
    filter.setIncludeClientInfo(true);
    filter.setIncludeQueryString(true);
    filter.setIncludePayload(true);
    return filter;
}

问题：我怎样才能为我发回给用户的 JSON 响应实现相同的效果？

Answer 1

我不知道为什么 spring 提供了一个使用ContentCachingRequestWrapper 的休息记录器，但不提供响应记录。因为它可以实现和req日志非常相似，如下：

public class MyFilter extends CommonsRequestLoggingFilter {
    @Override
    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) {
        //same mechanism as for request caching in superclass
        HttpServletResponse responseToUse = response;
        if (isIncludePayload() && !isAsyncDispatch(request) && !(response instanceof ContentCachingResponseWrapper)) {
            responseToUse = new ContentCachingResponseWrapper(response);
        }

        //outgoing request is logged in superclass
        super.doFilterInternal(request, responseToUse, filterChain);

        //log incoming response
        String rsp = getResponseMessage(responseToUse);
        LOGGER.info(rsp);
    }   

    //equivalent to super.createMessage() for request logging
    private String getResponseMessage(HttpServletResponse rsp) {
        StringBuilder msg = new StringBuilder();

        ContentCachingResponseWrapper wrapper =
                    WebUtils.getNativeResponse(request, ContentCachingResponseWrapper.class);

        if (wrapper != null) {
                byte[] buf = wrapper.getContentAsByteArray();
                if (buf.length > 0) {
                    int length = Math.min(buf.length, getMaxPayloadLength());
                    String payload;
                    try {
                        payload = new String(buf, 0, length, wrapper.getCharacterEncoding());
                    }
                    catch (UnsupportedEncodingException ex) {
                        payload = "[unknown]";
                    }
                    msg.append(";payload=").append(payload);
                }
         }
    } 
}

Answer 2

@Pointcut("@annotation(org.springframework.web.bind.annotation.RequestMapping)")
public void requestMapping() {}

@Pointcut("within(path.to your.controller.package.*)")
public void myController() {}

@Around("requestMapping() || myController()")
public MyRsp logAround(ProceedingJoinPoint joinPoint) throws Throwable {
    joinPoint.getArgs()// This will give you arguments if any being passed to your controller.
   ...............
   MyRsp myRsp = (MyRsp) joinPoint.proceed();
   ...............
   return myRsp;
}