Limit parallelism when loading Wasm modules

ocaml-wasm · May 2, 2024 · 0502d7d · 0502d7d
1 parent 136bd6d
commit 0502d7d
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/runtime/wasm/runtime.js b/runtime/wasm/runtime.js
@@ -366,18 +366,28 @@
     async function instantiateFromDir() {
       imports.OCaml = {};
       const deps = []
-      for (const module of link) {
+      async function loadModule(module, isRuntime) {
         const sync = module[1].constructor !== Array
         async function instantiate () {
           const code = loadCode(src + "/" + module[0] + ".wasm")
           await Promise.all(sync?deps:module[1].map((i)=>deps[i]));
           const wasmModule = await instantiateModule(code)
-          Object.assign(deps.length?imports.OCaml:imports.env,
+          Object.assign(isRuntime?imports.env:imports.OCaml,
                         wasmModule.instance.exports);
         }
-        deps.push(sync?await instantiate():instantiate())
+        const promise = instantiate();
+        deps.push(promise);
+        return promise;
       }
-      await deps.pop();
+      async function loadModules(lst) {
+        for (const module of lst) {
+          await loadModule(module);
+        }
+      }
+      await loadModule(link[0], 1);
+      await loadModule(link[1]);
+      const workers = new Array(20).fill(link.slice(2).values()).map(loadModules);
+      await Promise.all(workers);
       return {instance:{exports: Object.assign(imports.env, imports.OCaml)}}
     }
     const wasmModule =