feat: add chunk matcher for mf

umijs · Feb 20, 2025 · fd67d72 · fd67d72
1 parent 666455c
commit fd67d72
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/crates/mako/src/generate/chunk_pot/util.rs b/crates/mako/src/generate/chunk_pot/util.rs
@@ -105,6 +105,15 @@ pub(crate) fn runtime_code(context: &Arc<Context>) -> Result<String> {
     let chunk_graph = context.chunk_graph.read().unwrap();
     let has_dynamic_chunks = chunk_graph.get_all_chunks().len() > 1;
     let has_hmr = context.args.watch;
+    let chunk_matcher = context.config.module_federation.as_ref().and_then(|mf| {
+        mf.remotes.as_ref().and_then(|remotes| {
+            if remotes.is_empty() {
+                None
+            } else {
+                Some(r#"/^mako\/container\/remote\//"#.to_string())
+            }
+        })
+    });
     let app_runtime = AppRuntimeTemplate {
         has_dynamic_chunks,
         has_hmr,
@@ -127,6 +136,7 @@ pub(crate) fn runtime_code(context: &Arc<Context>) -> Result<String> {
             .as_ref()
             .map_or(false, |o| o.concatenate_modules.unwrap_or(false)),
         global_module_registry: context.config.output.global_module_registry,
+        chunk_matcher,
     };
     let app_runtime = app_runtime.render_once()?;
     let app_runtime = app_runtime.replace(

diff --git a/crates/mako/src/generate/runtime.rs b/crates/mako/src/generate/runtime.rs
@@ -14,4 +14,5 @@ pub struct AppRuntimeTemplate {
     pub concatenate_enabled: bool,
     pub cross_origin_loading: Option<String>,
     pub global_module_registry: bool,
+    pub chunk_matcher: Option<String>,
 }
diff --git a/crates/mako/templates/app_runtime.stpl b/crates/mako/templates/app_runtime.stpl
@@ -71,7 +71,7 @@ function createRuntime(makoModules, entryModuleId, global) {
   // required modules
   requireModule.m  = makoModules;
 
-  // required modules
+  // modules registry
   requireModule.c  = modulesRegistry;
 
 <% if concatenate_enabled { %>
@@ -136,7 +136,10 @@ function createRuntime(makoModules, entryModuleId, global) {
       if (data === 0) return;
 
       // skip federation remote chunk
-      if (chunkId.startsWith("mako/container/remote/")) return;
+
+      <% if chunk_matcher.is_some() { %>
+      if (<%= chunk_matcher.unwrap() %>.test(chunkId)) return
+      <% } %>
 
       if (data) {
         //     0       1        2