RouterFunction中的Spring-Webflux错误处理

Question

我是Spring gateway cloud（Spring Reactor）的新手。我想实现聚合层，它将从两个微服务（如RealAPI gateway）下载数据。据我所知，在Spring Cloud Gateway中，它需要以反应式方式实现。

我发现RouterFunction可以创建一个网关endpoint来实现POST到其中一个微服务。它可以工作，但当微服务返回BAD_状态或其他时，我的spring网关的响应只有500个内部服务器错误，但我想要完整的消息。

@Bean
  public RouterFunction<ServerResponse> createNodes(TokenFilter tokenFilter, NodesHandler nodesHandler) {

    return RouterFunctions.route()
        .POST("/nodeNew", nodesHandler::createNodes)
        .filter(tokenFilter)
        .filter(dataNotFoundToBadRequest())
        .build();
  }

我有handler（现在很原始，我正在学习webflux）

public Mono<ServerResponse> createNodes(ServerRequest request) {


    return request.bodyToMono(RestNodeCreate.class)
        .flatMap(newNodeCreate -> {
          if (newNodeCreate.getAccessMethod().equals("TCPIP")) {
            return oldCatalog.createNode(Mono.just(newNodeCreate), getTokenFromRequest(request))
                .onErrorResume(throwable -> {
                  System.out.println("OLD RESPONSE ");
                  return Mono.error(throwable));
                });
          } else {
            return builNodesToNewCatalog(Mono.just(newNodeCreate))
                .flatMap(newNodeCreate1 -> newCatalog.createNode(Mono.just(newNodeCreate1)))
                .onErrorResume(throwable -> {
                  System.out.println("NEW RESPONSE ");
                  return Mono.error(throwable);
                });
          }
        })
        .flatMap(o -> ServerResponse.ok().body(Mono.just(o), Object.class));
}

以及两个服务newCatalog和oldCatalog，其中WebClient通过方法向特定微服务发出post请求：

public Mono<RestNodeCreate> createNode(Mono<RestNodeCreate> node, String token) {
    return webClient
        .post()
        .uri("/nodes")
        .contentType(MediaType.APPLICATION_JSON)
        .header(HttpHeaders.AUTHORIZATION, token)
        .body(node, RestNodeCreate.class)
        .retrieve()
        .onStatus(HttpStatus::is5xxServerError,
            response -> response.bodyToMono(String.class).flatMap(s -> Mono.error(new RuntimeException(s))))
        .bodyToMono(RestNodeCreate.class);
  }

如果在webClient请求中引发异常，将状态代码、消息等转发给RouterFunction的响应，是否有一种方法可以让标准的错误处理程序具有响应等？

{
   "timestamp": "forwarded from webclient response",
   "statuscode": "forwarded from webclient response", 
   "message": "forwarded from webclient response"
} 

在spring cloud gateway中，是否可以使用阻塞RestTemplate和synchronus@RestController编写它？

Answer 1

我找到了解决方案：

import static org.springframework.boot.web.error.ErrorAttributeOptions.Include.EXCEPTION;
import static org.springframework.boot.web.error.ErrorAttributeOptions.Include.MESSAGE;

import java.util.Map;
import org.springframework.boot.web.error.ErrorAttributeOptions;
import org.springframework.boot.web.reactive.error.DefaultErrorAttributes;
import org.springframework.boot.web.reactive.error.ErrorAttributes;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.reactive.function.server.ServerRequest;

@Configuration
public class ErrorAttributesConfiguration {

  @Bean
  public ErrorAttributes errorAttributes() {
    return new DefaultErrorAttributes() {

      @Override
      public Map<String, Object> getErrorAttributes(ServerRequest request, ErrorAttributeOptions options) {
        return super.getErrorAttributes(request, options.including(MESSAGE, EXCEPTION));
      }
    };
  }
}