feat(wasm): support tracing-chrome in wasm (#10246)

* Writer

* Fix clippy

* Fix clippy

* cleanupGlobalTrace early

* Fix fmt

* Update writer api

* Fix fmt

Author: CPunisher

crates/rspack_tracing/src/chrome.rs

@@ -13,7 +13,7 @@ pub struct ChromeTracer {
 
 impl Tracer for ChromeTracer {
   fn setup(&mut self, output: &str) -> Option<Layered> {
-    let trace_writer = TraceWriter::from(output);
+    let trace_writer = TraceWriter::from(output.to_owned());
     let (chrome_layer, guard) = ChromeLayerBuilder::new()
       .trace_style(TraceStyle::Async)
       .include_args(true)

crates/rspack_tracing/src/lib.rs

@@ -2,31 +2,29 @@ mod chrome;
 mod stdout;
 mod tracer;
 
-use std::{fs, io, path::Path};
+use std::{fs, io, path::PathBuf};
 
 pub use chrome::ChromeTracer;
 pub use stdout::StdoutTracer;
 pub use tracer::Tracer;
 use tracing_subscriber::fmt::writer::BoxMakeWriter;
-pub(crate) enum TraceWriter<'a> {
+pub(crate) enum TraceWriter {
   Stdout,
   Stderr,
-  File { path: &'a Path },
+  File { path: PathBuf },
 }
 
-impl<'a> From<&'a str> for TraceWriter<'a> {
-  fn from(s: &'a str) -> Self {
-    match s {
+impl From<String> for TraceWriter {
+  fn from(s: String) -> Self {
+    match s.as_str() {
       "stdout" => Self::Stdout,
       "stderr" => Self::Stderr,
-      path => Self::File {
-        path: Path::new(path),
-      },
+      _ => Self::File { path: s.into() },
     }
   }
 }
 
-impl TraceWriter<'_> {
+impl TraceWriter {
   pub fn make_writer(&self) -> BoxMakeWriter {
     match self {
       TraceWriter::Stdout => BoxMakeWriter::new(io::stdout),

crates/rspack_tracing/src/stdout.rs

@@ -10,7 +10,7 @@ pub struct StdoutTracer;
 impl Tracer for StdoutTracer {
   fn setup(&mut self, output: &str) -> Option<Layered> {
     use tracing_subscriber::{fmt, prelude::*};
-    let trace_writer = TraceWriter::from(output);
+    let trace_writer = TraceWriter::from(output.to_owned());
     Some(
       fmt::layer()
         .pretty()

crates/rspack_tracing_chrome/src/lib.rs

@@ -52,7 +52,7 @@ pub struct ChromeLayerBuilder<S>
 where
   S: Subscriber + for<'span> LookupSpan<'span> + Send + Sync,
 {
-  out_writer: Option<Box<dyn Write + Send>>,
+  out_writer: Option<Box<dyn FnOnce() -> Box<dyn Write> + Send>>,
   name_fn: Option<NameFn<S>>,
   cat_fn: Option<NameFn<S>>,
   include_args: bool,
@@ -97,7 +97,7 @@ where
   ///
   /// If `file` could not be opened/created. To handle errors,
   /// open a file and pass it to [`writer`](crate::ChromeLayerBuilder::writer) instead.
-  pub fn file<P: AsRef<Path>>(self, file: P) -> Self {
+  pub fn file<P: AsRef<Path> + Send + 'static>(self, file: P) -> Self {
     self.writer(std::fs::File::create(file).expect("Failed to create trace file."))
   }
 
@@ -112,7 +112,7 @@ where
   /// # tracing_subscriber::registry().with(layer).init();
   /// ```
   pub fn writer<W: Write + Send + 'static>(mut self, writer: W) -> Self {
-    self.out_writer = Some(Box::new(writer));
+    self.out_writer = Some(Box::new(|| Box::new(writer)));
     self
   }
 
@@ -285,11 +285,10 @@ where
     let (tx, rx) = mpsc::channel();
     OUT.with(|val| val.replace(Some(tx.clone())));
 
-    let out_writer = builder
-      .out_writer
-      .unwrap_or_else(|| create_default_writer());
+    let out_writer = builder.out_writer;
 
     let handle = std::thread::spawn(move || {
+      let out_writer = out_writer.map_or_else(|| create_default_writer() as _, |f| f());
       let mut write = BufWriter::new(out_writer);
       write.write_all(b"[\n").unwrap();
 

packages/rspack/src/Compiler.ts

@@ -459,6 +459,7 @@ class Compiler {
 			this.hooks.afterDone.call(stats!);
 
 			instanceBinding.shutdownAsyncRuntime();
+			instanceBinding.cleanupGlobalTrace();
 			isRuntimeShutdown = true;
 		};