feat: support reloadAll

NoCoreNode · Jul 31, 2020 · 955c445 · 955c445
1 parent a155600
commit 955c445
Showing 1 changed file with 47 additions and 22 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -56,6 +56,25 @@ export default class Reloader {
         this.updateFiles();
     }
 
+    reloadAll(newFileMap: IFileMap) {
+        const reloadModules = new Set<string>();
+        for (const [name] of Object.entries(this.fileMap)) {
+            const moduleId = require.resolve(join(this.context, name));
+            if (require.cache[moduleId]) {
+                reloadModules.add(moduleId);
+            }
+        }
+        const modulesToReload = Array.from(reloadModules);
+        if (modulesToReload.length > 0) {
+            this.del(modulesToReload);
+        }
+        this.updateFileMap(Object.assign(this.fileMap, newFileMap));
+        return {
+            reloadModules: modulesToReload.map(a => a.replace(this.context + sep, '')),
+            errors: [],
+        };
+    }
+
     reload(newFileMap: IFileMap) {
 
         const reloadModules = new Set<string>();
@@ -75,30 +94,23 @@ export default class Reloader {
 
         const modulesToReload = Array.from(reloadModules);
 
+        const errors: IError[] = [];
+
         if (modulesToReload.length > 0) {
-            // 删除缓存
-            batchdelcache(
-                modulesToReload,
-                true, this.commonRootPath
-            );
-
-            /* istanbul ignore next */
-            if (typeof global.gc === 'function') {
-                global.gc();
-            }
-        }
 
-        const errors: IError[] = [];
-        for (const mod of modulesToReload) {
-            try {
-                // eslint-disable-next-line @typescript-eslint/no-require-imports
-                require(mod);
-            }
-            catch (e) {
-                errors.push({
-                    file: mod,
-                    code: 1,
-                });
+            this.del(modulesToReload);
+
+            for (const mod of modulesToReload) {
+                try {
+                    // eslint-disable-next-line @typescript-eslint/no-require-imports
+                    require(mod);
+                }
+                catch (e) {
+                    errors.push({
+                        file: mod,
+                        code: 1,
+                    });
+                }
             }
         }
 
@@ -115,6 +127,19 @@ export default class Reloader {
         this.updateFiles();
     }
 
+    private del(modulesToReload: string[]) {
+        // 删除缓存
+        batchdelcache(
+            modulesToReload,
+            true, this.commonRootPath
+        );
+
+        /* istanbul ignore next */
+        if (typeof global.gc === 'function') {
+            global.gc();
+        }
+    }
+
     private getKey(item: IFileMapItem): string {
         if (typeof item === 'string') {
             return item;