task group embedded workaround

Author: ktoso

stdlib/public/Concurrency/TaskGroup+Embedded.swift

@@ -0,0 +1,151 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2025 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+import Swift
+
+// FIXME: This is a workaround for trouble including gyb-generated sources in the embedded build.
+
+#if SWIFT_CONCURRENCY_EMBEDDED
+
+@available(SwiftStdlib 5.1, *)
+extension TaskGroup {
+
+  @available(SwiftStdlib 5.1, *)
+  @_alwaysEmitIntoClient
+  public mutating func addTask(
+    priority: TaskPriority? = nil,
+    operation: sending @escaping @isolated(any) () async -> ChildTaskResult
+  ) {
+    let flags = taskCreateFlags(
+      priority: priority,
+      isChildTask: true,
+      copyTaskLocals: false,
+      inheritContext: false,
+      enqueueJob: true,
+      addPendingGroupTaskUnconditionally: true,
+      isDiscardingTask: false
+    )
+
+    let builtinSerialExecutor =
+      Builtin.extractFunctionIsolation(operation)?.unownedExecutor.executor
+
+    var task: Builtin.NativeObject? =
+      Builtin.createTask(
+        flags: flags,
+        initialSerialExecutor: builtinSerialExecutor,
+        taskGroup: _group,
+        operation: operation).0
+
+    // Assert that we did create the task, but there's no need to store it,
+    // as it was added to the group itself.
+    assert(task != nil, "Expected task to be created!")
+  }
+
+  @available(SwiftStdlib 5.1, *)
+  @_alwaysEmitIntoClient
+  public mutating func addTaskUnlessCancelled(
+    priority: TaskPriority? = nil,
+    operation: sending @escaping @isolated(any) () async -> ChildTaskResult
+  ) {
+    let flags = taskCreateFlags(
+      priority: priority,
+      isChildTask: true,
+      copyTaskLocals: false,
+      inheritContext: false,
+      enqueueJob: true,
+      addPendingGroupTaskUnconditionally: false,
+      isDiscardingTask: false
+    )
+
+    let builtinSerialExecutor =
+      Builtin.extractFunctionIsolation(operation)?.unownedExecutor.executor
+
+    var task: Builtin.NativeObject? =
+      Builtin.createTask(
+        flags: flags,
+        initialSerialExecutor: builtinSerialExecutor,
+        taskGroup: _group,
+        operation: operation).0
+
+    // Assert that we did create the task, but there's no need to store it,
+    // as it was added to the group itself.
+    assert(task != nil, "Expected task to be created!")
+  }
+}
+
+@available(SwiftStdlib 5.1, *)
+extension ThrowingTaskGroup {
+
+  @available(SwiftStdlib 5.1, *)
+  @_alwaysEmitIntoClient
+  public mutating func addTask(
+    priority: TaskPriority? = nil,
+    operation: sending @escaping @isolated(any) () async throws -> ChildTaskResult
+  ) {
+    let flags = taskCreateFlags(
+      priority: priority,
+      isChildTask: true,
+      copyTaskLocals: false,
+      inheritContext: false,
+      enqueueJob: true,
+      addPendingGroupTaskUnconditionally: true,
+      isDiscardingTask: false
+    )
+
+    let builtinSerialExecutor =
+      Builtin.extractFunctionIsolation(operation)?.unownedExecutor.executor
+
+    var task: Builtin.NativeObject? =
+      Builtin.createTask(
+        flags: flags,
+        initialSerialExecutor: builtinSerialExecutor,
+        taskGroup: _group,
+        operation: operation).0
+
+    // Assert that we did create the task, but there's no need to store it,
+    // as it was added to the group itself.
+    assert(task != nil, "Expected task to be created!")
+  }
+
+  @available(SwiftStdlib 5.1, *)
+  @_alwaysEmitIntoClient
+  public mutating func addTaskUnlessCancelled(
+    priority: TaskPriority? = nil,
+    operation: sending @escaping @isolated(any) () async throws -> ChildTaskResult
+  ) {
+    let flags = taskCreateFlags(
+      priority: priority,
+      isChildTask: true,
+      copyTaskLocals: false,
+      inheritContext: false,
+      enqueueJob: true,
+      addPendingGroupTaskUnconditionally: false,
+      isDiscardingTask: false
+    )
+
+    let builtinSerialExecutor =
+      Builtin.extractFunctionIsolation(operation)?.unownedExecutor.executor
+
+    var task: Builtin.NativeObject? =
+      Builtin.createTask(
+        flags: flags,
+        initialSerialExecutor: builtinSerialExecutor,
+        taskGroup: _group,
+        operation: operation).0
+
+    // Assert that we did create the task, but there's no need to store it,
+    // as it was added to the group itself.
+    assert(task != nil, "Expected task to be created!")
+  }
+}
+
+#endif