- fixed leak in CoroutineScheduler.startCoroutine() when pass `CoSc…

…ope`.
belozierov · Feb 1, 2021 · 9c395df · 9c395df
1 parent c137153
commit 9c395df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/SwiftCoroutine/Coroutine/TaskScheduler/CoroutineScheduler.swift b/Sources/SwiftCoroutine/Coroutine/TaskScheduler/CoroutineScheduler.swift
@@ -67,9 +67,9 @@ extension CoroutineScheduler {
     ///   - task: The closure that will be executed inside coroutine. If the task throws an error, then the coroutine will be terminated.
     public func startCoroutine(in scope: CoScope? = nil, task: @escaping () throws -> Void) {
         guard let scope = scope else { return _startCoroutine { try? task() } }
-        _startCoroutine {
+        _startCoroutine { [weak scope] in
             guard let coroutine = try? Coroutine.current(),
-                let completion = scope.add(coroutine.cancel) else { return }
+                let completion = scope?.add(coroutine.cancel) else { return }
             try? task()
             completion()
         }