Spring Cloud Gateway之全局異常攔截器

/**
 * @version 2019/8/14
 * @description: 異常攔截器
 * @modified:
 */
@Slf4j
public class JsonExceptionHandler implements ErrorWebExceptionHandler {

    /**
     * MessageReader
     */
    private List<HttpMessageReader<?>> messageReaders = Collections.emptyList();

    /**
     * MessageWriter
     */
    private List<HttpMessageWriter<?>> messageWriters = Collections.emptyList();

    /**
     * ViewResolvers
     */
    private List<ViewResolver> viewResolvers = Collections.emptyList();

    /**
     * 存儲處理異常后的信息
     */
    private ThreadLocal<ResEntity> exceptionHandlerResult = new ThreadLocal<>();

    /**
     * 參考AbstractErrorWebExceptionHandler
     */
    public void setMessageReaders(List<HttpMessageReader<?>> messageReaders) {
        Assert.notNull(messageReaders, "'messageReaders' must not be null");
        this.messageReaders = messageReaders;
    }

    /**
     * 參考AbstractErrorWebExceptionHandler
     */
    public void setViewResolvers(List<ViewResolver> viewResolvers) {
        this.viewResolvers = viewResolvers;
    }

    /**
     * 參考AbstractErrorWebExceptionHandler
     */
    public void setMessageWriters(List<HttpMessageWriter<?>> messageWriters) {
        Assert.notNull(messageWriters, "'messageWriters' must not be null");
        this.messageWriters = messageWriters;
    }

    @Override
    public Mono<Void> handle(ServerWebExchange exchange, Throwable ex) {
        // 按照異常類型進行處理 默認500
　　　　　　int httpStatus = HttpStatus.INTERNAL_SERVER_ERROR.value();
　　　　　　String body = "系統異常，請聯系管理員";
　　　　　　if (ex instanceof NotFoundException) {
    　　　　　　httpStatus = HttpStatus.NOT_FOUND.value();
   　　　　　　 body = ex.getMessage();
　　　　　　}else if(ex instanceof ResponseStatusException){
    　　　　　　ResponseStatusException responseStatusException = (ResponseStatusException) ex;
    　　　　　　httpStatus = responseStatusException.getStatus().value();
    　　　　　　body = responseStatusException.getMessage();
　　　　　　} else if (ex instanceof BusinessException) {
    　　　　　　body  = ex.getMessage();
    　　　　　　httpStatus = ((BusinessException) ex).toResEntity().getHttpStatus();
　　　　　　} else if (ex instanceof RuntimeException) {
    　　　　　　Throwable cause = ex.getCause();
   　　　　　　 body  = ex.getMessage();
    　　　　　　if(null != cause && cause.getMessage().contains("Load balancer does not have available server for client")){
       　　　　　　 body = "服務不存在";
   　　　　　　 }
　　　　　　}
　　　　　　//錯誤記錄
        ServerHttpRequest request = exchange.getRequest();
        log.error("[全局異常處理]異常請求路徑:{}", request.getPath());
        log.error("異常詳細信息：{}",ex);
        //封裝響應體
        ResEntity res = new ResEntity();
        res.setHttpStatus(httpStatus);
        res.setMsg(body);

        //參考AbstractErrorWebExceptionHandler
        if (exchange.getResponse().isCommitted()) {
            return Mono.error(ex);
        }
        exceptionHandlerResult.set(res);
        ServerRequest newRequest = ServerRequest.create(exchange, this.messageReaders);
        return RouterFunctions.route(RequestPredicates.all(), this::renderErrorResponse).route(newRequest)
                .switchIfEmpty(Mono.error(ex))
                .flatMap((handler) -> handler.handle(newRequest))
                .flatMap((response) -> write(exchange, response));

    }

    /**
     * 參考DefaultErrorWebExceptionHandler
     */
    protected Mono<ServerResponse> renderErrorResponse(ServerRequest request) {
        ResEntity result = exceptionHandlerResult.get();

        //404 因前端框架問題，不轉換處理 ，因此目前只有系統報500才轉換為自定義的狀態碼
        if(HttpStatus.INTERNAL_SERVER_ERROR.value() == result.getHttpStatus()){
            return ServerResponse.status(HttpStatus.OK.value())
                    .contentType(MediaType.APPLICATION_JSON_UTF8)
                    .body(BodyInserters.fromObject(result));
        }
        return ServerResponse.status(result.getHttpStatus())
                .contentType(MediaType.APPLICATION_JSON_UTF8)
                .body(BodyInserters.fromObject(result));
    }

    /**
     * 參考AbstractErrorWebExceptionHandler
     */
    private Mono<? extends Void> write(ServerWebExchange exchange,
                                       ServerResponse response) {
        exchange.getResponse().getHeaders()
                .setContentType(response.headers().getContentType());
        return response.writeTo(exchange, new ResponseContext());
    }

    /**
     * 參考AbstractErrorWebExceptionHandler
     */
    private class ResponseContext implements ServerResponse.Context {

        @Override
        public List<HttpMessageWriter<?>> messageWriters() {
            return JsonExceptionHandler.this.messageWriters;
        }

        @Override
        public List<ViewResolver> viewResolvers() {
            return JsonExceptionHandler.this.viewResolvers;
        }
    }
}