perf(bootstrap): Implement using getProjectVersion() in @ts-morph/boo…

…tstrap.
dsherret · Nov 21, 2020 · 646f797 · 646f797
1 parent 7c524ec
commit 646f797
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 3 deletions.
diff --git a/packages/bootstrap/src/Project.ts b/packages/bootstrap/src/Project.ts
@@ -151,6 +151,7 @@ export class Project {
             compilerOptions: this.compilerOptions,
             getNewLine: () => newLineKind,
             resolutionHost: resolutionHost || {},
+            getProjectVersion: () => this._sourceFileCache.getProjectVersion().toString(),
         });
         this.languageServiceHost = languageServiceHost;
         this.compilerHost = compilerHost;

diff --git a/packages/bootstrap/src/SourceFileCache.ts b/packages/bootstrap/src/SourceFileCache.ts
@@ -3,6 +3,7 @@ import { DocumentRegistry, ScriptKind, ts, StringUtils, TsSourceFileContainer, F
 
 export class SourceFileCache implements TsSourceFileContainer {
     private readonly sourceFilesByFilePath = new Map<StandardizedFilePath, ts.SourceFile>();
+    private projectVersion = 0;
 
     readonly documentRegistry: DocumentRegistry;
 
@@ -25,6 +26,10 @@ export class SourceFileCache implements TsSourceFileContainer {
         return this.sourceFilesByFilePath.values();
     }
 
+    getProjectVersion() {
+        return this.projectVersion;
+    }
+
     getSourceFileVersion(sourceFile: ts.SourceFile) {
         return this.documentRegistry.getSourceFileVersion(sourceFile);
     }
@@ -91,6 +96,7 @@ export class SourceFileCache implements TsSourceFileContainer {
             this.fileSystemWrapper.queueMkdir(dirPath);
 
         this.sourceFilesByFilePath.set(standardizedFilePath, sourceFile);
+        this.projectVersion++;
     }
 
     removeSourceFile(filePath: StandardizedFilePath) {

diff --git a/packages/common/lib/ts-morph-common.d.ts b/packages/common/lib/ts-morph-common.d.ts
@@ -108,7 +108,8 @@ export interface CreateHostsOptions {
     getNewLine: () => "\r\n" | "\n";
     /** The resolution host used for resolving modules and type reference directives. */
     resolutionHost: ResolutionHost;
-    /** Provides the current project version to be used for up-to-date checks.  */
+    /** Provides the current project version to be used to tell if source files have
+     * changed. Provide this for a performance improvement. */
     getProjectVersion?: () => string;
 }
 

diff --git a/packages/common/src/compiler/createHosts.ts b/packages/common/src/compiler/createHosts.ts
@@ -18,7 +18,8 @@ export interface CreateHostsOptions {
     getNewLine: () => "\r\n" | "\n";
     /** The resolution host used for resolving modules and type reference directives. */
     resolutionHost: ResolutionHost;
-    /** Provides the current project version to be used for up-to-date checks.  */
+    /** Provides the current project version to be used to tell if source files have
+     * changed. Provide this for a performance improvement. */
     getProjectVersion?: () => string;
 }
 

diff --git a/packages/ts-morph/src/compiler/tools/LanguageService.ts b/packages/ts-morph/src/compiler/tools/LanguageService.ts
@@ -27,7 +27,7 @@ export class LanguageService {
     /** @internal */
     private _context: ProjectContext;
     /** @internal */
-    private _projectVersion: number  = 0;
+    private _projectVersion = 0;
 
     /**
      * Gets the compiler language service.