1. 程式人生 > >只需一步,在Spring Boot中統一Restful API返回值格式與統一處理異常

只需一步,在Spring Boot中統一Restful API返回值格式與統一處理異常

統一返回值

在前後端分離大行其道的今天,有一個統一的返回值格式不僅能使我們的介面看起來更漂亮,而且還可以使前端可以統一處理很多東西,避免很多問題的產生。

比較通用的返回值格式如下:

public class Result<T> {
    // 介面呼叫成功或者失敗
    private Integer code = 0;
    // 失敗的具體code
    private String errorCode = "";
    // 需要傳遞的資訊,例如錯誤資訊
    private String msg;
    // 需要傳遞的資料
    private T data;
    ...
}

最原始的介面如下:

	@GetMapping("/test")
    public User test() {
        return new User();
    }

當我們需要統一返回值時,可能會使用這樣一個辦法:

	@GetMapping("/test")
    public Result test() {
        return Result.success(new User());
    }

這個方法確實達到了統一介面返回值的目的,但是卻有幾個新問題誕生了:

  • 介面返回值不明顯,不能一眼看出來該介面的返回值。
  • 每一個介面都需要增加額外的程式碼量。

所幸Spring Boot已經為我們提供了更好的解決辦法,只需要在專案中加上以下程式碼,就可以無感知的為我們統一全域性返回值。

/**
 * 全域性返回值統一封裝
 */
@EnableWebMvc
@Configuration
public class GlobalReturnConfig {

    @RestControllerAdvice
    static class ResultResponseAdvice implements ResponseBodyAdvice<Object> {
        @Override
        public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
            return true;
        }

        @Override
        public Object beforeBodyWrite(Object body, MethodParameter methodParameter, MediaType mediaType, Class<? extends HttpMessageConverter<?>> aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
            if (body instanceof Result) {
                return body;
            }
            return new Result(body);
        }
    }
}

而我們的介面只需要寫成最原始的樣子就行了。

	@GetMapping("/test")
    public User test() {
        return new User();
    }

統一處理異常

將返回值統一封裝時我們沒有考慮當介面丟擲異常的情況。當介面丟擲異常時讓使用者直接看到服務端的異常肯定是不夠友好的,而我們也不可能每一個介面都去try/catch進行處理,此時只需要使用@ExceptionHandler註解即可無感知的全域性統一處理異常。

@RestControllerAdvice
public class GlobalExceptionHandler {

    private static final Logger LOG = LoggerFactory.getLogger(GlobalExceptionHandler.class);

    /**
     * 全域性異常處理
     */
    @ExceptionHandler
    public JsonData handleException(HttpServletRequest request, HttpServletResponse response, final Exception e) {
        LOG.error(e.getMessage(), e);
        if (e instanceof AlertException) {//可以在前端Alert的異常
            if (((AlertException) e).getRetCode() != null) {//預定義異常
                return new Result(((AlertException) e).getRetCode());
            } else {
                return new Result(1, e.getMessage() != null ? e.getMessage() : "");
            }
        } else {//其它異常
            if (Util.isProduct()) {//如果是正式環境,統一提示
                return new Result(RetCode.ERROR);
            } else {//測試環境,alert異常資訊
                return new Result(1, StringUtils.isNotBlank(e.getMessage()) ? e.getMessage() : e.toString());
            }
        }
    }

}

其中的AlertException為我們自定義的異常,因此當業務中需要丟擲錯誤時,可以手動丟擲AlertException

以上就是統一處理返回值和統一處