WIP: WebClient

[facebook-errors] / src / main / java / de / juplo / facebook / errors / GraphApiExchangeFilterFunction.java

diff --git a/src/main/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunction.java b/src/main/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunction.java
index 768ec1d..34ca5a7 100644 (file)
--- a/src/main/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunction.java
+++ b/src/main/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunction.java
@@ -1,13 +1,12 @@
 package de.juplo.facebook.errors;
 
 
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
 import org.springframework.http.HttpStatus;
 import org.springframework.web.reactive.function.client.ClientRequest;
 import org.springframework.web.reactive.function.client.ClientResponse;
 import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
 import org.springframework.web.reactive.function.client.ExchangeFunction;
+import org.springframework.web.reactive.function.client.WebClientResponseException;
 import reactor.core.publisher.Mono;
 
 
@@ -17,13 +16,6 @@ import reactor.core.publisher.Mono;
  */
 public class GraphApiExchangeFilterFunction implements ExchangeFilterFunction
 {
-  private static final Logger LOG =
-      LoggerFactory.getLogger(GraphApiExchangeFilterFunction.class);
-
-  public static GraphApiExchangeFilterFunction INSTANCE =
-      new GraphApiExchangeFilterFunction();
-
-
   @Override
   public Mono<ClientResponse> filter(ClientRequest request, ExchangeFunction next)
   {
@@ -36,10 +28,21 @@ public class GraphApiExchangeFilterFunction implements ExchangeFilterFunction
                   HttpStatus.Series.CLIENT_ERROR.equals(response.statusCode().series())
                       ? response
                           .bodyToMono(String.class)
-                          .flatMap(errorBody -> Mono.error(GraphApiException.create(
+                          .map(errorBody -> GraphApiException.create(
                               response.statusCode(),
                               response.headers().asHttpHeaders(),
-                              errorBody.getBytes())))
+                              errorBody.getBytes()))
+                          .map(e ->
+                              e.getClass().equals(ErrorResponseParsingErrorException.class)
+                                  ? WebClientResponseException
+                                      .create(
+                                          e.getStatus().value(),
+                                          e.getStatus().getReasonPhrase(),
+                                          e.getHeaders(),
+                                          null,
+                                          null)
+                                  : e)
+                          .flatMap(e -> Mono.error(e))
                       : Mono.just(response);
             });
   }