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

使用 HandlerMethodReturnValueHandler 在SpringBoot项目 实现 RESTful API 返回值自动封装,简化开发

参考文献 Springboot 之 HandlerMethodReturnValueHandler 运用简介 现在项目中大部分采用 - 掘金

在开发 RESTful API 时,为了确保接口返回数据的一致性,我们通常会采用统一的响应格式。例如:

{"status": "success","message": "","data": { /* 业务数据 */ }
}

然而,传统的做法往往需要在每个控制器方法中手动封装 Result.success(data),这导致了大量的重复代码。今天,我们将探讨如何通过自定义 HandlerMethodReturnValueHandler 来实现返回值的自动封装,从而简化我们的工作流程,提高开发效率。

🎯问题背景:重复的返回值封装

在没有进行统一处理的情况下,我们的 Controller 方法可能会像这样编写:

@RequestMapping(value = "/getBaseInfo")
public Result<StaffInfo> getBaseInfo(SessMerchant sessMerchant, String id) {Assert.notNull(id, "ID不能为空");StaffInfo item = staffInfoService.getById(id);if (null == item) {Assert.isTrue(false, "ID错误");}return Result.success(item); // 每个方法都要手动封装
}

这种方法不仅繁琐,而且容易遗漏,增加了出错的风险。

二、解决方案:自定义 HandlerMethodReturnValueHandler

Spring MVC 提供了 HandlerMethodReturnValueHandler 接口,允许我们在控制器方法执行后、响应写出前,对返回值进行拦截和处理。我们可以利用它来自动包装返回结果

1. 核心组件:HandlerMethodReturnValueHandlerProxy

import org.springframework.core.MethodParameter;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodReturnValueHandler;
import org.springframework.web.method.support.ModelAndViewContainer;import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.Map;public class HandlerMethodReturnValueHandlerProxy implements HandlerMethodReturnValueHandler {private final HandlerMethodReturnValueHandler proxyObject;public HandlerMethodReturnValueHandlerProxy(HandlerMethodReturnValueHandler proxyObject) {this.proxyObject = proxyObject;}@Overridepublic boolean supportsReturnType(MethodParameter returnType) {// 仅代理支持 @ResponseBody 的处理器return proxyObject.supportsReturnType(returnType);}@Overridepublic void handleReturnValue(Object returnValue, MethodParameter returnType,ModelAndViewContainer mavContainer,NativeWebRequest webRequest) throws Exception {HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);String uri = request.getRequestURI(); // 可用于日志或特殊处理Map<String, Object> resultMap = new HashMap<>();resultMap.put("status", "success");resultMap.put("message", "");resultMap.put("data", returnValue); // 将原始返回值包装进data字段// 继续交给原始处理器(如 RequestResponseBodyMethodProcessor)处理proxyObject.handleReturnValue(resultMap, returnType, mavContainer, webRequest);}private static final String STATUS_CODE_SUCCEEDED = "success";
}
🔍 解析:
  • 构造函数注入原始处理器:保存原始的 HandlerMethodReturnValueHandler(如 RequestResponseBodyMethodProcessor),用于最终序列化。
  • supportsReturnType:判断是否支持当前返回类型,直接委托给原始处理器。
  • handleReturnValue:这是核心逻辑,在这里将原始返回值 returnValue 包装成我们期望的统一结构 {status, message, data},然后再交给原始处理器进行JSON序列化。

2. 配置类:替换默认处理器链

import com.atguigu.common.utils.aspect.HandlerMethodReturnValueHandlerProxy;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.method.support.HandlerMethodReturnValueHandler;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter;
import org.springframework.web.servlet.mvc.method.annotation.RequestResponseBodyMethodProcessor;import java.util.ArrayList;
import java.util.List;@Configuration
public class RestReturnValueHandlerConfigurer implements InitializingBean {@Autowiredprivate RequestMappingHandlerAdapter handlerAdapter;@Overridepublic void afterPropertiesSet() throws Exception {List<HandlerMethodReturnValueHandler> originalHandlers = handlerAdapter.getReturnValueHandlers();List<HandlerMethodReturnValueHandler> newHandlers = new ArrayList<>();for (HandlerMethodReturnValueHandler handler : originalHandlers) {if (handler instanceof RequestResponseBodyMethodProcessor) {// 对支持 @ResponseBody 的处理器进行代理包装newHandlers.add(new HandlerMethodReturnValueHandlerProxy(handler));} else {newHandlers.add(handler);}}// 替换原有的处理器链handlerAdapter.setReturnValueHandlers(newHandlers);}
}
🔍 关键点解析
  • @Autowired RequestMappingHandlerAdapter:这是Spring MVC处理请求的核心组件之一,管理着所有 HandlerMethodReturnValueHandler
  • afterPropertiesSet:在Bean初始化完成后执行,确保 handlerAdapter 已注入。
  • 只代理 RequestResponseBodyMethodProcessor:该处理器负责处理带有 @ResponseBody 或 @RestController 的返回值并进行JSON转换,是我们关注的重点。
  • 保留其他处理器:如视图解析、ModelAndView 等不受影响。

三、效果对比:从繁琐到简洁

✅ 改造前(需手动封装)

@RequestMapping("/getBaseInfo")
public Result<StaffInfo> getBaseInfo(String id) {StaffInfo item = staffInfoService.getById(id);return Result.success(item); // 每次都要写
}

✅ 改造后(自动封装)

@RequestMapping("/getBaseInfo")
public StaffInfo getBaseInfo(String id) {Assert.notNull(id, "ID不能为空");StaffInfo item = staffInfoService.getById(id);if (item == null) throw new IllegalArgumentException("ID错误");return item; // 直接返回业务对象,框架自动包装
}

响应结果自动变为:

{"status": "success","message": "","data": {"id": "123","name": "张三"}
}

可进一步优化,仅对特定注解(如 @RestResult)的方法进行包装,避免影响所有接口。

如果你的项目正在使用Spring Boot构建REST API,强烈建议引入此类统一返回值处理机制,让代码更干净!

http://www.dtcms.com/a/349786.html

相关文章:

  • 数据结构青铜到王者第三话---ArrayList与顺序表(2)
  • 零知开源——基于STM32F103RBT6和ADXL335实现SG90舵机姿态控制系统
  • three.js+WebGL踩坑经验合集(9.1):polygonOffsetUnits工作原理大揭秘
  • 【数据结构】LeetCode160.相交链表 138.随即链表复制 牛客——链表回文问题
  • [SC]SystemC动态进程概述及案例
  • LinkedIn 自动消息发送工具
  • 网络编程——TCP、UDP
  • 人工智能(AI)与网络安全
  • 【Linux】协议的本质
  • 一键脚本:自动安装 Nginx + Certbot + HTTPS(Let‘s Encrypt)
  • QT-QSS样式表
  • 面试:计算机网络
  • 《输赢》电视剧总结学习
  • 数据结构:红黑树(Red-Black Tree)
  • 电商秒杀场景下,深挖JVM内存泄漏与多线程死锁的解决方案
  • Python3.14安装包下载与保姆级图文安装教程!!
  • PyTorch实战(1)——深度学习概述
  • 【动态规划】309. 买卖股票的最佳时机含冷冻期及动态规划模板
  • webpack文件指纹:hash、chunkhash与contenthash详解
  • 基于 OpenCV 与 Mediapipe 的二头肌弯举追踪器构建指南:从环境搭建到实时计数的完整实现
  • 【CV】图像基本操作——①图像的IO操作
  • 系统架构设计师-计算机系统存储管理-页式、段氏、段页式模拟题
  • [系统架构设计师]专业英语(二十二)
  • Python爬虫第四课:selenium自动化
  • Qwt7.0-打造更美观高效的Qt开源绘图控件库
  • macbook国内源安装rust
  • leetcode LCR 012.寻找数组的中心下标
  • 如何在 Jenkins 中安装 Master 和 Slave 节点以优化 CI/CD 流程
  • init.environ.rc详解
  • CORS解决跨域问题的多个方案 - nginx站点配置 / thinkphp框架内置中间件 / 纯前端vue、vite的server.proxy代理