feat: introduce allocative

Author: quininer

Cargo.lock

@@ -113,6 +113,17 @@ wasmparser          = { version = "0.222.0", default-features = false }
 winnow              = { version = "0.7.12", default-features = false, features = ["std", "simd"] }
 xxhash-rust         = { version = "0.8.14", default-features = false }
 
+allocative       = { package = "rspack-allocative", version = "0.3.5", default-features = false, features = [
+  "camino",
+  "dashmap",
+  "hashbrown",
+  "indexmap",
+  "tokio",
+  "once_cell",
+  "ustr",
+  "atomic_refcell",
+] }
+
 # Pinned
 napi        = { version = "3.1.6", default-features = false }
 napi-build  = { version = "2.2.3", default-features = false }
@@ -418,6 +429,10 @@ unused_import_braces                   = "warn"
 unused_lifetimes                       = "warn"
 unused_macro_rules                     = "warn"
 
+[workspace.lints.rust.unexpected_cfgs]
+level = "warn"
+check-cfg = ['cfg(allocative)']
+
 [workspace.lints.clippy]
 cargo_common_metadata   = "allow"
 empty_docs              = "allow" # there are some false positives inside biome_wasm

Cargo.toml

@@ -17,6 +17,7 @@ debug_tool      = ["rspack_binding_api/debug_tool"]
 info-level      = ["tracing/release_max_level_info"]
 plugin          = ["rspack_binding_api/plugin"]
 sftrace-setup   = ["rspack_binding_api/sftrace-setup"]
+allocative      = ["rspack_binding_api/allocative"]
 
 [package.metadata.cargo-shear]
 # Adding napi-derive as a dependency to workaround an issue where `dts` will no longer work without it.

crates/node_binding/Cargo.toml

@@ -78,6 +78,10 @@ async function build() {
 			features.push("sftrace-setup");
 			rustflags.push("-Zinstrument-xray=always");
 		}
+		if (process.env.ALLOCATIVE) {
+			features.push("allocative");
+			rustflags.push("--cfg=allocative");
+		}
 		if (values.profile === "release") {
 			features.push("info-level");
 		}

crates/node_binding/scripts/build.js

@@ -16,6 +16,12 @@ color-backtrace = ["dep:color-backtrace"]
 debug_tool      = ["rspack_core/debug_tool"]
 plugin          = ["rspack_loader_swc/plugin", "rspack_util/plugin"]
 sftrace-setup   = ["dep:sftrace-setup", "rspack_allocator/sftrace-setup"]
+allocative      = [
+  "rspack_util/allocative",
+  "rspack_collections/allocative",
+  "rspack_cacheable/allocative",
+  "rspack_tracing_perfetto/allocative",
+]
 
 [dependencies]
 anyhow                    = { workspace = true }
@@ -41,7 +47,8 @@ rspack_tasks              = { workspace = true }
 rspack_util               = { workspace = true }
 rspack_workspace          = { workspace = true }
 
-rspack_tracing = { workspace = true }
+rspack_tracing          = { workspace = true }
+rspack_tracing_perfetto = { workspace = true }
 
 async-trait        = { workspace = true }
 cow-utils          = { workspace = true }
@@ -120,7 +127,7 @@ tokio                                  = { workspace = true, features = ["rt", "
 ustr                                   = { workspace = true }
 
 [package.metadata.cargo-shear]
-ignored = ["parking_lot"]
+ignored = ["parking_lot", "rspack_tracing_perfetto"]
 
 [target.'cfg(target_family = "wasm")'.dependencies]
 # Pin parking_lot version to the same version within the workspace

crates/rspack_binding_api/Cargo.toml

@@ -10,6 +10,7 @@ version.workspace = true
 noop = []
 
 [dependencies]
+allocative      = { workspace = true, optional = true }
 camino          = { workspace = true }
 dashmap         = { workspace = true }
 hashlink        = { workspace = true }
@@ -27,3 +28,7 @@ smol_str        = { workspace = true }
 swc_core        = { workspace = true, features = ["ecma_ast"] }
 ustr            = { workspace = true }
 xxhash-rust     = { workspace = true, features = ["const_xxh64"] }
+
+[lints.rust.unexpected_cfgs]
+level = "warn"
+check-cfg = ['cfg(allocative)']

crates/rspack_cacheable/Cargo.toml

@@ -69,7 +69,9 @@ use std::sync::{Arc, LazyLock};
 
 use dashmap::DashSet;
 
+#[cfg_attr(allocative, allocative::root)]
 pub static CACHED_ARC_STR: LazyLock<DashSet<Arc<str>>> = LazyLock::new(Default::default);
+
 impl AsRefStrConverter for Arc<str> {
   fn as_str(&self) -> &str {
     self.as_ref()

crates/rspack_cacheable/src/with/as_ref_str.rs

@@ -16,3 +16,9 @@ rayon            = { workspace = true }
 rspack_cacheable = { workspace = true }
 serde            = { workspace = true, features = ["derive"] }
 ustr             = { workspace = true, features = ["serde"] }
+allocative       = { workspace = true, optional = true }
+
+[lints.rust.unexpected_cfgs]
+level = "warn"
+check-cfg = ['cfg(allocative)']
+

crates/rspack_collections/Cargo.toml

@@ -36,6 +36,7 @@ pub type IdentifierLinkedSet = LinkedHashSet<Identifier, BuildHasherDefault<Iden
 
 #[cacheable(hashable)]
 #[derive(Debug, Default, Clone, Copy, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize)]
+#[cfg_attr(allocative, derive(allocative::Allocative))]
 pub struct Identifier(#[cacheable(with=AsPreset)] Ustr);
 
 impl Deref for Identifier {

crates/rspack_collections/src/identifier.rs

@@ -35,6 +35,7 @@ pub trait ItemUkey {
 /// Ukey stands for Unique key
 #[rspack_cacheable::cacheable]
 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
+#[cfg_attr(allocative, derive(allocative::Allocative))]
 pub struct Ukey(u32);
 
 impl Ukey {