spring 统一处理包装返回值为Result, 并打印

@Slf4j
@ControllerAdvice
public class JsonResponseBodyAdvice implements ResponseBodyAdvice<Object> {

    @Override
    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(
            Object body, 
            MethodParameter returnType, 
            MediaType selectedContentType, 
            Class<? extends HttpMessageConverter<?>> selectedConverterType, 
            ServerHttpRequest request, 
            ServerHttpResponse response) {
        
        // 处理特殊路径,不需要做包装
        String url = request.getURI().getPath();
        // 如果匹配到了完全不包装的url,直接返回
        if (UrlUtil.check(url).anyMatchExclude()) {
            return body;
        }
        
        Object result;
        if (body instanceof Result) {
            result = body;
        } else if (body instanceof String) {
            // 因为StringHttpMessageConverter会直接把字符串写入body, 所以字符串特殊处理
            result = JsonUtil.toJson(Result.success(body));
        } else {
            // 其他类型进行统一包装
            result = Result.success(body);
        }

        log.info("response = {}", JsonUtil.toJson(result));
        return result;
    }
}
发布了44 篇原创文章 · 获赞 13 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_19457117/article/details/88350758