聊聊Spring里那个不打扰Controller就能统一改响应的“神器”

2025-12-12 0 331

01 引言

在程序编码的时候,我们总会统一返回响应参数,以规范开发。但是面对老项目改造的时候,有没有遇到过返回的参数不统一的情况,自己又不原因一个个的纠正。那么这篇文章将从不同的角度给你带来思路。

02 ResponseBodyAdvice

类似一个切面,在响应参数返回之前执行。按照官方文档的释义就是在使用HttpMessageConverter之前,执行@ResponseBody或者ResponseEntity之后被触发。

可以直接向RequestMappingHandlerAdapterExceptionHandlerExceptionResolver注册,也可以直接通过@ControllerAdvice被触发。

2.1 场景假设

假设我们系统中有以下三种返回方式:

@GetMapping(\"/testResponseEntity\")
public ResponseEntity testResponseEntity() {
    return ResponseEntity.ok(new Wjson(\"test_ResponseEntity\", \"v1\"));
}

@GetMapping(\"/testWjson\")
public Wjson testWjson() {
    return new Wjson(\"test_wjson\", \"v2\");
}


@GetMapping(\"/testJsonResult\")
public JsonResult testJsonResult() {
    return JsonResult.success(new Wjson(\"test_json\", \"v3\"));
}

现在来了一个技术Leader,想要重构现有的结构,需要统一返回参数,决定最终的返回结果如下:

{
  \"code\": \"0000\",
  \"data\": {
    \"content\": \"test_json\",
    \"version\": \"v3\"
  },
  \"success\": true
}

并给出了统一返回参数的实体:

@Data
public class JsonResult {

    private String code;
    private String msg;
    private T data;
    private boolean success;

    public static  JsonResult success(T data) {
        JsonResult jr = new JsonResult();
        jr.setData(data);
        jr.setSuccess(true);
        jr.setCode(\"0000\");
        return jr;
    }

    public static  JsonResult fail(String code, String msg) {
        JsonResult jr = new JsonResult();
        jr.setSuccess(false);
        jr.setCode(code);
        jr.setMsg(msg);
        return jr;
    }
}

2.2 问题解决

强迫症的解决方案,就是一个个的修改每一个方法。这种方案自然可行,但是当方法很多时,工作量就非常大,显然不太适合。

于是,就想从框架层面解决统一返回的问题。ResponseBodyAdvice该上场了。我们需要定于属于自己的请求体切面。

@ControllerAdvice
public class MyResponseBodyAdvice implements ResponseBodyAdvice {
    @Override
    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter> converterType) {
        return returnType.getContainingClass().isAnnotationPresent(RestController.class);
    }

    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType,
                                  Class<? extends HttpMessageConverter> selectedConverterType,
                                  ServerHttpRequest request, ServerHttpResponse response) {
        if (body instanceof JsonResult) {
            return body;
        }

        if (body instanceof Wjson) {
            return JsonResult.success(body);
        }
        if (body instanceof ResponseEntity) {
            return JsonResult.success(((ResponseEntity) body).getBody());
        }

        return JsonResult.fail(\"30002\", \"请求失败\");
    }
}

supports:什么样的条件进入,允许处理,类似开关。案例中,返回的方法上是否包含RestController.class

beforeBodyWritesupports返回true之后进入。案例中,如果返回JsonResult不处理。如果是Wjson或者ResponseEntity,需要转化成JsonResult,否则返回错误信息。

2.3 结果演示

这样就不在修改Controller的情况下,完成响应参数的统一封装。

03 番外

原本上面的解决方案已经是常见的解决方案了。但是前两天在整理SpringBoot配置项时,发现HandlerMethodReturnValueHandler同样可以解决此类问题,功能甚至更丰富。

3.1 自定统一返回处理

public class JsonResultHandlerMethodReturnValueHandler implements HandlerMethodReturnValueHandler {

    @Override
    public boolean supportsReturnType(MethodParameter returnType) {
        return returnType.getContainingClass().isAnnotationPresent(RestController.class);
    }

    @Override
    public void handleReturnValue(Object returnValue, MethodParameter returnType, ModelAndViewContainer mavContainer, NativeWebRequest webRequest) throws Exception {
        // 标记请求已处理,避免视图解析
        mavContainer.setRequestHandled(true);
        HttpServletResponse response = webRequest.getNativeResponse(HttpServletResponse.class);
        response.setContentType(MediaType.APPLICATION_JSON_VALUE);
        response.setCharacterEncoding(\"UTF-8\");
        PrintWriter writer = response.getWriter();
        
        if (returnValue instanceof JsonResult) {
            writer.write(JSON.toJSONString(returnValue));
            return;
        }

        if (returnValue instanceof Wjson) {
            writer.write(JSON.toJSONString(JsonResult.success(returnValue)));
            return;
        }

        if (returnValue instanceof ResponseEntity) {
            writer.write(JSON.toJSONString(JsonResult.success(((ResponseEntity) returnValue).getBody())));
            return;
        }
        
        writer.write(JSON.toJSONString(JsonResult.fail(\"30001\", \"请求失败\")));
    }
}

3.2 配置

为防止自定义的方法返回解析器被系统默认的解析器有限取代执行,我们需要调整解析器的位置。

@Autowired
private RequestMappingHandlerAdapter handlerAdapter;

@PostConstruct
public void init() {
    List returnValueHandlers = handlerAdapter.getReturnValueHandlers();
    List list = new ArrayList(returnValueHandlers);
    list.add(0, new JsonResultHandlerMethodReturnValueHandler());
    handlerAdapter.setReturnValueHandlers(list);
}

结果这里就不演示了,和上面的结果一致。

04 小结

条条道路通罗马,每一种方式都能解决问题。但是不同方案处理数据的时机不同,HandlerMethodReturnValueHandler处理的方式更加灵活,可以随意定制。而ResponseBodyAdvice更像一个定制好的半成品,使用起来更加简单。

收藏 (0) 打赏

感谢您的支持,我会继续努力的!

打开微信/支付宝扫一扫,即可进行扫码打赏哦,分享从这里开始,精彩与您同在
点赞 (0)

申明:本文由第三方发布,内容仅代表作者观点,与本网站无关。对本文以及其中全部或者部分内容的真实性、完整性、及时性本站不作任何保证或承诺,请读者仅作参考,并请自行核实相关内容。本网发布或转载文章出于传递更多信息之目的,并不意味着赞同其观点或证实其描述,也不代表本网对其真实性负责。

左子网 编程相关 聊聊Spring里那个不打扰Controller就能统一改响应的“神器” https://www.zuozi.net/35825.html

常见问题
  • 1、自动:拍下后,点击(下载)链接即可下载;2、手动:拍下后,联系卖家发放即可或者联系官方找开发者发货。
查看详情
  • 1、源码默认交易周期:手动发货商品为1-3天,并且用户付款金额将会进入平台担保直到交易完成或者3-7天即可发放,如遇纠纷无限期延长收款金额直至纠纷解决或者退款!;
查看详情
  • 1、描述:源码描述(含标题)与实际源码不一致的(例:货不对板); 2、演示:有演示站时,与实际源码小于95%一致的(但描述中有”不保证完全一样、有变化的可能性”类似显著声明的除外); 3、发货:不发货可无理由退款; 4、安装:免费提供安装服务的源码但卖家不履行的; 5、收费:价格虚标,额外收取其他费用的(但描述中有显著声明或双方交易前有商定的除外); 6、其他:如质量方面的硬性常规问题BUG等。 注:经核实符合上述任一,均支持退款,但卖家予以积极解决问题则除外。
查看详情
  • 1、左子会对双方交易的过程及交易商品的快照进行永久存档,以确保交易的真实、有效、安全! 2、左子无法对如“永久包更新”、“永久技术支持”等类似交易之后的商家承诺做担保,请买家自行鉴别; 3、在源码同时有网站演示与图片演示,且站演与图演不一致时,默认按图演作为纠纷评判依据(特别声明或有商定除外); 4、在没有”无任何正当退款依据”的前提下,商品写有”一旦售出,概不支持退款”等类似的声明,视为无效声明; 5、在未拍下前,双方在QQ上所商定的交易内容,亦可成为纠纷评判依据(商定与描述冲突时,商定为准); 6、因聊天记录可作为纠纷评判依据,故双方联系时,只与对方在左子上所留的QQ、手机号沟通,以防对方不承认自我承诺。 7、虽然交易产生纠纷的几率很小,但一定要保留如聊天记录、手机短信等这样的重要信息,以防产生纠纷时便于左子介入快速处理。
查看详情

相关文章

猜你喜欢
发表评论
暂无评论
官方客服团队

为您解决烦忧 - 24小时在线 专业服务