Fix cancellation order in CancellableRateLimitedFluxIterator

Closes elastic#103054
fcofdez · Jan 11, 2024 · a318cad · a318cad
1 parent a8d0e9c
commit a318cad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...rc/main/java/org/elasticsearch/repositories/azure/CancellableRateLimitedFluxIterator.java b/...rc/main/java/org/elasticsearch/repositories/azure/CancellableRateLimitedFluxIterator.java
@@ -165,9 +165,9 @@ public void onNext(T element) {
     }
 
     public void cancel() {
+        done = true;
         cancelSubscription();
         clearQueue();
-        done = true;
         // cancel should be called from the consumer
         // thread, but to avoid potential deadlocks
         // we just try to release a possibly blocked
@@ -177,9 +177,9 @@ public void cancel() {
 
     @Override
     public void onError(Throwable t) {
+        done = true;
         clearQueue();
         error = t;
-        done = true;
         signalConsumer();
     }