springboot统一拦截包装接口返回值

1、作用

代替手动封装每个接口的返回值,否则每个有返回值的controller接口,都需要手动使用Resonse类包装返回结果

2、实现

方法返回值handler,需要实现接口HandlerMethodReturnValueHandler,并重写supportsReturnType和handleReturnValue方法。只有在supportsReturnType方法返回值为true的情况下才会执行handleReturnValue方法。

@Service
public class ResponseReturnValueHandler implements HandlerMethodReturnValueHandler {

    /**
     * 当前handler是否能够处理 MethodParameter returnType
     *
     * @param returnType
     * @return
     */
    @Override
    public boolean supportsReturnType(MethodParameter returnType) {
        ApiResponseBody methodAnnotation = returnType.getMethodAnnotation(ApiResponseBody.class);
        if (Objects.nonNull(methodAnnotation)) {
            ApiResponseBodyType type = methodAnnotation.type();
            if (ApiResponseBodyType.HTML.equals(type) || ApiResponseBodyType.XML.equals(type) || ApiResponseBodyType.IGNORE.equals(type)) {
                return false;
            }
        }
        return true;
    }

    /**
     * 处理返回结果
     *
     * @param returnValue
     * @param returnType
     * @param mavContainer
     * @param webRequest
     * @throws Exception
     */
    @Override
    public void handleReturnValue(Object returnValue, MethodParameter returnType, ModelAndViewContainer mavContainer, NativeWebRequest webRequest) throws Exception {
        // 使用Result包装返回值returnValue
        Result jsonResult = Result.success(returnValue);
        HttpServletResponse response = webRequest.getNativeResponse(HttpServletResponse.class);
        PrintWriter writer = response.getWriter();
        writer.write(JsonHelper.parseToJson(jsonResult));
        writer.flush();
        writer.close();
    }
}

包装器wrapper(把处理器handler加入到RequestMappingHandlerAdapter的list中)

@Configuration
public class RequestMappingHandlerAdapterWrapper {
    /**
     * 解析参数和返回结果
     */
    @Autowired
    private RequestMappingHandlerAdapter requestMappingHandlerAdapter;

    /**
     * 自定义的返回值处理器
     */
    @Autowired
    private ResponseReturnValueHandler responseReturnInterceptor;

    @PostConstruct
    public void init() {
        List<HandlerMethodReturnValueHandler> originalHandlers = new ArrayList<>();
        originalHandlers.add(responseReturnInterceptor);
        requestMappingHandlerAdapter.setReturnValueHandlers(originalHandlers);
    }
}

方法返回值类型注解(此注解使用在controller的方法上,用来标识接口返回的数据类型,可以有json、html或者ingore),只要不是ignore、html或xml,就当做json,在方法handleReturnValue中使用Result来统一包装。

@Getter
public enum ApiResponseBodyType {
    /**
     * JSON序列化格式
     */
    JSON,
    HTML,
    IGNORE,
    XML
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值