接口数据统一处理返回

根据@ControllerAdvice注解实现ResponseBodyAdvice接口来完成此功能

package com.vshare.decision.config;

import com.vshare.decision.api.Result;
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;

import java.net.URI;

/**
 * @Author djw
 * @Description 统一接口返回处理
 * @Date 2019/12/4 14:23
 */
@ControllerAdvice
public class ResponseBody implements ResponseBodyAdvice<Object> {
    @Override
    public boolean supports(MethodParameter methodParameter, Class aClass) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object o,
                                  MethodParameter methodParameter,
                                  MediaType mediaType,
                                  Class aClass,
                                  ServerHttpRequest serverHttpRequest,
                                  ServerHttpResponse serverHttpResponse) {

        URI uri = serverHttpRequest.getURI();
        String path = uri.toString();
        //有swagger接口文档时
        if (!path.contains("swagger") && !path.contains("v2")) {
            Result result = new Result();
            if (o == null) {
                return result;
            //当接口返回的数据已经被处理过就不需要在处理了，直接返回
            } else if (o instanceof Result) {
                result = (Result) o;
            } else {
                result.setResult(o);
            }
            // 返回接口数据
            return result;
        }
        return o;
    }
}