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

使用 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.xdnf.cn/news/18745.html

相关文章:

  • Linux系统网络管理
  • 积分排行样式
  • 动态住宅代理:跨境电商数据抓取的稳定解决方案
  • 3785定期复盘代码实现设计模式的越识应用
  • Java接口调用第三方接口时的超时处理策略
  • 浅谈为什么尾递归更高效?——从调用栈和汇编的视角
  • 开源零信任本地化部署实战指南:Keycloak + OpenZiti 完整方案
  • 机器学习-朴素贝叶斯
  • 常用的分布式ID设计方案
  • 可信医疗大数据来源、院内数据、病种数据及编程使用方案分析
  • 【MTCNN网络结构记忆卡片】--003nets.py
  • 嵌入式第三十六天(网络编程(TCP))
  • Java的数字计算
  • More Effective C++ 条款06: 区分自增自减操作符的前缀和后缀形式
  • 若依4.7.8(springboot2.5.15)升级到4.8.1(springboot3.3.5)并集成Dubbo3客户端
  • 工程师的自我修养
  • Python JSON数据格式
  • 【数据结构】-4-顺序表(上)
  • 复杂水域场景识别率↑89%!陌讯多模态融合算法在岸边垃圾检测的落地实践
  • CUDA安装,pytorch库安装
  • 小米AX3600访问桥接的光猫
  • 图解SpringMVC工作流程,以及源码分析。
  • Hibernate详解
  • 爆肝三周,我终于上线了自己的第一个小程序
  • Vue 项目 package.json 终极详解(主流实践 / 逐项说明)
  • 大型 C/C++ 项目中 AI 助手(Cursor / Claude Code)日常操作清单与发散思路
  • 详解triton.jit及PTX
  • 微服务-19.什么是网关
  • AI重塑跨境电商:选品成功率提升53%+物流效率加快34%,多语种运营成破局关键
  • 试析微剧《云端爱人》:AI时代的数字爱情寓言与情感觉醒