perf: avoid create new dependency for add include

Author: ahabhgk

crates/node_binding/src/compilation/mod.rs

@@ -22,6 +22,7 @@ use rspack_napi::napi::bindgen_prelude::*;
 use rspack_napi::NapiResultExt;
 use rspack_napi::OneShotRef;
 use rspack_plugin_runtime::RuntimeModuleFromJs;
+use rustc_hash::FxHashMap;
 
 use super::{JsFilename, PathWithInfo};
 use crate::entry::JsEntryOptions;
@@ -38,6 +39,7 @@ use crate::JsStatsOptimizationBailout;
 use crate::LocalJsFilename;
 use crate::RawDependency;
 use crate::ToJsCompatSource;
+use crate::COMPILER_REFERENCES;
 use crate::{AssetInfo, JsAsset, JsPathData, JsStats};
 use crate::{JsRspackDiagnostic, JsRspackError};
 
@@ -700,23 +702,52 @@ impl JsCompilation {
   ) -> napi::Result<()> {
     let compilation = self.as_mut()?;
 
+    let Some(compiler_reference) = COMPILER_REFERENCES.with(|ref_cell| {
+      let references = ref_cell.borrow();
+      references.get(&compilation.compiler_id()).cloned()
+    }) else {
+      return Err(napi::Error::from_reason(
+        "Unable to addInclude now. The Compiler has been garbage collected by JavaScript.",
+      ));
+    };
+    let Some(js_compiler) = compiler_reference.get() else {
+      return Err(napi::Error::from_reason(
+        "Unable to addInclude now. The Compiler has been garbage collected by JavaScript.",
+      ));
+    };
+    let include_dependencies_map = &js_compiler.include_dependencies_map;
+
     let args = js_args
       .into_iter()
       .map(|(js_context, js_dependency, js_options)| {
         let layer = match &js_options {
           Some(options) => options.layer.clone(),
           None => None,
         };
-        let dependency = Box::new(EntryDependency::new(
-          js_dependency.request,
-          js_context.into(),
-          layer,
-          false,
-        )) as BoxDependency;
         let options = match js_options {
           Some(js_opts) => js_opts.into(),
           None => EntryOptions::default(),
         };
+        let dependency = if let Some(map) = include_dependencies_map.get(&js_dependency.request)
+          && let Some(dependency) = map.get(&options)
+        {
+          dependency.clone()
+        } else {
+          let dependency: BoxDependency = Box::new(EntryDependency::new(
+            js_dependency.request.clone(),
+            js_context.into(),
+            layer,
+            false,
+          ));
+          if let Some(mut map) = include_dependencies_map.get_mut(&js_dependency.request) {
+            map.insert(options.clone(), dependency.clone());
+          } else {
+            let mut map = FxHashMap::default();
+            map.insert(options.clone(), dependency.clone());
+            include_dependencies_map.insert(js_dependency.request, map);
+          }
+          dependency
+        };
         (dependency, options)
       })
       .collect::<Vec<(BoxDependency, EntryOptions)>>();

crates/node_binding/src/lib.rs

@@ -13,7 +13,9 @@ use std::sync::{Arc, Mutex};
 use compiler::{Compiler, CompilerState, CompilerStateGuard};
 use napi::{bindgen_prelude::*, CallContext};
 use rspack_collections::UkeyMap;
-use rspack_core::{Compilation, CompilerId, ModuleIdentifier, PluginExt};
+use rspack_core::{
+  BoxDependency, Compilation, CompilerId, EntryOptions, ModuleIdentifier, PluginExt,
+};
 use rspack_error::Diagnostic;
 use rspack_fs::IntermediateFileSystem;
 use rspack_fs_node::{NodeFileSystem, ThreadsafeNodeFS};
@@ -85,7 +87,9 @@ use resolver_factory::*;
 pub use resource_data::*;
 pub use rsdoctor::*;
 use rspack_tracing::{ChromeTracer, OtelTracer, StdoutTracer, Tracer};
+use rspack_util::fx_hash::FxDashMap;
 pub use runtime::*;
+use rustc_hash::FxHashMap;
 pub use source::*;
 pub use stats::*;
 use swc_core::common::util::take::Take;
@@ -112,6 +116,7 @@ pub struct JsCompiler {
   js_hooks_plugin: JsHooksAdapterPlugin,
   compiler: Pin<Box<Compiler>>,
   state: CompilerState,
+  include_dependencies_map: FxDashMap<String, FxHashMap<EntryOptions, BoxDependency>>,
 }
 
 #[napi]
@@ -187,6 +192,7 @@ impl JsCompiler {
       compiler: Box::pin(Compiler::from(rspack)),
       state: CompilerState::init(),
       js_hooks_plugin,
+      include_dependencies_map: Default::default(),
     })
   }