feat: support tree shaking

crates/rspack_core/src/build_chunk_graph/code_splitter.rs

@@ -659,7 +659,7 @@ Or do you want to use the entrypoints '{name}' and '{runtime}' independently on
       assign_depths(
         &mut assign_depths_map,
         &compilation.get_module_graph(),
-        modules.iter().collect(),
+        modules.iter(),
       );
       input_entrypoints_and_modules.insert(entry_point, modules);
     }
@@ -1133,7 +1133,8 @@ Or do you want to use the entrypoints '{name}' and '{runtime}' independently on
 
     self.stat_processed_blocks += 1;
 
-    let chunk_group_info = self.chunk_group_infos.expect_get(&item.chunk_group_info);
+    let chunk_group_info: &ChunkGroupInfo =
+      self.chunk_group_infos.expect_get(&item.chunk_group_info);
 
     let runtime = chunk_group_info.runtime.clone();
 
@@ -1538,7 +1539,11 @@ Or do you want to use the entrypoints '{name}' and '{runtime}' independently on
       return modules.clone();
     }
 
-    self.extract_block_modules(module.get_root_block(compilation), runtime, compilation);
+    self.extract_block_modules(
+      module.get_root_block(&compilation.get_module_graph()),
+      runtime,
+      compilation,
+    );
     self
       .block_modules_runtime_map
       .get::<OptionalRuntimeSpec>(&runtime.cloned().into())
@@ -1937,11 +1942,10 @@ pub(crate) enum DependenciesBlockIdentifier {
 }
 
 impl DependenciesBlockIdentifier {
-  pub fn get_root_block<'a>(&'a self, compilation: &'a Compilation) -> ModuleIdentifier {
+  pub fn get_root_block<'a>(&'a self, module_graph: &'a ModuleGraph) -> ModuleIdentifier {
     match self {
       DependenciesBlockIdentifier::Module(m) => *m,
-      DependenciesBlockIdentifier::AsyncDependenciesBlock(id) => *compilation
-        .get_module_graph()
+      DependenciesBlockIdentifier::AsyncDependenciesBlock(id) => *module_graph
         .block_by_id(id)
         .expect("should have block")
         .parent(),

crates/rspack_core/src/build_chunk_graph/incremental.rs

@@ -128,7 +128,7 @@ impl CodeSplitter {
         continue;
       };
 
-      parent_cg.children.remove(&chunk_group_ukey);
+      parent_cg.children.swap_remove_full(&chunk_group_ukey);
 
       if let Some(parent_cgi) = self.chunk_group_info_map.get(parent) {
         if let Some(parent_cgi) = self.chunk_group_infos.get_mut(parent_cgi) {

crates/rspack_core/src/build_chunk_graph/mod.rs

@@ -50,6 +50,6 @@ pub(crate) fn build_chunk_graph(compilation: &mut Compilation) -> rspack_error::
 
 #[instrument(skip_all)]
 pub(crate) fn build_chunk_graph_new(compilation: &mut Compilation) -> rspack_error::Result<()> {
-  new_code_splitter::code_split(compilation);
+  new_code_splitter::code_split(compilation)?;
   Ok(())
 }