Clean up memoization when done

andrewbranch · andrewbranch · commit 9cdad91f805e · 2019-07-22T14:47:17.000-07:00
diff --git a/src/compiler/core.ts b/src/compiler/core.ts
@@ -1634,10 +1634,16 @@ namespace ts {
         };
     }
 
-    export function memoizeOne<ArgsT extends unknown[], ReturnT>(callback: (...args: ArgsT) => ReturnT) {
+    export function memoizeOne<ArgsT extends unknown[], ReturnT>(callback: (...args: ArgsT) => ReturnT): typeof callback & { clear: () => void } {
         let value: ReturnT;
         let cachedArgs: ArgsT;
-        return (...args: ArgsT) => {
+        runMemoized.clear = () => {
+            value = undefined!;
+            cachedArgs = undefined!;
+        };
+        return runMemoized;
+
+        function runMemoized(...args: ArgsT) {
             const length = args.length;
             if (cachedArgs && cachedArgs.length === length) {
                 for (let i = 0; i < length; i++) {
@@ -1653,7 +1659,6 @@ namespace ts {
         };
     }
 
-
     /**
      * High-order function, composes functions. Note that functions are composed inside-out;
      * for example, `compose(a, b)` is the equivalent of `x => b(a(x))`.
diff --git a/src/compiler/moduleSpecifiers.ts b/src/compiler/moduleSpecifiers.ts
@@ -185,6 +185,10 @@ namespace ts.moduleSpecifiers {
         return result;
     });
 
+    export function clearSymlinksCache() {
+        discoverProbableSymlinks.clear();
+    }
+
     /**
      * Looks for existing imports that use symlinks to this module.
      * Symlinks will be returned first so they are preferred over the real path.
diff --git a/src/services/codefixes/importFixes.ts b/src/services/codefixes/importFixes.ts
@@ -591,6 +591,7 @@ namespace ts.codefix {
                 }
             }
         });
+        moduleSpecifiers.clearSymlinksCache();
 
         function getNodeModulesPackageNameFromFileName(importedFileName: string): string | undefined {
             if (!stringContains(importedFileName, "node_modules")) {

Original file line number	Diff line number	Diff line change
`@@ -591,6 +591,7 @@ namespace ts.codefix {`
`591`	`591`	`}`
`592`	`592`	`}`
`593`	`593`	`});`
	`594`	`+ moduleSpecifiers.clearSymlinksCache();`
`594`	`595`
`595`	`596`	`function getNodeModulesPackageNameFromFileName(importedFileName: string): string \| undefined {`
`596`	`597`	`if (!stringContains(importedFileName, "node_modules")) {`