add revokeObjectURL

vitejs · May 20, 2021 · 57bdd4f · 57bdd4f
1 parent 662e909
commit 57bdd4f
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/packages/vite/src/node/plugins/worker.ts b/packages/vite/src/node/plugins/worker.ts
@@ -56,9 +56,14 @@ export function webWorkerPlugin(config: ResolvedConfig): Plugin {
               sourcemap: config.build.sourcemap
             })
 
-            return `export default function WorkerWrapper() {
-              const blob = new Blob([atob(\"${Buffer.from(output[0].code).toString('base64')}\")], { type: 'text/javascript;charset=utf-8' });
-              return new Worker((window.URL || window.webkitURL).createObjectURL(blob));
+            return `const blob = new Blob([atob(\"${Buffer.from(output[0].code).toString('base64')}\")], { type: 'text/javascript;charset=utf-8' });
+            export default function WorkerWrapper() {
+              const objURL = (window.URL || window.webkitURL).createObjectURL(blob);
+              try {
+                return new Worker(objURL);
+              } finally {
+                (window.URL || window.webkitURL).revokeObjectURL(objURL);
+              }
             }`
           } finally {
             await bundle.close()