cloub spring 拦截器_Spring Cloud Gateway之全局异常拦截器

/**

* @version 2019/8/14

* @description: 异常拦截器

* @modified:

*/

@Slf4j

public class JsonExceptionHandler implements ErrorWebExceptionHandler {

/**

* MessageReader

*/

private List> messageReaders = Collections.emptyList();

/**

* MessageWriter

*/

private List> messageWriters = Collections.emptyList();

/**

* ViewResolvers

*/

private List viewResolvers = Collections.emptyList();

/**

* 存储处理异常后的信息

*/

private ThreadLocal exceptionHandlerResult = new ThreadLocal<>();

/**

* 参考AbstractErrorWebExceptionHandler

*/

public void setMessageReaders(List> messageReaders) {

Assert.notNull(messageReaders, "'messageReaders' must not be null");

this.messageReaders = messageReaders;

}

/**

* 参考AbstractErrorWebExceptionHandler

*/

public void setViewResolvers(List viewResolvers) {

this.viewResolvers = viewResolvers;

}

/**

* 参考AbstractErrorWebExceptionHandler

*/

public void setMessageWriters(List> messageWriters) {

Assert.notNull(messageWriters, "'messageWriters' must not be null");

this.messageWriters = messageWriters;

}

@Override

public Mono 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 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> messageWriters() {

return JsonExceptionHandler.this.messageWriters;

}

@Override

public List viewResolvers() {

return JsonExceptionHandler.this.viewResolvers;

}

}

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值