Spring Cloud - 通过 Gateway webflux 编程实现网关异常处理-CSDN博客

阿里云国内75折回扣微信号：monov8

阿里云国际，腾讯云国际，低至75折。AWS 93折免费开户实名账号代冲值优惠多多微信号：monov8 飞机：@monov6

一、webflux 编程实现网关异常处理

我们知道在某一个服务中出现异常可以通过 @ControllerAdvice + @ExceptionHandler 来统一异常处理即使是在微服务架构中我们也可以将上述统一异常处理放入到公共的微服务中这样哪一个微服务需要直接引入模块即可。但是如果我们的网关微服务出现了异常该怎么处理呢

步骤如下

在网关微服务中创建一个类添加 @Configuration 注解将其作为配置类.
实现自定义 ErrorWebExceptionHandler 接口重写 handle 方法.
在 handle 方法中实现统一异常处理.

代码注释如下

@Slf4j
@Configuration
public class GlobalExceptionConfig implements ErrorWebExceptionHandler {


    @Autowired
    private ObjectMapper objectMapper;

    /**
     * 全局异常处理
     * @param exchange 交换器(request, response)
     * @param ex 出现异常时的异常对象
     * @return
     */
    @Override
    public Mono<Void> handle(ServerWebExchange exchange, Throwable ex) {
        Map<String, String> result = new HashMap<>();
        //1.获取响应对象
        ServerHttpResponse response = exchange.getResponse();

        //2. response 是否结束(一般不用可以用来处理多个异常的场景)
        if(response.isCommitted()) {
            return Mono.error(ex);
        }

        //3.设置响应头类型(JSON)
        response.getHeaders().setContentType(MediaType.APPLICATION_JSON);

        //4.设置响应状态码
        if (ex instanceof IllegalTokenException) {
            //是我们自定义的异常
            response.setStatusCode(HttpStatus.FORBIDDEN);
        } else {
            //不是我们自定义的异常就通过 "服务器异常" 来处理
            response.setStatusCode(HttpStatus.INTERNAL_SERVER_ERROR);
        }

        //5.处理统一异常响应
        return response.writeWith(Mono.fromSupplier(() -> {
            DataBufferFactory bufferFactory = response.bufferFactory();
            result.put("msg", ex.getMessage());
            try {
                return bufferFactory.wrap(objectMapper.writeValueAsBytes(result));
            } catch (JsonProcessingException e) {
                throw new RuntimeException(e);
            }
        }));
    }

}

IllegalTokenException是我们自定义的异常通常在上述代码的第四步中就需要根据我们判断 ex 属于我们的哪种自定义异常然后进行对应的状态码处理.
Mono.fromSupplier()是一个函数式接口参数这里通过一 lamada 表达式实现无参返回值是一个 Mono泛型. 这里要实现的逻辑就是把异常的响应进行封装.
bufferFactory.wrap这里就是按照 webflux 编程的方式参数是 json 格式通过 ObjectMapper 转化得到一个 Mono 对象.