Merge pull request #40 from allaboutapps/sharedAsyncChannel

SharedAsyncChannel

Sources/Toolbox/Utilities/SharedAsyncChannel.swift

@@ -0,0 +1,48 @@
+#if canImport(AsyncAlgorithms)
+import AsyncAlgorithms
+import Foundation
+
+public final class SharedAsyncChannel<Element: Sendable>: AsyncSequence, @unchecked Sendable {
+    public typealias Element = Element
+    public typealias AsyncIterator = Iterator
+    private let lock = NSLock()
+    private var storage = [UUID: AsyncChannel<Element>]()
+
+    public init() {}
+
+    public func send(_ element: Element) async {
+        for channel in storage.values {
+            await channel.send(element)
+        }
+    }
+
+    public func makeAsyncIterator() -> Iterator {
+        let channel = AsyncChannel<Element>()
+        let id = UUID()
+        lock.lock()
+        storage[id] = channel
+        lock.unlock()
+        return Iterator(innerIterator: channel.makeAsyncIterator(), parent: self, id: id)
+    }
+
+    private func gotCancelled(id: UUID) {
+        lock.lock()
+        storage[id] = nil
+        lock.unlock()
+    }
+
+    public struct Iterator: AsyncIteratorProtocol {
+        var innerIterator: AsyncChannel<Element>.AsyncIterator
+        let parent: SharedAsyncChannel<Element>
+        let id: UUID
+
+        public mutating func next() async -> Element? {
+            return await withTaskCancellationHandler {
+                await innerIterator.next()
+            } onCancel: { [parent, id] in
+                parent.gotCancelled(id: id)
+            }
+        }
+    }
+}
+#endif

Tests/ToolboxTests/TabBarCoordinatorTests.swift

@@ -10,7 +10,7 @@ final class TabBarCoordinatorTests: XCTestCase {
 
         // Switch to the main actor context.
         await MainActor.run {
-            let tabBarCoordinator = TabBarCoordinator()
+            let tabBarCoordinator = TabBarCoordinator(tabBarController: UITabBarController())
 
             // Test your coordinator instance here, e.g.:
             XCTAssertNotNil(tabBarCoordinator)