装饰SpringMVC的适配器实现响应自动包装

embedded/2025/1/31 21:37:34/

文章目录

    • 1.common-tool-starter
        • 1.目录结构
        • 2.ResultWrapper.java
    • 2.common-web-starter
        • 1.目录结构
        • 2.IgnoredResultWrapper.java 自定义注解,忽略对返回结果的自动包装
        • 3.ReturnValueHandlersDecorator.java适配器进行扩展的装饰器
        • 4.WebAutoConfiguration.java 将装饰器注入容器
    • 3.common-web-starter-demo
        • 1.WebController.java 测试三种使用方式
        • 2.测试
          • 1.第一种:直接使用自动包装成功结果
          • 2.第二种:使用 @IgnoredResultWrapper注解忽略掉自动包装
          • 3.第三种:直接使用ResultWrapper来自己封装结果

1.common-tool-starter

1.目录结构

CleanShot 2025-01-06 at 18.36.32@2x

java_8">2.ResultWrapper.java
java">package com.sunxiansheng.tool.response;import lombok.Data;import java.io.Serializable;/*** Description: 通用响应封装类,通过枚举来获取code和message** @Author sun* @Create 2024/10/11* @Version 1.0*/
@Data
public class ResultWrapper<T> implements Serializable {private static final long serialVersionUID = 1L;// 是否成功private boolean success;// 响应代码private int code;// 响应消息private String message;// 响应数据private T data;// 私有构造器,防止外部直接创建private ResultWrapper() {}// 使用内部建造者类进行对象构建public static class Builder<T> {private boolean success;private int code;private String message;private T data;// ============================== 链式调用设置建造者对象 ==============================public Builder<T> success(boolean success) {this.success = success;return this;}public Builder<T> code(int code) {this.code = code;return this;}public Builder<T> message(String message) {this.message = message;return this;}public Builder<T> data(T data) {this.data = data;return this;}// ============================== 链式调用设置建造者对象 ==============================// ============================== 构建ResultWrapper对象 ==============================public ResultWrapper<T> build() {ResultWrapper<T> result = new ResultWrapper<>();result.success = this.success;result.code = this.code;result.message = this.message;result.data = this.data;return result;}// ============================== 构建ResultWrapper对象 ==============================}// ============================== 快捷方法 ==============================public static <T> ResultWrapper<T> ok() {return new Builder<T>().success(true).code(RespBeanEnum.SUCCESS.getCode()).message(RespBeanEnum.SUCCESS.getMessage()).build();}public static <T> ResultWrapper<T> ok(T data) {return new Builder<T>().success(true).code(RespBeanEnum.SUCCESS.getCode()).message(RespBeanEnum.SUCCESS.getMessage()).data(data).build();}public static <T> ResultWrapper<T> ok(T data, String message) {return new Builder<T>().success(true).code(RespBeanEnum.SUCCESS.getCode()).message(message).data(data).build();}public static <T> ResultWrapper<T> fail() {return new Builder<T>().success(false).code(RespBeanEnum.ERROR.getCode()).message(RespBeanEnum.ERROR.getMessage()).build();}public static <T> ResultWrapper<T> fail(String message) {return new Builder<T>().success(false).code(RespBeanEnum.ERROR.getCode()).message(message).build();}public static <T> ResultWrapper<T> fail(int code, String message) {return new Builder<T>().success(false).code(code).message(message).build();}public static <T> ResultWrapper<T> fail(int code, String message, T data) {return new Builder<T>().success(false).code(code).message(message).data(data).build();}public static <T> ResultWrapper<T> fail(RespBeanEnum respBeanEnum) {return new Builder<T>().success(false).code(respBeanEnum.getCode()).message(respBeanEnum.getMessage()).build();}// ============================== 快捷方法 ==============================
}

2.common-web-starter

1.目录结构

CleanShot 2025-01-06 at 18.37.27@2x

java__167">2.IgnoredResultWrapper.java 自定义注解,忽略对返回结果的自动包装
java">package com.sunxiansheng.web.annotation;import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;/*** Description: 忽略对返回结果的自动包装** @Author sun* @Create 2025/1/6 15:58* @Version 1.0*/
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface IgnoredResultWrapper {}
java__191">3.ReturnValueHandlersDecorator.java适配器进行扩展的装饰器
java">package com.sunxiansheng.web.config;import com.sunxiansheng.tool.response.ResultWrapper;
import com.sunxiansheng.web.annotation.IgnoredResultWrapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.core.MethodParameter;
import org.springframework.core.annotation.AnnotatedElementUtils;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodReturnValueHandler;
import org.springframework.web.method.support.ModelAndViewContainer;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter;
import org.springframework.web.servlet.mvc.method.annotation.RequestResponseBodyMethodProcessor;import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;/*** Description: 装饰器** @Author sun* @Create 2025/1/6 14:40* @Version 1.0*/
@Slf4j
public class ReturnValueHandlersDecorator implements InitializingBean {/*** 注入适配器,可以获取*/@Resourceprivate RequestMappingHandlerAdapter requestMappingHandlerAdapter;@Overridepublic void afterPropertiesSet() {List<HandlerMethodReturnValueHandler> returnValueHandlers = requestMappingHandlerAdapter.getReturnValueHandlers();// 因为默认List<HandlerMethodReturnValueHandler>是不可修改列表,所以需要给他转换成一个可修改的列表assert returnValueHandlers != null;List<HandlerMethodReturnValueHandler> handlers = new ArrayList<>(returnValueHandlers);// 装饰一下然后重新设置this.decorateHandlers(handlers);// 将装饰后的handlers替换原来的不可变列表requestMappingHandlerAdapter.setReturnValueHandlers(handlers);}private void decorateHandlers(List<HandlerMethodReturnValueHandler> returnValueHandlers) {for (HandlerMethodReturnValueHandler returnValueHandler : returnValueHandlers) {if (returnValueHandler instanceof RequestResponseBodyMethodProcessor) {// 找到RequestResponseBodyMethodProcessorControllerReturnValueHandler controllerReturnValueHandler = new ControllerReturnValueHandler(returnValueHandler);// 找到在原列表中的索引位置int index = returnValueHandlers.indexOf(returnValueHandler);// 将装饰后的HandlerMethodReturnValueHandler放到原来的位置returnValueHandlers.set(index, controllerReturnValueHandler);}}}private static class ControllerReturnValueHandler implements HandlerMethodReturnValueHandler {// 拿到被装饰的对象private final HandlerMethodReturnValueHandler handler;public ControllerReturnValueHandler(HandlerMethodReturnValueHandler handler) {this.handler = handler;}/*** 保持跟原生的一致** @param returnType* @return*/@Overridepublic boolean supportsReturnType(MethodParameter returnType) {return AnnotatedElementUtils.hasAnnotation(returnType.getContainingClass(), ResponseBody.class) || returnType.hasMethodAnnotation(ResponseBody.class);}@Overridepublic void handleReturnValue(Object o, MethodParameter methodParameter, ModelAndViewContainer modelAndViewContainer, NativeWebRequest nativeWebRequest) throws Exception {IgnoredResultWrapper methodAnnotation = methodParameter.getMethodAnnotation(IgnoredResultWrapper.class);// 如果有IgnoredResultWrapper注解,就不进行包装,走原来的逻辑if (Objects.nonNull(methodAnnotation)) {handler.handleReturnValue(o, methodParameter, modelAndViewContainer, nativeWebRequest);return;}// 如果返回值是ResultWrapper类型,也不进行包装,走原来的逻辑if (o instanceof ResultWrapper) {handler.handleReturnValue(o, methodParameter, modelAndViewContainer, nativeWebRequest);return;}// 其余情况,进行包装log.info("Controller返回值已被自动包装,如果上传文件,请加@IgnoredResultWrapper注解取消自动包装!");ResultWrapper<Object> ok = ResultWrapper.ok(o);handler.handleReturnValue(ok, methodParameter, modelAndViewContainer, nativeWebRequest);}}
}
java__297">4.WebAutoConfiguration.java 将装饰器注入容器
java">    /*** 注入对返回结果增强的装饰器** @return*/@Bean@ConditionalOnMissingBeanpublic ReturnValueHandlersDecorator returnValueHandlersDecorator() {return new ReturnValueHandlersDecorator();}

3.common-web-starter-demo

java__314">1.WebController.java 测试三种使用方式
java">package com.sunxiansheng.web.controller;import com.sunxiansheng.tool.response.ResultWrapper;
import com.sunxiansheng.web.annotation.IgnoredResultWrapper;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;/*** Description: 测试** @Author sun* @Create 2024/10/4 22:56* @Version 1.0*/
@RestController
public class WebController {/*** 第一种方式:直接使用自动包装成功结果** @return*/@RequestMapping("/method1")public String method1() {return "method1";}/*** 第二种方式:使用 @IgnoredResultWrapper注解忽略掉自动包装** @return*/@IgnoredResultWrapper@RequestMapping("/method2")public String method2() {return "method2";}/*** 第三种方式:直接使用ResultWrapper来自己封装结果** @return*/@RequestMapping("/method3")public ResultWrapper<String> method3() {return ResultWrapper.fail("method3");}
}
2.测试
1.第一种:直接使用自动包装成功结果

CleanShot 2025-01-06 at 18.40.09@2x

2.第二种:使用 @IgnoredResultWrapper注解忽略掉自动包装

CleanShot 2025-01-06 at 18.40.48@2x

3.第三种:直接使用ResultWrapper来自己封装结果

CleanShot 2025-01-06 at 18.41.18@2x


http://www.ppmy.cn/embedded/158454.html

相关文章

参照和谐色调为PPT图形设置统一格式的要点

大家好&#xff5e;今天和大家聊一聊PPT设计中如何通过和谐色调为图形设置统一格式&#xff0c;让整体设计看起来既专业又有视觉吸引力。PPT不仅仅是一个展示工具&#xff0c;它更是传达信息的载体。无论是公司汇报、学术展示&#xff0c;还是产品推广&#xff0c;一个有条理、…

React 低代码项目:项目创建

Date: January 29, 2025 项目创建 思路&#xff1a; 使用 Create-React-App 创建 React 项目使用 Vite 创建 React 项目使用 eslint prettier husty 等&#xff0c;制定编码规则 创建项目 注&#xff1a;在这之前&#xff0c;推荐 node 版本&#xff1a;node/18.20.6 &#…

股指期货的基差套利有什么样的风险?

基差说的就是期货价格减去现货价格的那个差。如果期货价格比现货价格高&#xff0c;咱们就叫它升水&#xff1b;反过来&#xff0c;期货价格比现货价格低&#xff0c;那就是贴水。股指期货和股票现货&#xff0c;它们之间有个默契&#xff0c;那就是到了期货交割日&#xff0c;…

消息队列:春招面试的重要知识模块

在之前的文章中&#xff0c;我们深入探讨了 Redis 缓存&#xff0c;了解了它的数据结构、应用场景以及缓存淘汰策略&#xff0c;这对于提升系统性能起到了关键作用。而在现代分布式系统中&#xff0c;消息队列同样是不可或缺的组件。消息队列能够实现系统间的异步通信、解耦以及…

视频外绘技术总结:Be-Your-Outpainter、Follow-Your-Canvas、M3DDM

Diffusion Models专栏文章汇总:入门与实战 前言:视频Inpaint的技术很火,但是OutPaint却热度不高,这篇博客总结比较经典的几篇视频Outpaint技术。其实Outpaint在runway等工具上很火,可是学术界对此关注比较少,博主从这三年的顶会中找到了最具代表性的三篇论文解读。 目录 …

【C++】类和对象

面向对象编程 学习过C语言的小伙伴知道&#xff1a;C语言是面向过程的&#xff0c;关注的是过程&#xff0c;分析出求解问题的步骤&#xff0c;通过函数调用逐步解决问题。 面向过程编程也叫结构化编程。虽然结构化编程的理念提高了程序的清晰度&#xff0c;可靠性&#xff0c…

灰色预测模型

特点&#xff1a; 利用少量、不完全的信息 预测的是指数型的数值 预测的是比较近的数据 灰色生成数列原理&#xff1a; 累加生成&#xff1a; 累减生成&#xff1a;通过累减生成还原成原始数列。 加权相邻生成&#xff1a;&#xff08;会更接近每月中旬&#xff0c;更推荐…

粒子群算法 笔记 数学建模

引入: 如何找到全局最大值&#xff1a;如果只是贪心的话&#xff0c;容易被局部最大解锁定 方法有&#xff1a;盲目搜索&#xff0c;启发式搜索 盲目搜索&#xff1a;枚举法和蒙特卡洛模拟&#xff0c;但是样例太多花费巨量时间 所以启发式算法就来了&#xff0c;通过经验和规…