refactor: code splitting

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);
     }
@@ -1535,7 +1535,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())
@@ -1928,11 +1932,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

@@ -129,7 +129,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

@@ -7,6 +7,7 @@ use crate::{incremental::IncrementalPasses, Compilation};
 
 pub(crate) mod code_splitter;
 pub(crate) mod incremental;
+pub(crate) mod new_code_splitter;
 
 #[instrument(skip_all)]
 pub(crate) fn build_chunk_graph(compilation: &mut Compilation) -> rspack_error::Result<()> {
@@ -46,3 +47,9 @@ pub(crate) fn build_chunk_graph(compilation: &mut Compilation) -> rspack_error::
 
   Ok(())
 }
+
+#[instrument(skip_all)]
+pub(crate) fn build_chunk_graph_new(compilation: &mut Compilation) -> rspack_error::Result<()> {
+  new_code_splitter::code_split(compilation)?;
+  Ok(())
+}