当前位置： 首页 > news >正文

Spring MVC HandlerInterceptor 拦截请求及响应体

news 2025/7/8 13:26:50

Spring MVC HandlerInterceptor 拦截请求及响应体

使用Spring MVC HandlerInterceptor拦截请求和响应体的实现方案。
通过自定义LoggingInterceptor在preHandle、postHandle和afterCompletion方法中记录请求信息，并利用RequestBodyCachingFilter缓存请求体以便多次读取。

具体步骤：

  1. 使用InterceptorRequest对象存储请求信息；
  2. 通过Filter实现请求体缓存；
  3. 使用RequestWrapper处理一次性读取的InputStream问题。

该方案适用于需要记录完整请求/响应信息的应用场景。

  • 前期想法
@Slf4j
public class LoggingInterceptor implements HandlerInterceptor {// create a InterceptorRequest object to store request info@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {// save request header, request body into InterceptorRequest// set InterceptorRequest into request attribute, eg "InterceptorRequest"log.info("preHandle");return true;}@Overridepublic void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {log.info("postHandle");}@Overridepublic void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {// get InterceptorRequest from request attribute "InterceptorRequest"// update response body, response code into InterceptorRequest// save into persistence systemif (null == ex) {log.info("afterCompletion");} else {            log.error("afterCompletion -ex - {}", ex.getMessage());}}
}
  • 将请求结果一次性塞入持久层
@Slf4j
public class LoggingInterceptor implements HandlerInterceptor {// create a InterceptorRequest object to store request info// ...// update code as below@Overridepublic void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {// get InterceptorRequest from request attribute "InterceptorRequest"// update response body, response code into InterceptorRequest// save into persistence system}
}
  • 使用 Filter 拦截获取Request Body，获取请求体
@Slf4j
public class HttpRequestContext {public static String getRequestBody(ServletRequest request) {StringBuilder builder = new StringBuilder();try (InputStream inputStream = request.getInputStream();BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8))) {String line;while ((null != (line = reader.readLine()))) {builder.append(line);}} catch (IOException e) {log.warn("Error reading request body", e);throw new RuntimeException(e);}return builder.toString();}
}
/*** It is a filter class used to cache request bodies, commonly employed in web applications,* especially when handling POST or PUT requests. Since the input stream (InputStream) of an **HTTP request*** can only be read once, it is necessary to cache the request body in scenarios where the content* of the request body needs to be accessed multiple times (such as logging, verification, filtering, etc.).*/
@Slf4j
public class RequestBodyCachingFilter implements Filter {@Overridepublic void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {String requestId = request.getAttribute("requestId")if (null == requestId) {requestId  = UUID.randomUUID().toString().replaceAll("-", "");}MDC.put("requestId", requestId);ContentCachingResponseWrapper responseWrapper = new ContentCachingResponseWrapper((HttpServletResponse) response);try {// Wrap the request to cache the request bodyServletRequest requestWrapper = new RequestWrapper((HttpServletRequest) request);chain.doFilter(requestWrapper, responseWrapper);} finally {responseWrapper.copyBodyToResponse();MDC.remove("requestId");}}@Getterstatic class RequestWrapper extends ContentCachingRequestWrapper {private final String requestBody;public RequestWrapper(HttpServletRequest request) {super(request);requestBody = HttpRequestContext.getRequestBody(request);}@Overridepublic ServletInputStream getInputStream() throws IOException {// Return a ServletInputStream that reads from the cached request bodyfinal ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(requestBody.getBytes());return new ServletInputStream() {@Overridepublic int read() throws IOException {return byteArrayInputStream.read();}@Overridepublic boolean isFinished() {return false;}@Overridepublic boolean isReady() {return false;}@Overridepublic void setReadListener(ReadListener listener) {}};}}
}

需要将自定义的Filter注册到FilterRegistrationBean中

    @Beanpublic FilterRegistrationBean<RequestBodyCachingFilter> requestBodyCachingFilter() {FilterRegistrationBean<RequestBodyCachingFilter> registration = new FilterRegistrationBean<>();registration.setFilter(new RequestBodyCachingFilter());registration.addUrlPatterns("/*");registration.setOrder(1);return registration;}
  • 统一接口封装响应实体
@Getter
@Setter
public class ApiObj<T> {private  String code;private  String message;private  T data;public ApiObj(String code, String message, T data) {this.code = code;this.data = data;this.message = message;}public static <T> ApiObj<T> success(T data) {return new ApiObj<>("200", "success", data);}public static <T> ApiObj<T> failure(String code, String message) {return new ApiObj<>(code, message, null);}
}


封装异常返回。这样返回结果可以与对外接口提供的数据一致。
@Slf4j
@Order(Ordered.HIGHEST_PRECEDENCE)
@RestControllerAdvice
public class GlobalException {static final ObjectMapper objectMapper = new ObjectMapper();@ExceptionHandler(MethodArgumentNotValidException.class)public ResponseEntity<ApiObj<String>> handleValidationExceptions(MethodArgumentNotValidException ex) throws JsonProcessingException {log.error("handleValidationExceptions - {}", ex.getMessage());Map<String, String> errors = new HashMap<>();ex.getBindingResult().getAllErrors().forEach((error) -> {String fieldName = ((FieldError) error).getField();String errorMessage = error.getDefaultMessage();errors.put(fieldName, errorMessage);});String message = objectMapper.writeValueAsString(errors);return new ResponseEntity<>(ApiObj.failure(String.valueOf(HttpStatus.BAD_REQUEST.value()), message), HttpStatus.BAD_REQUEST);}@ExceptionHandler(value = Exception.class)public ResponseEntity<ApiObj<String>> defaultErrorHandler(HttpServletRequest req, Exception e) throws Exception {log.error("defaultErrorHandler - {}", e.getMessage());Map<String, String> map = Map.of("k1", "k2");String message = objectMapper.writeValueAsString(map);return new ResponseEntity<>(ApiObj.failure(String.valueOf(HttpStatus.BAD_REQUEST.value()), message), HttpStatus.BAD_REQUEST);}@ExceptionHandler(value = IOException.class)public ResponseEntity<ApiObj<String>> defaultIOException(HttpServletRequest req, IOException e) throws Exception {log.error("defaultIOException - {}", e.getMessage());Map<String, String> map = Map.of("k3", "k4");String message = objectMapper.writeValueAsString(map);return new ResponseEntity<>(ApiObj.failure(String.valueOf(HttpStatus.BAD_REQUEST.value()), message), HttpStatus.BAD_REQUEST);}
}

= 完整更新 LoggingInterceptor

@Slf4j
public class LoggingInterceptor implements HandlerInterceptor {final static String INTERNAL_REQUEST_BODY = "INTERNAL_REQUEST_BODY";@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {// save request header, request body into InterceptorRequest// set InterceptorRequest into request attribute, eg "InterceptorRequest"log.info("preHandle");String requestBody = HttpRequestContext.getRequestBody(request);log.info("Request Body: {}", requestBody);request.setAttribute(INTERNAL_REQUEST_BODY, requestBody);return true;}@Overridepublic void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {log.info("postHandle");}@Overridepublic void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {// get InterceptorRequest from request attribute "InterceptorRequest"// update response body, response code into InterceptorRequest// save into persistence systemlog.info("afterCompletion");String requestBody = (String) request.getAttribute(INTERNAL_REQUEST_BODY);log.info("Request Body: {}", requestBody);if (response instanceof ContentCachingResponseWrapper responseWrapper) {byte[] responseBody = responseWrapper.getContentAsByteArray();String responseBodyStr = new String(responseBody, response.getCharacterEncoding());log.info("Response Body: {}", responseBodyStr);// write response body to responseresponseWrapper.copyBodyToResponse();}request.removeAttribute(INTERNAL_REQUEST_BODY);}
}

logback-spring.xml

%X{requestId:-MISSING} 输出MDC requestId

<configuration><appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender"><encoder><pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} %-5level %X{requestId:-MISSING} %logger{36} - %msg%n</pattern>        </encoder></appender><root level="info"><appender-ref ref="STDOUT" /></root>
</configuration>
http://www.lryc.cn/news/582213.html

相关文章：

  • 【Netty高级】Netty的技术内幕
  • token非对称加密
  • AI的出现，是否能替代IT从业者
  • React19 新增Hooks：useOptimistic
  • 系统学习Python——并发模型和异步编程：进程、线程和GIL
  • 量子计算+AI芯片：光子计算如何重构神经网络硬件生态
  • 动手学深度学习13.7. 单发多框检测（SSD）-笔记练习（PyTorch）
  • Android 10 Gnss数据流程
  • Java 大视界 -- Java 大数据在智能安防视频监控系统中的视频质量评估与智能修复（337）
  • uniapp的navigator跳转功能
  • STEP 7 MicroWIN SMART V2.2 的详细安装步骤及注意事项
  • 【世纪龙科技】汽车钣金虚拟仿真教学实训软件
  • 源码推送到gitee码云仓库
  • 华为OD 二维伞的雨滴效应
  • JDBC 注册驱动的常用方法详解
  • Spring Data JPA基本方法调用规律
  • RK3588 Android SDK 实战全解析 —— 架构、原理与开发关键点
  • linux qt 使用log4cpp库
  • 对象存储-OSS
  • centos停止维护后更换yum源
  • Centos Docker 安装(100%成功)
  • 腾讯云 CDN 不支持 WebSocket 的现状与华为云 CDN 的替代方案-优雅草卓伊凡
  • 【DPDK应用篇】事件驱动架构：eventdev异步处理模型的设计与实现
  • 循环移位网络设计
  • ubuntu server系统 安装宝塔
  • 【build.gradle中的各种jdk或者是jvm,sdk版本作用区别,详细说明】
  • RKAndroid11-系统设置新增开关选项
  • Kotlin流操作符简介
  • 力扣网编程274题：H指数之计数排序（中等）
  • 分布式推客系统架构设计：从微服务到高性能计算的实践路径