Fix actuallyCancel logic

supabase · Jan 9, 2024 · 0f9be90 · 0f9be90
1 parent f20c806
commit 0f9be90
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Sources/_Helpers/URLSession+AsyncAwait.swift b/Sources/_Helpers/URLSession+AsyncAwait.swift
@@ -54,8 +54,7 @@ private actor URLSessionTaskCancellationHelper {
   }
 
   private func actuallyCancel() {
-    state = .cancelled
-
+    // Handle whatever needs to be done based on the current state
     switch state {
     case let .registered(task):
       task.cancel()
@@ -64,6 +63,9 @@ private actor URLSessionTaskCancellationHelper {
     case .initialized:
       break
     }
+
+    // Set state into cancelled to short circuit subsequent cancellations or registrations.
+    state = .cancelled
   }
 }