Calls clearData() on all data propagation providers.

Signed-off-by: Santiago Pericasgeertsen <[email protected]>
helidon-io · Feb 2, 2024 · 2a97af4 · 2a97af4
1 parent 6996630
commit 2a97af4
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/webclient/webclient/src/main/java/io/helidon/webclient/WebClientRequestBuilderImpl.java b/webclient/webclient/src/main/java/io/helidon/webclient/WebClientRequestBuilderImpl.java
@@ -646,8 +646,12 @@ private Single<WebClientResponse> invoke(Flow.Publisher<DataChunk> requestEntity
 
     @SuppressWarnings(value = "unchecked")
     private void runInContext(Map<Class<?>, Object> data, Runnable command) {
-        PROPAGATION_PROVIDERS.forEach(provider -> provider.propagateData(data.get(provider.getClass())));
-        Contexts.runInContext(context, command);
+        try{
+            PROPAGATION_PROVIDERS.forEach(provider -> provider.propagateData(data.get(provider.getClass())));
+            Contexts.runInContext(context, command);
+        } finally{
+            PROPAGATION_PROVIDERS.forEach(provider -> provider.clearData(data.get(provider.getClass())));
+        }
     }
 
     /**
@@ -674,13 +678,17 @@ public void onNext(T item) {
             }
 
             @Override
+            @SuppressWarnings("unchecked")
             public void onError(Throwable throwable) {
                 runInContext(contextProperties, () -> subscriber.onError(throwable));
+                PROPAGATION_PROVIDERS.forEach(provider -> provider.clearData(provider.getClass()));
             }
 
             @Override
+            @SuppressWarnings("unchecked")
             public void onComplete() {
                 runInContext(contextProperties, subscriber::onComplete);
+                PROPAGATION_PROVIDERS.forEach(provider -> provider.clearData(provider.getClass()));
             }
         }));
     }