perf(es/codegen): Use scoped allocator (swc-project#9248)

**Description:** Main: <img width="711" alt="스크린샷 2024-07-15 오후 8 05 50" src="https://github.com/user-attachments/assets/508a29e8-744f-4017-a819-b7576e61d287"> new: <img width="711" alt="스크린샷 2024-07-15 오후 8 05 19" src="https://github.com/user-attachments/assets/bcf3107d-e843-4fab-8b8a-fc6376705273">
kdy1 · Jul 16, 2024 · 970f323 · 970f323
1 parent 5cc6449
commit 970f323
Show file tree

Hide file tree

Showing 16 changed files with 34 additions and 14 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/swc_compiler_base/Cargo.toml b/crates/swc_compiler_base/Cargo.toml
@@ -21,6 +21,7 @@ rustc-hash = { workspace = true }
 serde      = { workspace = true, features = ["derive"] }
 serde_json = { workspace = true }
 sourcemap  = { workspace = true }
+swc_allocator = { version = "0.1.5", path = "../swc_allocator", default-features = false }
 
 swc_atoms = { version = "0.6.5", path = "../swc_atoms" }
 swc_common = { version = "0.35.0", path = "../swc_common", features = [

diff --git a/crates/swc_compiler_base/src/lib.rs b/crates/swc_compiler_base/src/lib.rs
@@ -171,10 +171,10 @@ where
 {
     let _timer = timer!("Compiler::print");
 
-    let mut src_map_buf = vec![];
+    let mut src_map_buf = swc_allocator::vec::Vec::new();
 
     let src = {
-        let mut buf = vec![];
+        let mut buf = Vec::new();
         {
             let mut w = swc_ecma_codegen::text_writer::JsWriter::new(
                 cm.clone(),

diff --git a/crates/swc_ecma_ast/src/stmt.rs b/crates/swc_ecma_ast/src/stmt.rs
@@ -164,7 +164,7 @@ impl Stmt {
     }
 }
 
-// Memory layout depedns on the version of rustc.
+// Memory layout depends on the version of rustc.
 // #[cfg(target_pointer_width = "64")]
 // assert_eq_size!(Stmt, [u8; 56]);
 

diff --git a/crates/swc_ecma_codegen/Cargo.toml b/crates/swc_ecma_codegen/Cargo.toml
@@ -25,6 +25,7 @@ serde      = { workspace = true }
 sourcemap  = { workspace = true }
 tracing    = { workspace = true }
 
+swc_allocator           = { version = "0.1.5", path = "../swc_allocator" }
 swc_atoms               = { version = "0.6.5", path = "../swc_atoms" }
 swc_common              = { version = "0.35.0", path = "../swc_common" }
 swc_ecma_ast            = { version = "0.116.0", path = "../swc_ecma_ast" }

diff --git a/crates/swc_ecma_codegen/benches/bench.rs b/crates/swc_ecma_codegen/benches/bench.rs
@@ -1,6 +1,7 @@
 extern crate swc_malloc;
 
 use codspeed_criterion_compat::{black_box, criterion_group, criterion_main, Bencher, Criterion};
+use swc_allocator::{vec::Vec, Allocator};
 use swc_common::FileName;
 use swc_ecma_codegen::Emitter;
 use swc_ecma_parser::{Parser, StringInput, Syntax};
@@ -84,7 +85,6 @@ fn bench_emitter(b: &mut Bencher, s: &str) {
         let fm = cm.new_source_file(FileName::Anon.into(), s.into());
         let mut parser = Parser::new(Syntax::default(), StringInput::from(&*fm), None);
 
-        let mut src_map_buf = vec![];
         let module = parser
             .parse_module()
             .map_err(|e| e.into_diagnostic(handler).emit())
@@ -95,7 +95,11 @@ fn bench_emitter(b: &mut Bencher, s: &str) {
         }
 
         b.iter(|| {
-            let mut buf = vec![];
+            let alloc = Allocator::default();
+            let _guard = unsafe { alloc.guard() };
+            let mut src_map_buf = Vec::new();
+
+            let mut buf = Vec::new();
             {
                 let mut emitter = Emitter {
                     cfg: Default::default(),

diff --git a/crates/swc_ecma_codegen/benches/with_parse.rs b/crates/swc_ecma_codegen/benches/with_parse.rs
@@ -1,6 +1,7 @@
 extern crate swc_malloc;
 
 use codspeed_criterion_compat::{black_box, criterion_group, criterion_main, Bencher, Criterion};
+use swc_allocator::vec::Vec;
 use swc_common::FileName;
 use swc_ecma_codegen::Emitter;
 use swc_ecma_parser::{Parser, StringInput, Syntax};
@@ -84,7 +85,7 @@ fn bench_emitter(b: &mut Bencher, s: &str) {
         b.iter(|| {
             let fm = cm.new_source_file(FileName::Anon.into(), s.into());
             let mut parser = Parser::new(Syntax::default(), StringInput::from(&*fm), None);
-            let mut src_map_buf = vec![];
+            let mut src_map_buf = Vec::new();
             let module = parser
                 .parse_module()
                 .map_err(|e| e.into_diagnostic(handler).emit())

diff --git a/crates/swc_ecma_codegen/examples/sourcemap.rs b/crates/swc_ecma_codegen/examples/sourcemap.rs
@@ -9,6 +9,7 @@ use std::{
     time::Instant,
 };
 
+use swc_allocator::vec::Vec;
 use swc_common::input::SourceFileInput;
 use swc_ecma_ast::*;
 use swc_ecma_codegen::{text_writer::JsWriter, Emitter};
@@ -30,7 +31,7 @@ fn parse_and_gen(entry: &Path) {
             .expect("failed to parse input as a module");
 
         let mut code = vec![];
-        let mut srcmap = vec![];
+        let mut srcmap = Vec::new();
 
         {
             let mut emitter = Emitter {
@@ -56,7 +57,7 @@ fn parse_and_gen(entry: &Path) {
     .expect("failed to process a module");
 }
 
-/// Usage: ./scripts/instruements path/to/input/file
+/// Usage: ./scripts/instruments path/to/input/file
 fn main() {
     let main_file = env::args().nth(1).unwrap();
 

diff --git a/crates/swc_ecma_codegen/src/tests.rs b/crates/swc_ecma_codegen/src/tests.rs
@@ -1,5 +1,6 @@
 use std::path::PathBuf;
 
+use swc_allocator::vec::Vec;
 use swc_common::{comments::SingleThreadedComments, FileName, SourceMap};
 use swc_ecma_parser;
 use swc_ecma_testing::{exec_node_js, JsExecOptions};
@@ -44,11 +45,11 @@ impl Builder {
     where
         F: for<'aa> FnOnce(&mut Emitter<'aa, Box<(dyn WriteJs + 'aa)>, SourceMap>),
     {
-        let mut buf = vec![];
+        let mut buf = Vec::new();
 
         self.with(src, &mut buf, op);
 
-        String::from_utf8(buf).unwrap()
+        String::from_utf8_lossy(&buf).into_owned()
     }
 }
 

diff --git a/crates/swc_ecma_codegen/src/text_writer/basic_impl.rs b/crates/swc_ecma_codegen/src/text_writer/basic_impl.rs
@@ -1,6 +1,7 @@
 use std::io::Write;
 
 use rustc_hash::FxHashSet;
+use swc_allocator::vec::Vec;
 use swc_common::{sync::Lrc, BytePos, LineCol, SourceMap, Span};
 
 use super::{Result, WriteJs};

diff --git a/crates/swc_ecma_codegen/tests/sourcemap.rs b/crates/swc_ecma_codegen/tests/sourcemap.rs
@@ -3,6 +3,7 @@ use std::{fs::read_to_string, path::PathBuf};
 use base64::prelude::{Engine, BASE64_STANDARD};
 use rustc_hash::FxHashSet;
 use sourcemap::SourceMap;
+use swc_allocator::vec::Vec;
 use swc_common::{comments::SingleThreadedComments, source_map::SourceMapGenConfig};
 use swc_ecma_ast::EsVersion;
 use swc_ecma_codegen::{text_writer::WriteJs, Emitter};
@@ -315,7 +316,7 @@ fn identity(entry: PathBuf) {
             Some(&comments),
         );
         let mut parser: Parser<Lexer> = Parser::new_from(lexer);
-        let mut src_map = vec![];
+        let mut src_map = Vec::new();
 
         {
             let mut wr = Box::new(swc_ecma_codegen::text_writer::JsWriter::new(

diff --git a/crates/swc_ecma_transforms_base/Cargo.toml b/crates/swc_ecma_transforms_base/Cargo.toml
@@ -34,6 +34,7 @@ swc_ecma_ast    = { version = "0.116.0", path = "../swc_ecma_ast" }
 swc_ecma_parser = { version = "0.147.0", path = "../swc_ecma_parser" }
 swc_ecma_utils  = { version = "0.131.0", path = "../swc_ecma_utils" }
 swc_ecma_visit  = { version = "0.102.0", path = "../swc_ecma_visit" }
+swc_allocator = { version = "0.1.5", path = "../swc_allocator", default-features = false }
 
 [dev-dependencies]
 codspeed-criterion-compat = { workspace = true }

diff --git a/crates/swc_ecma_transforms_react/Cargo.toml b/crates/swc_ecma_transforms_react/Cargo.toml
@@ -27,6 +27,7 @@ serde     = { workspace = true, features = ["derive"], optional = true }
 sha1      = { workspace = true }
 
 string_enum                = { version = "0.4.4", path = "../string_enum" }
+swc_allocator = { version = "0.1.5", path = "../swc_allocator", default-features = false }
 swc_atoms                  = { version = "0.6.5", path = "../swc_atoms" }
 swc_common                 = { version = "0.35.0", path = "../swc_common" }
 swc_config                 = { version = "0.1.13", path = "../swc_config" }

diff --git a/crates/swc_ecma_transforms_react/src/pure_annotations/tests.rs b/crates/swc_ecma_transforms_react/src/pure_annotations/tests.rs
@@ -1,3 +1,4 @@
+use swc_allocator::vec::Vec;
 use swc_common::{comments::SingleThreadedComments, sync::Lrc, FileName, Mark, SourceMap};
 use swc_ecma_codegen::{text_writer::JsWriter, Emitter};
 use swc_ecma_parser::{Parser, StringInput};
@@ -40,8 +41,8 @@ fn emit(
     comments: Lrc<SingleThreadedComments>,
     program: &Module,
 ) -> String {
-    let mut src_map_buf = vec![];
-    let mut buf = vec![];
+    let mut src_map_buf = Vec::new();
+    let mut buf = std::vec::Vec::new();
     {
         let writer = Box::new(JsWriter::new(
             source_map.clone(),

diff --git a/crates/swc_fast_ts_strip/Cargo.toml b/crates/swc_fast_ts_strip/Cargo.toml
@@ -13,6 +13,7 @@ version       = "0.2.2"
 [dependencies]
 anyhow = { workspace = true }
 serde  = { workspace = true, features = ["derive"] }
+swc_allocator = { version = "0.1.5", path = "../swc_allocator", default-features = false }
 
 swc_common = { version = "0.35.0", path = "../swc_common", features = [
   "sourcemap",

diff --git a/crates/swc_fast_ts_strip/src/lib.rs b/crates/swc_fast_ts_strip/src/lib.rs
@@ -241,7 +241,7 @@ pub fn operate(
 
             let mut src = vec![];
             let mut src_map_buf = if options.source_map {
-                Some(vec![])
+                Some(swc_allocator::vec::Vec::new())
             } else {
                 None
             };