Merge branch 'main' into memory-pool

Author: SyMind

Cargo.lock

@@ -35,7 +35,6 @@ serde_json = "1.0.133"
 dyn-clone = "1.0.17"
 rustc-hash = "2.1.0"
 memchr = "2.7.4"
-itertools = "0.13"
 
 
 codspeed-criterion-compat = { version = "2.7.2", default-features = false, optional = true }

Cargo.toml

@@ -30,6 +30,8 @@ use benchmark_repetitive_react_components::{
   benchmark_repetitive_react_components_source,
 };
 
+use crate::bench_complex_replace_source::benchmark_complex_replace_source_size;
+
 const HELLOWORLD_JS: &str = include_str!(concat!(
   env!("CARGO_MANIFEST_DIR"),
   "/benches/fixtures/transpile-minify/files/helloworld.js"
@@ -164,6 +166,11 @@ fn bench_rspack_sources(criterion: &mut Criterion) {
     benchmark_complex_replace_source_source,
   );
 
+  group.bench_function(
+    "complex_replace_source_size",
+    benchmark_complex_replace_source_size,
+  );
+
   group.bench_function(
     "parse_source_map_from_json",
     benchmark_parse_source_map_from_json,

benches/bench.rs

@@ -36738,3 +36738,11 @@ pub fn benchmark_complex_replace_source_source(b: &mut Bencher) {
     black_box(source.source());
   });
 }
+
+pub fn benchmark_complex_replace_source_size(b: &mut Bencher) {
+  let source = LARGE_REPLACE_SOURCE.clone();
+
+  b.iter(|| {
+    black_box(source.size());
+  });
+}

benches/bench_complex_replace_source.rs

@@ -19,6 +19,7 @@ use crate::{
 #[derive(Default)]
 struct CachedData {
   hash: OnceLock<u64>,
+  size: OnceLock<usize>,
   line_only_map: OnceLock<Option<SourceMap>>,
   columns_map: OnceLock<Option<SourceMap>>,
 }
@@ -95,7 +96,7 @@ impl Source for CachedSource {
   }
 
   fn size(&self) -> usize {
-    self.inner.size()
+    *self.cache.size.get_or_init(|| self.inner.size())
   }
 
   fn map(&self, options: &MapOptions) -> Option<SourceMap> {