Add resource_set attr to all expensive rules

This allows better control over resource usage for expensive rules in
resource-constrained systems.

Author: illicitonion

MODULE.bazel

@@ -16,6 +16,7 @@ bazel_dep(name = "rules_cc", version = "0.1.1")
 bazel_dep(name = "rules_license", version = "1.0.0")
 bazel_dep(name = "rules_shell", version = "0.3.0")
 bazel_dep(name = "apple_support", version = "1.17.1", repo_name = "build_bazel_apple_support")
+bazel_dep(name = "aspect_bazel_lib", version = "2.19.4")
 
 internal_deps = use_extension("//rust/private:internal_extensions.bzl", "i")
 use_repo(

rust/private/rust.bzl

@@ -14,6 +14,7 @@
 
 """Rust rule implementations"""
 
+load("@aspect_bazel_lib//lib:resource_sets.bzl", "resource_set_attr")
 load("@bazel_skylib//lib:paths.bzl", "paths")
 load("@rules_cc//cc/common:cc_info.bzl", "CcInfo")
 load("//rust/private:common.bzl", "COMMON_PROVIDERS", "rust_common")
@@ -817,7 +818,7 @@ _rust_test_attrs = {
 rust_library = rule(
     implementation = _rust_library_impl,
     provides = COMMON_PROVIDERS,
-    attrs = _common_attrs | {
+    attrs = _common_attrs | resource_set_attr | {
         "disable_pipelining": attr.bool(
             default = False,
             doc = dedent("""\
@@ -915,7 +916,7 @@ _rust_static_library_transition = transition(
 
 rust_static_library = rule(
     implementation = _rust_static_library_impl,
-    attrs = _common_attrs | {
+    attrs = _common_attrs | resource_set_attr | {
         "platform": attr.label(
             doc = "Optional platform to transition the static library to.",
             default = None,
@@ -964,7 +965,7 @@ _rust_shared_library_transition = transition(
 
 rust_shared_library = rule(
     implementation = _rust_shared_library_impl,
-    attrs = _common_attrs | _experimental_use_cc_common_link_attrs | {
+    attrs = _common_attrs | _experimental_use_cc_common_link_attrs | resource_set_attr | {
         "platform": attr.label(
             doc = "Optional platform to transition the shared library to.",
             default = None,
@@ -1017,7 +1018,7 @@ rust_proc_macro = rule(
     # need to declare `_allowlist_function_transition`, see
     # https://docs.bazel.build/versions/main/skylark/config.html#user-defined-transitions.
     attrs = dict(
-        _common_attrs.items(),
+        (_common_attrs | resource_set_attr).items(),
         _allowlist_function_transition = attr.label(
             default = Label("//tools/allowlists/function_transition_allowlist"),
         ),
@@ -1105,7 +1106,7 @@ _rust_binary_transition = transition(
 rust_binary = rule(
     implementation = _rust_binary_impl,
     provides = COMMON_PROVIDERS,
-    attrs = _common_attrs | _rust_binary_attrs | {
+    attrs = _common_attrs | _rust_binary_attrs | resource_set_attr | {
         "platform": attr.label(
             doc = "Optional platform to transition the binary to.",
             default = None,
@@ -1294,7 +1295,7 @@ _rust_test_transition = transition(
 rust_test = rule(
     implementation = _rust_test_impl,
     provides = COMMON_PROVIDERS,
-    attrs = _common_attrs | _rust_test_attrs | {
+    attrs = _common_attrs | _rust_test_attrs | resource_set_attr | {
         "platform": attr.label(
             doc = "Optional platform to transition the test to.",
             default = None,

rust/private/rustc.bzl

@@ -14,6 +14,7 @@
 
 """Functionality for constructing actions that invoke the Rust compiler"""
 
+load("@aspect_bazel_lib//lib:resource_sets.bzl", "resource_set")
 load("@bazel_skylib//rules:common_settings.bzl", "BuildSettingInfo")
 load(
     "@bazel_tools//tools/build_defs/cc:action_names.bzl",
@@ -1357,6 +1358,9 @@ def rustc_compile_action(
             action_outputs.append(dsym_folder)
 
     if ctx.executable._process_wrapper:
+        rustc_resource_set = resource_set(ctx.attr)
+        if rustc_resource_set == None:
+            rustc_resource_set = get_rustc_resource_set(toolchain)
         # Run as normal
         ctx.actions.run(
             executable = ctx.executable._process_wrapper,
@@ -1371,8 +1375,8 @@ def rustc_compile_action(
                 formatted_version,
                 len(crate_info.srcs.to_list()),
             ),
+            resource_set = rustc_resource_set,
             toolchain = "@rules_rust//rust:toolchain_type",
-            resource_set = get_rustc_resource_set(toolchain),
         )
         if args_metadata:
             ctx.actions.run(
@@ -1388,6 +1392,7 @@ def rustc_compile_action(
                     formatted_version,
                     len(crate_info.srcs.to_list()),
                 ),
+                resource_set = resource_set(ctx.attr),
                 toolchain = "@rules_rust//rust:toolchain_type",
             )
     elif hasattr(ctx.executable, "_bootstrap_process_wrapper"):

test/unit/pipelined_compilation/wrap.bzl

@@ -1,6 +1,7 @@
 """A custom rule that wraps a crate called to_wrap."""
 
 load("@rules_cc//cc/common:cc_info.bzl", "CcInfo")
+load("@aspect_bazel_lib//lib:resource_sets.bzl", "resource_set_attr")
 
 # buildifier: disable=bzl-visibility
 load("//rust/private:providers.bzl", "BuildInfo", "CrateInfo", "DepInfo", "DepVariantInfo")
@@ -82,7 +83,7 @@ def _wrap_impl(ctx):
 
 wrap = rule(
     implementation = _wrap_impl,
-    attrs = {
+    attrs = resource_set_attr | {
         "crate_name": attr.string(),
         "generate_metadata": attr.bool(default = False),
         "target": attr.label(),