Spring OAuth2 自定义返回数据

Spring OAuth2 自定义返回数据

在使用spring oauth2的时候，返回的数据进行自定义处理，以下是使用ResponseBodyAdvice进行实现的，此类可以统一处理Controller返回ResponseBody的结果，View页面不受影响。

实现ResponseBodyAdvice接口

package com.daoles.authorizationserver.config;

import lombok.extern.slf4j.Slf4j;
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 com.krishantha.rentcloud.authorizationserver.common.JsonResult;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

@Slf4j
@ControllerAdvice
public class MyResponseBodyAdvice implements ResponseBodyAdvice {
    @Override
    public boolean supports(MethodParameter methodParameter, Class aClass) {
        //此处返回true,表示对任何handler的responsebody都调用beforeBodyWrite方法，如果有特殊方法不使用可以考虑使用注解等方式过滤
        return true;
    }

    /**
     * 对Controller的所有返回结果进行处理
     * @param body 是controller方法中返回的值，对其进行修改后再return
     * @param methodParameter
     * @param mediaType
     * @param aClass
     * @param serverHttpRequest
     * @param serverHttpResponse
     * @return
     */
    @Override
    public Object beforeBodyWrite(Object body, MethodParameter methodParameter, MediaType mediaType, Class aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
        log.info("请求返回数据类型class="+ body.getClass().getName());
        if (body.toString().contains("error") && body.toString().contains("Unauthorized")){
            return new JsonResult<Object>(401,body);
        }else if (body.toString().contains("error")){
            return new JsonResult<Object>(10001,body);
        }
        if (log.isDebugEnabled()) {
            log.debug("请求返回数据body=     " + body.toString());
        }
        return new JsonResult<Object>(body);
    }
}

简单的JsonResult封装

package com.daoles.authorizationserver.common;

import lombok.Data;
import java.io.Serializable;

@Data
public class JsonResult<T> implements Serializable {
    private Integer code;
    private String msg;
    private T data;

    public JsonResult() {
    }

    public JsonResult( String msg, T data) {
        this.code = 200;
        this.msg = msg;
        this.data = data;
    }

    public JsonResult(T data) {
        this.code = 200;
        this.msg = "success";
        this.data = data;
    }
    public JsonResult(Integer code, T data) {
        this.code = code;
        this.msg = "error";
        this.data = data;
    }
    public JsonResult(Integer code, String msg, T data) {
        this.code = code;
        this.msg = msg;
        this.data = data;
    }
}

使用以上JsonResult类进行数据统一返回，这样前端对接过程中可以统一处理比较好些。

logging:
  level:
    org:
      hibernate:
        type: trace
    com:
      daoles:
       authorizationserver:
          config: info  --此处的日志级别如果是debug会打印body内容

if (log.isDebugEnabled()) {
    log.debug("请求返回数据body=     " + body.toString());
}