[Concurrency] Include more pieces of Concurrency in Embedded (AsyncStream, continuations)

We have been only including a subset of files and functionality on Embedded Concurrency, let's instead include all the
source files, and have a fine grained opt out on things that don't yet work. Namely, this is still avoiding clocks, task
sleeping and custom executors.

Add a test for AsyncStream and continuations on Embedded Concurrency.

Author: kubamracek

stdlib/public/Concurrency/CMakeLists.txt

@@ -263,39 +263,13 @@ if(SWIFT_SHOULD_BUILD_EMBEDDED_STDLIB AND SWIFT_SHOULD_BUILD_EMBEDDED_CONCURRENC
       IS_STDLIB IS_FRAGILE
 
       ${SWIFT_RUNTIME_CONCURRENCY_C_SOURCES}
+      ${SWIFT_RUNTIME_CONCURRENCY_SWIFT_SOURCES}
       CooperativeGlobalExecutor.cpp
 
-      # TODO: Only a subset of Swift Concurrency .swift sources, for now.
-      Actor.swift
-      AsyncLet.swift
-      CheckedContinuation.swift
-      Errors.swift
-      Executor.swift
-      ExecutorAssertions.swift
-      AsyncCompactMapSequence.swift
-      AsyncDropFirstSequence.swift
-      AsyncDropWhileSequence.swift
-      AsyncFilterSequence.swift
-      AsyncFlatMapSequence.swift
-      AsyncIteratorProtocol.swift
-      AsyncMapSequence.swift
-      AsyncPrefixSequence.swift
-      AsyncPrefixWhileSequence.swift
-      AsyncSequence.swift
-      AsyncThrowingCompactMapSequence.swift
-      AsyncThrowingDropWhileSequence.swift
-      AsyncThrowingFilterSequence.swift
-      AsyncThrowingFlatMapSequence.swift
-      AsyncThrowingMapSequence.swift
-      AsyncThrowingPrefixWhileSequence.swift
-      GlobalActor.swift
-      PartialAsyncTask.swift
-      Task.swift
-      TaskCancellation.swift
-
       SWIFT_COMPILE_FLAGS
         ${extra_swift_compile_flags} -enable-experimental-feature Embedded
         -parse-stdlib -DSWIFT_CONCURRENCY_EMBEDDED
+        -Xfrontend -emit-empty-object-file
         ${SWIFT_RUNTIME_CONCURRENCY_SWIFT_FLAGS}
       C_COMPILE_FLAGS
         ${extra_c_compile_flags} ${SWIFT_RUNTIME_CONCURRENCY_C_FLAGS} -DSWIFT_CONCURRENCY_EMBEDDED=1

stdlib/public/Concurrency/CheckedContinuation.swift

@@ -19,16 +19,15 @@ internal func logFailedCheck(_ message: UnsafeRawPointer)
 /// Implementation class that holds the `UnsafeContinuation` instance for
 /// a `CheckedContinuation`.
 @available(SwiftStdlib 5.1, *)
-@_unavailableInEmbedded
 internal final class CheckedContinuationCanary: @unchecked Sendable {
   // The instance state is stored in tail-allocated raw memory, so that
   // we can atomically check the continuation state.
 
   private init() { fatalError("must use create") }
 
   private static func _create(continuation: UnsafeRawPointer, function: String)
-      -> Self {
-    let instance = Builtin.allocWithTailElems_1(self,
+      -> CheckedContinuationCanary {
+    let instance = Builtin.allocWithTailElems_1(CheckedContinuationCanary.self,
       1._builtinWordValue,
       (UnsafeRawPointer?, String).self)
 
@@ -52,7 +51,7 @@ internal final class CheckedContinuationCanary: @unchecked Sendable {
   }
 
   internal static func create<T, E>(continuation: UnsafeContinuation<T, E>,
-                                 function: String) -> Self {
+                                 function: String) -> CheckedContinuationCanary {
     return _create(
         continuation: unsafeBitCast(continuation, to: UnsafeRawPointer.self),
         function: function)
@@ -79,7 +78,11 @@ internal final class CheckedContinuationCanary: @unchecked Sendable {
     // Log if the continuation was never consumed before the instance was
     // destructed.
     if _continuationPtr.pointee != nil {
+      #if !$Embedded
       logFailedCheck("SWIFT TASK CONTINUATION MISUSE: \(function) leaked its continuation without resuming it. This may cause tasks waiting on it to remain suspended forever.\n")
+      #else
+      fatalError("SWIFT TASK CONTINUATION MISUSE")
+      #endif
     }
   }
 }
@@ -120,7 +123,6 @@ internal final class CheckedContinuationCanary: @unchecked Sendable {
 /// you can replace one with the other in most circumstances,
 /// without making other changes.
 @available(SwiftStdlib 5.1, *)
-@_unavailableInEmbedded
 public struct CheckedContinuation<T, E: Error>: Sendable {
   private let canary: CheckedContinuationCanary
 
@@ -163,7 +165,11 @@ public struct CheckedContinuation<T, E: Error>: Sendable {
     if let c: UnsafeContinuation<T, E> = canary.takeContinuation() {
       c.resume(returning: value)
     } else {
+      #if !$Embedded
       fatalError("SWIFT TASK CONTINUATION MISUSE: \(canary.function) tried to resume its continuation more than once, returning \(value)!\n")
+      #else
+      fatalError("SWIFT TASK CONTINUATION MISUSE")
+      #endif
     }
   }
 
@@ -183,13 +189,16 @@ public struct CheckedContinuation<T, E: Error>: Sendable {
     if let c: UnsafeContinuation<T, E> = canary.takeContinuation() {
       c.resume(throwing: error)
     } else {
+      #if !$Embedded
       fatalError("SWIFT TASK CONTINUATION MISUSE: \(canary.function) tried to resume its continuation more than once, throwing \(error)!\n")
+      #else
+      fatalError("SWIFT TASK CONTINUATION MISUSE")
+      #endif
     }
   }
 }
 
 @available(SwiftStdlib 5.1, *)
-@_unavailableInEmbedded
 extension CheckedContinuation {
   /// Resume the task awaiting the continuation by having it either
   /// return normally or throw an error based on the state of the given
@@ -282,7 +291,6 @@ extension CheckedContinuation {
 /// - SeeAlso: `withUnsafeContinuation(function:_:)`
 /// - SeeAlso: `withUnsafeThrowingContinuation(function:_:)`
 @inlinable
-@_unavailableInEmbedded
 @available(SwiftStdlib 5.1, *)
 #if !$Embedded
 @backDeployed(before: SwiftStdlib 6.0)
@@ -307,7 +315,6 @@ public func withCheckedContinuation<T>(
 // introduction of #isolation.
 @available(SwiftStdlib 5.1, *)
 @_unsafeInheritExecutor // ABI compatibility with Swift 5.1
-@_unavailableInEmbedded
 @_silgen_name("$ss23withCheckedContinuation8function_xSS_yScCyxs5NeverOGXEtYalF")
 public func _unsafeInheritExecutor_withCheckedContinuation<T>(
   function: String = #function,
@@ -348,7 +355,6 @@ public func _unsafeInheritExecutor_withCheckedContinuation<T>(
 /// - SeeAlso: `withUnsafeContinuation(function:_:)`
 /// - SeeAlso: `withUnsafeThrowingContinuation(function:_:)`
 @inlinable
-@_unavailableInEmbedded
 @available(SwiftStdlib 5.1, *)
 #if !$Embedded
 @backDeployed(before: SwiftStdlib 6.0)
@@ -373,7 +379,6 @@ public func withCheckedThrowingContinuation<T>(
 // introduction of #isolation.
 @available(SwiftStdlib 5.1, *)
 @_unsafeInheritExecutor // ABI compatibility with Swift 5.1
-@_unavailableInEmbedded
 @_silgen_name("$ss31withCheckedThrowingContinuation8function_xSS_yScCyxs5Error_pGXEtYaKlF")
 public func _unsafeInheritExecutor_withCheckedThrowingContinuation<T>(
   function: String = #function,

stdlib/public/Concurrency/ContinuousClock.swift

@@ -11,6 +11,7 @@
 //===----------------------------------------------------------------------===//
 import Swift
 
+#if !$Embedded
 /// A clock that measures time that always increments and does not stop 
 /// incrementing while the system is asleep. 
 ///
@@ -33,6 +34,7 @@ public struct ContinuousClock: Sendable {
 
   public init() { }
 }
+#endif
 
 @available(SwiftStdlib 5.7, *)
 extension Duration {
@@ -49,6 +51,8 @@ extension Duration {
   }
 }
 
+#if !$Embedded
+
 @available(SwiftStdlib 5.7, *)
 extension Clock where Self == ContinuousClock {
   /// A clock that measures time that always increments but does not stop 
@@ -189,3 +193,5 @@ extension ContinuousClock.Instant: InstantProtocol {
     rhs.duration(to: lhs)
   }
 }
+
+#endif

stdlib/public/Concurrency/Deque/Deque+Codable.swift

@@ -14,6 +14,8 @@
 
 import Swift
 
+#if !$Embedded
+
 extension _Deque: Encodable where Element: Encodable {
   /// Encodes the elements of this deque into the given encoder in an unkeyed
   /// container.
@@ -50,3 +52,5 @@ extension _Deque: Decodable where Element: Decodable {
     }
   }
 }
+
+#endif

stdlib/public/Concurrency/Deque/Deque+CustomDebugStringConvertible.swift

@@ -12,7 +12,8 @@
 /// This file is copied from swift-collections and should not be modified here.
 /// Rather all changes should be made to swift-collections and copied back.
 
-#if !SWIFT_STDLIB_STATIC_PRINT
+#if !SWIFT_STDLIB_STATIC_PRINT && !$Embedded
+
 import Swift
 
 extension _Deque: CustomDebugStringConvertible {
@@ -32,4 +33,5 @@ extension _Deque: CustomDebugStringConvertible {
     return result
   }
 }
-#endif
+
+#endif

stdlib/public/Concurrency/Deque/Deque+CustomStringConvertible.swift

@@ -12,7 +12,8 @@
 /// This file is copied from swift-collections and should not be modified here.
 /// Rather all changes should be made to swift-collections and copied back.
 
-#if !SWIFT_STDLIB_STATIC_PRINT
+#if !SWIFT_STDLIB_STATIC_PRINT && !$Embedded
+
 import Swift
 
 extension _Deque: CustomStringConvertible {
@@ -32,4 +33,5 @@ extension _Deque: CustomStringConvertible {
     return result
   }
 }
-#endif
+
+#endif

stdlib/public/Concurrency/GlobalConcurrentExecutor.swift

@@ -31,6 +31,7 @@ import Swift
 ///
 /// Customizing the global concurrent executor is currently not supported.
 @available(SwiftStdlib 6.0, *)
+@_unavailableInEmbedded
 public var globalConcurrentExecutor: any TaskExecutor {
   get {
     _DefaultGlobalConcurrentExecutor.shared
@@ -43,6 +44,7 @@ public var globalConcurrentExecutor: any TaskExecutor {
 /// thread pool that is used as the default executor for Swift concurrency
 /// tasks.
 @available(SwiftStdlib 6.0, *)
+@_unavailableInEmbedded
 internal final class _DefaultGlobalConcurrentExecutor: TaskExecutor {
   public static let shared: _DefaultGlobalConcurrentExecutor = .init()
 

stdlib/public/Concurrency/MainActor.swift

@@ -12,6 +12,8 @@
 
 import Swift
 
+#if !$Embedded
+
 #if SWIFT_STDLIB_TASK_TO_THREAD_MODEL_CONCURRENCY
 @available(SwiftStdlib 5.1, *)
 @available(*, unavailable, message: "Unavailable in task-to-thread concurrency model")
@@ -157,3 +159,5 @@ extension MainActor {
   }
 }
 #endif
+
+#endif

stdlib/public/Concurrency/SuspendingClock.swift

@@ -11,6 +11,8 @@
 //===----------------------------------------------------------------------===//
 import Swift
 
+#if !$Embedded
+
 /// A clock that measures time that always increments but stops incrementing 
 /// while the system is asleep. 
 ///
@@ -179,3 +181,4 @@ extension SuspendingClock.Instant: InstantProtocol {
   }
 }
 
+#endif

stdlib/public/Concurrency/Task+TaskExecutor.swift

@@ -190,6 +190,7 @@ public func _unsafeInheritExecutor_withTaskExecutorPreference<T: Sendable>(
 /// Task with specified executor -----------------------------------------------
 
 @available(SwiftStdlib 6.0, *)
+@_unavailableInEmbedded
 extension Task where Failure == Never {
   /// Runs the given nonthrowing operation asynchronously
   /// as part of a new top-level task on behalf of the current actor.
@@ -257,6 +258,7 @@ extension Task where Failure == Never {
 }
 
 @available(SwiftStdlib 6.0, *)
+@_unavailableInEmbedded
 extension Task where Failure == Error {
   /// Runs the given throwing operation asynchronously
   /// as part of a new top-level task on behalf of the current actor.
@@ -321,6 +323,7 @@ extension Task where Failure == Error {
 // ==== Detached tasks ---------------------------------------------------------
 
 @available(SwiftStdlib 6.0, *)
+@_unavailableInEmbedded
 extension Task where Failure == Never {
   /// Runs the given nonthrowing operation asynchronously
   /// as part of a new top-level task.
@@ -379,6 +382,7 @@ extension Task where Failure == Never {
 }
 
 @available(SwiftStdlib 6.0, *)
+@_unavailableInEmbedded
 extension Task where Failure == Error {
   /// Runs the given throwing operation asynchronously
   /// as part of a new top-level task.
@@ -441,6 +445,7 @@ extension Task where Failure == Error {
 // ==== Unsafe Current Task ----------------------------------------------------
 
 @available(SwiftStdlib 6.0, *)
+@_unavailableInEmbedded
 extension UnsafeCurrentTask {
 
   /// The current ``TaskExecutor`` preference, if this task has one configured.