Merge pull request #45 from outfoxx/fix/data-stream-cancel

Cancel dataEventStream’s URLSession task when stream cancelled
outfoxx · May 24, 2023 · 67e71a8 · 67e71a8
2 parents f2abafd + 30332ee
commit 67e71a8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Sources/Sunday/NetworkSession.swift b/Sources/Sunday/NetworkSession.swift
@@ -115,8 +115,13 @@ public class NetworkSession {
     }
 
     return AsyncThrowingStream(DataEvent.self) {
+
       let task = session.dataTask(with: request)
+
       setTaskDelegate(DataStreamDelegate(continuation: $0), for: task)
+
+      $0.onTermination = { _ in task.cancel() }
+
       task.resume()
     }