test(benchmark): add basic rust benchmark (#9054)

h-a-n-a · web-flow · commit ad79086197f8 · 2025-01-21T15:09:26.000+08:00
* test: add basic rust benchmark

* chore: lint

* feat: include symbol
diff --git a/.github/workflows/reusable-build.yml b/.github/workflows/reusable-build.yml
@@ -453,6 +453,8 @@ jobs:
           tool: cargo-codspeed
 
       - name: Build Benchmark
+        env:
+          RUSTFLAGS: "-C debuginfo=1 -C strip=none -g --cfg codspeed"
         run: cargo codspeed build -p rspack_benchmark --features codspeed
 
       - name: Run benchmark
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/tasks/benchmark/Cargo.toml b/tasks/benchmark/Cargo.toml
@@ -12,7 +12,11 @@ default = []
 codspeed = ["criterion2/codspeed"]
 
 [dependencies]
-criterion2 = { default-features = false, version = "2.0.0"}
+criterion2 = { default-features = false, version = "2.0.0", features = ["async_tokio"]}
+rspack = { workspace = true }
+rspack_fs = { workspace = true }
+rspack_core = { workspace = true }
+tokio = { workspace = true }
 
 [[bench]]
 name    = "basic"
diff --git a/tasks/benchmark/benches/basic.rs b/tasks/benchmark/benches/basic.rs
@@ -1,15 +1,58 @@
-use rspack_benchmark::{black_box, criterion_group, criterion_main, Criterion};
+#![allow(clippy::unwrap_used)]
+#![feature(trait_upcasting)]
 
-fn fibonacci(n: u64) -> u64 {
-  match n {
-    0 => 1,
-    1 => 1,
-    n => fibonacci(n - 1) + fibonacci(n - 2),
-  }
+use std::sync::Arc;
+
+use rspack::builder::Builder as _;
+use rspack_benchmark::{criterion_group, criterion_main, Criterion};
+use rspack_core::Compiler;
+use rspack_fs::{MemoryFileSystem, ReadableFileSystem, WritableFileSystem};
+use tokio::runtime::Builder;
+
+trait FileSystem: ReadableFileSystem + WritableFileSystem + Send + Sync {}
+impl<T: ReadableFileSystem + WritableFileSystem + Send + Sync> FileSystem for T {}
+
+async fn basic(fs: Arc<dyn FileSystem>) {
+  let mut compiler = Compiler::builder()
+    .context("/")
+    .entry("main", "./src/index.js")
+    .input_filesystem(fs.clone())
+    .output_filesystem(fs.clone())
+    .build();
+
+  compiler.run().await.unwrap();
+  assert!(compiler
+    .compilation
+    .get_errors()
+    .collect::<Vec<_>>()
+    .is_empty());
 }
 
 pub fn criterion_benchmark(c: &mut Criterion) {
-  c.bench_function("fib 20", |b| b.iter(|| fibonacci(black_box(20))));
+  c.bench_function("basic", |b| {
+    let rt = Builder::new_multi_thread().build().unwrap();
+
+    let fs = MemoryFileSystem::default();
+    rt.block_on(async {
+      fs.create_dir_all("/src".into()).await.unwrap();
+      fs.write(
+        "/src/index.js".into(),
+        br#"if(process.env.NODE_ENV === "production") {
+    console.log(123);
+  } else {
+    console.log(456);
+  }
+  "#,
+      )
+      .await
+      .unwrap();
+    });
+    let fs = Arc::new(fs);
+    b.to_async(&rt).iter(|| {
+      let fs = fs.clone();
+      basic(fs)
+    });
+  });
 }
 
 criterion_group!(benches, criterion_benchmark);
