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

S-X-S2025-01-29 1:07

文章目录

1.common-tool-starter

1.目录结构
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.目录结构
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 {

}
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 {

    /**
     * 注入适配器，可以获取
     */
    @Resource
    private RequestMappingHandlerAdapter requestMappingHandlerAdapter;

    @Override
    public 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) {
                // 找到RequestResponseBodyMethodProcessor
                ControllerReturnValueHandler 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
         */
        @Override
        public boolean supportsReturnType(MethodParameter returnType) {
            return AnnotatedElementUtils.hasAnnotation(returnType.getContainingClass(), ResponseBody.class) || returnType.hasMethodAnnotation(ResponseBody.class);
        }

        @Override
        public 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);
        }
    }
}
4.WebAutoConfiguration.java 将装饰器注入容器
java 复制代码 
    /**
     * 注入对返回结果增强的装饰器
     *
     * @return
     */
    @Bean
    @ConditionalOnMissingBean
    public ReturnValueHandlersDecorator returnValueHandlersDecorator() {
        return new ReturnValueHandlersDecorator();
    }

3.common-web-starter-demo

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.第一种：直接使用自动包装成功结果
2.第二种：使用 @IgnoredResultWrapper注解忽略掉自动包装
3.第三种：直接使用ResultWrapper来自己封装结果
相关推荐
一只韩非子7 小时前
Spring AI Alibaba 快速上手教程：10 分钟接入大模型
java·后端·ai编程
叫我阿柒啊7 小时前
从Java全栈到云原生：一场技术深度对话
java·spring boot·docker·微服务·typescript·消息队列·vue3
ONLYOFFICE7 小时前
【技术教程】如何将文档编辑器集成至基于Java的Web应用程序
java·编辑器·onlyoffice
lbwxxc8 小时前
手写 Tomcat
java·tomcat
CHEN5_028 小时前
【CouponHub项目开发】使用RocketMQ5.x实现延时修改优惠券状态，并通过使用模板方法模式重构消息队列发送功能
java·重构·模板方法模式·项目
杨杨杨大侠8 小时前
实战案例：商品详情页数据聚合服务的技术实现
java·spring·github
杨杨杨大侠8 小时前
实战案例：保险理赔线上审核系统的技术实现
java·spring·github
计算机毕设定制辅导-无忧学长8 小时前
MQTT 与 Java 框架集成：Spring Boot 实战（一）
java·网络·spring boot
叫我阿柒啊8 小时前
从Java全栈到Vue3实战：一次真实面试的深度复盘
java·spring boot·微服务·vue3·响应式编程·前后端分离·restful api
快乐非自愿8 小时前
掌握设计模式--模板方法模式
java·设计模式·模板方法模式
热门推荐
012025 年高教社杯全国大学生数学建模竞赛C 题 NIPT 的时点选择与胎儿的异常判定 完整成品思路模型代码分享，全网首发高质量！！！022025年数学建模国赛C题超详细解题思路032025全国大学生数学建模C题保姆级思路模型（持续更新）：NIPT 的时点选择与胎儿的异常判定042025高教社杯国赛数学建模选题建议+初步分析05UV安装并设置国内源062025国赛B题保姆级教程思路分析 碳化硅外延层厚度的确定07奈飞工厂官网，国内Netflix影视在线看|中文网页电脑版入口08KGG转MP3工具|非KGM文件|解密音频09（E题｜AI 辅助智能体测）2025年高教杯全国大学生数学建模国赛解题思路|完整代码论文集合10A股预测还能更准？开源大模型Kronos带你跑通预测+回测全流程