fix: Exception propagation policy for Filtered retry

PlaytikaOSS · Dec 30, 2023 · dca239c · dca239c
1 parent 01ff94d
commit dca239c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/feign-reactor-core/src/main/java/reactivefeign/retry/FilteredReactiveRetryPolicy.java b/feign-reactor-core/src/main/java/reactivefeign/retry/FilteredReactiveRetryPolicy.java
@@ -1,5 +1,6 @@
 package reactivefeign.retry;
 
+import feign.ExceptionPropagationPolicy;
 import org.reactivestreams.Publisher;
 import reactor.core.Exceptions;
 import reactor.core.publisher.Flux;
@@ -14,6 +15,8 @@ public class FilteredReactiveRetryPolicy implements ReactiveRetryPolicy {
 
     private final ReactiveRetryPolicy retryPolicy;
     private final Predicate<Throwable> toRetryOn;
+    private final ExceptionPropagationPolicy exceptionPropagationPolicy;
+
 
     public static FilteredReactiveRetryPolicy notRetryOn(ReactiveRetryPolicy retryPolicy, Class<? extends Throwable>... errorClasses) {
         return new FilteredReactiveRetryPolicy(retryPolicy,
@@ -22,8 +25,18 @@ public static FilteredReactiveRetryPolicy notRetryOn(ReactiveRetryPolicy retryPo
     }
 
     public FilteredReactiveRetryPolicy(ReactiveRetryPolicy retryPolicy, Predicate<Throwable> toRetryOn) {
+        this(retryPolicy, toRetryOn, retryPolicy.exceptionPropagationPolicy());
+    }
+
+    public FilteredReactiveRetryPolicy(ReactiveRetryPolicy retryPolicy, Predicate<Throwable> toRetryOn, ExceptionPropagationPolicy exceptionPropagationPolicy) {
         this.retryPolicy = retryPolicy;
         this.toRetryOn = toRetryOn;
+        this.exceptionPropagationPolicy = exceptionPropagationPolicy;
+    }
+
+    @Override
+    public ExceptionPropagationPolicy exceptionPropagationPolicy(){
+        return exceptionPropagationPolicy;
     }
 
     @Override