Lock async stream inits (#11)

* Lock async stream inits

Fixes #10.

* wip

Author: stephencelis

Sources/ConcurrencyExtras/AsyncStream.swift

@@ -1,3 +1,5 @@
+import Foundation
+
 extension AsyncStream {
   /// Produces an `AsyncStream` from an `AsyncSequence` by consuming the sequence till it
   /// terminates, ignoring any failure.
@@ -51,10 +53,13 @@ extension AsyncStream {
   ///
   /// - Parameter sequence: An async sequence.
   public init<S: AsyncSequence>(_ sequence: S) where S.Element == Element {
+    let lock = NSLock()
     var iterator: S.AsyncIterator?
     self.init {
-      if iterator == nil {
-        iterator = sequence.makeAsyncIterator()
+      lock.withLock {
+        if iterator == nil {
+          iterator = sequence.makeAsyncIterator()
+        }
       }
       return try? await iterator?.next()
     }

Sources/ConcurrencyExtras/AsyncThrowingStream.swift

@@ -1,13 +1,18 @@
+import Foundation
+
 extension AsyncThrowingStream where Failure == Error {
   /// Produces an `AsyncThrowingStream` from an `AsyncSequence` by consuming the sequence till it
   /// terminates, rethrowing any failure.
   ///
   /// - Parameter sequence: An async sequence.
   public init<S: AsyncSequence>(_ sequence: S) where S.Element == Element {
+    let lock = NSLock()
     var iterator: S.AsyncIterator?
     self.init {
-      if iterator == nil {
-        iterator = sequence.makeAsyncIterator()
+      lock.withLock {
+        if iterator == nil {
+          iterator = sequence.makeAsyncIterator()
+        }
       }
       return try await iterator?.next()
     }

Sources/ConcurrencyExtras/Internal/Locking.swift

@@ -0,0 +1,11 @@
+import Foundation
+
+#if !(os(iOS) || os(macOS) || os(tvOS) || os(watchOS))
+  extension NSLock {
+    func withLock<R>(_ body: () throws -> R) rethrows -> R {
+      self.lock()
+      defer { self.unlock() }
+      return try body()
+    }
+  }
+#endif