SpringBoot使用ResponseBodyAdvice進行統一響應處理

適用場景 , 返回給調用方一個統一的響應對象 , 即Controller中使用了@ResponseBody註解的方法 , 可以隨意返回Object , String , List 等 , 在該對象中進行統一處理並返回

package com.xbz.common.web.exception;

import com.alibaba.fastjson.JSONObject;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

/**
 * @title 全局統一響應處理
 * @author Xingbz
 * @createDate 2019-8-2
 */
@ControllerAdvice
public class GlobalResponseBodyAdvice implements ResponseBodyAdvice {


    /** 此處如果返回false , 則不執行當前Advice的業務 */
    @Override
    public boolean supports(MethodParameter returnType, Class converterType) {
        return true;
    }

    /** 處理響應的具體業務方法 */
    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
        if(body instanceof JSONModel){
            return body;
        }else{
            if (body instanceof JSONObject){
                return body;
            }
            JSONModel jsonModel = new JSONModel();
            jsonModel.setCode(JSONModel.CODE_SUCCESS);
            jsonModel.setMessage(JSONModel.MESSAGE_SUCCESS);
            jsonModel.setData(body);
            return jsonModel;
        }
    }
}