Allow setting exec_group in more places

Signed-off-by: Brentley Jones <[email protected]>

Author: brentleyjones

doc/api.md

@@ -411,9 +411,9 @@ A tuple of `(CcLinkingContext, CcLinkingOutputs)` containing the linking
 
 <pre>
 swift_common.extract_symbol_graph(*, <a href="#swift_common.extract_symbol_graph-actions">actions</a>, <a href="#swift_common.extract_symbol_graph-compilation_contexts">compilation_contexts</a>, <a href="#swift_common.extract_symbol_graph-emit_extension_block_symbols">emit_extension_block_symbols</a>,
-                                  <a href="#swift_common.extract_symbol_graph-feature_configuration">feature_configuration</a>, <a href="#swift_common.extract_symbol_graph-include_dev_srch_paths">include_dev_srch_paths</a>, <a href="#swift_common.extract_symbol_graph-minimum_access_level">minimum_access_level</a>,
-                                  <a href="#swift_common.extract_symbol_graph-module_name">module_name</a>, <a href="#swift_common.extract_symbol_graph-output_dir">output_dir</a>, <a href="#swift_common.extract_symbol_graph-swift_infos">swift_infos</a>, <a href="#swift_common.extract_symbol_graph-swift_toolchain">swift_toolchain</a>,
-                                  <a href="#swift_common.extract_symbol_graph-toolchain_type">toolchain_type</a>)
+                                  <a href="#swift_common.extract_symbol_graph-exec_group">exec_group</a>, <a href="#swift_common.extract_symbol_graph-feature_configuration">feature_configuration</a>, <a href="#swift_common.extract_symbol_graph-include_dev_srch_paths">include_dev_srch_paths</a>,
+                                  <a href="#swift_common.extract_symbol_graph-minimum_access_level">minimum_access_level</a>, <a href="#swift_common.extract_symbol_graph-module_name">module_name</a>, <a href="#swift_common.extract_symbol_graph-output_dir">output_dir</a>, <a href="#swift_common.extract_symbol_graph-swift_infos">swift_infos</a>,
+                                  <a href="#swift_common.extract_symbol_graph-swift_toolchain">swift_toolchain</a>, <a href="#swift_common.extract_symbol_graph-toolchain_type">toolchain_type</a>)
 </pre>
 
 Extracts the symbol graph from a Swift module.
@@ -426,6 +426,7 @@ Extracts the symbol graph from a Swift module.
 | <a id="swift_common.extract_symbol_graph-actions"></a>actions |  The object used to register actions.   |  none |
 | <a id="swift_common.extract_symbol_graph-compilation_contexts"></a>compilation_contexts |  A list of `CcCompilationContext`s that represent C/Objective-C requirements of the target being compiled, such as Swift-compatible preprocessor defines, header search paths, and so forth. These are typically retrieved from the `CcInfo` providers of a target's dependencies.   |  none |
 | <a id="swift_common.extract_symbol_graph-emit_extension_block_symbols"></a>emit_extension_block_symbols |  A `bool` that indicates whether `extension` block information should be included in the symbol graph.   |  `None` |
+| <a id="swift_common.extract_symbol_graph-exec_group"></a>exec_group |  Runs the Swift compilation action under the given execution group's context. If `None`, the default execution group is used.   |  `None` |
 | <a id="swift_common.extract_symbol_graph-feature_configuration"></a>feature_configuration |  The Swift feature configuration.   |  none |
 | <a id="swift_common.extract_symbol_graph-include_dev_srch_paths"></a>include_dev_srch_paths |  A `bool` that indicates whether the developer framework search paths will be added to the compilation command.   |  none |
 | <a id="swift_common.extract_symbol_graph-minimum_access_level"></a>minimum_access_level |  The minimum access level of the declarations that should be extracted into the symbol graphs. The default value is `None`, which means the Swift compiler's default behavior should be used (at the time of this writing, the default behavior is "public").   |  `None` |

proto/swift_proto_utils.bzl

@@ -196,6 +196,7 @@ def compile_swift_protos_for_target(
         additional_swift_proto_compiler_info,
         attr,
         ctx,
+        exec_group = None,
         module_name,
         proto_infos,
         swift_proto_compilers,
@@ -208,6 +209,8 @@ def compile_swift_protos_for_target(
         additional_compiler_deps: Additional dependencies passed directly to the Swift compiler.
         attr: The attributes of the target for which the module is being compiled.
         ctx: The context of the aspect or rule.
+        exec_group: Runs the Swift compilation action under the given execution
+            group's context. If `None`, the default execution group is used.
         module_name: The name of the Swift module that should be compiled from the protos.
         proto_infos: List of `ProtoInfo` providers to compile into Swift source files.
         swift_proto_compilers: List of targets propagating `SwiftProtoCompiler` providers.
@@ -276,6 +279,7 @@ def compile_swift_protos_for_target(
         additional_inputs = additional_inputs,
         cc_infos = get_providers(compiler_deps, CcInfo),
         copts = ["-parse-as-library"] + copts,
+        exec_group = exec_group,
         feature_configuration = feature_configuration,
         include_dev_srch_paths = include_dev_srch_paths,
         module_name = module_name,

swift/internal/symbol_graph_extracting.bzl

@@ -25,6 +25,7 @@ def extract_symbol_graph(
         actions,
         compilation_contexts,
         emit_extension_block_symbols = None,
+        exec_group = None,
         feature_configuration,
         include_dev_srch_paths,
         minimum_access_level = None,
@@ -44,6 +45,8 @@ def extract_symbol_graph(
             a target's dependencies.
         emit_extension_block_symbols: A `bool` that indicates whether `extension` block
             information should be included in the symbol graph.
+        exec_group: Runs the Swift compilation action under the given execution
+            group's context. If `None`, the default execution group is used.
         feature_configuration: The Swift feature configuration.
         include_dev_srch_paths: A `bool` that indicates whether the developer
             framework search paths will be added to the compilation command.
@@ -114,6 +117,7 @@ def extract_symbol_graph(
     run_toolchain_action(
         actions = actions,
         action_name = SWIFT_ACTION_SYMBOL_GRAPH_EXTRACT,
+        exec_group = exec_group,
         feature_configuration = feature_configuration,
         outputs = [output_dir],
         prerequisites = prerequisites,