Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
perf(ngcc): introduce cache for sharing data across entry-points (#38840
) ngcc creates typically two `ts.Program` instances for each entry-point, one for processing sources and another one for processing the typings. The creation of these programs is somewhat expensive, as it concerns module resolution and parsing of source files. This commit implements several layers of caching to optimize the creation of programs: 1. A shared module resolution cache across all entry-points within a single invocation of ngcc. Both the sources and typings program benefit from this cache. 2. Sharing the parsed `ts.SourceFile` for a single entry-point between the sources and typings program. 3. Sharing parsed `ts.SourceFile`s of TypeScript's default libraries across all entry-points within a single invocation. Some of these default library typings are large and therefore expensive to parse, so sharing the parsed source files across all entry-points offers a significant performance improvement. Using a bare CLI app created using `ng new` + `ng add @angular/material`, the above changes offer a 3-4x improvement in ngcc's processing time when running synchronously and ~2x improvement for asynchronous runs. PR Close #38840
- Loading branch information
1 parent
84bd1a2
commit 58411e7
Showing
8 changed files
with
509 additions
and
20 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
197 changes: 197 additions & 0 deletions
197
packages/compiler-cli/ngcc/src/packages/source_file_cache.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,197 @@ | ||
/** | ||
* @license | ||
* Copyright Google LLC All Rights Reserved. | ||
* | ||
* Use of this source code is governed by an MIT-style license that can be | ||
* found in the LICENSE file at https://angular.io/license | ||
*/ | ||
import * as ts from 'typescript'; | ||
import {AbsoluteFsPath, FileSystem} from '../../../src/ngtsc/file_system'; | ||
|
||
/** | ||
* A cache that holds on to source files that can be shared for processing all entry-points in a | ||
* single invocation of ngcc. In particular, the following files are shared across all entry-points | ||
* through this cache: | ||
* | ||
* 1. Default library files such as `lib.dom.d.ts` and `lib.es5.d.ts`. These files don't change | ||
* and some are very large, so parsing is expensive. Therefore, the parsed `ts.SourceFile`s for | ||
* the default library files are cached. | ||
* 2. The typings of @angular scoped packages. The typing files for @angular packages are typically | ||
* used in the entry-points that ngcc processes, so benefit from a single source file cache. | ||
* Especially `@angular/core/core.d.ts` is large and expensive to parse repeatedly. In contrast | ||
* to default library files, we have to account for these files to be invalidated during a single | ||
* invocation of ngcc, as ngcc will overwrite the .d.ts files during its processing. | ||
* | ||
* The lifecycle of this cache corresponds with a single invocation of ngcc. Separate invocations, | ||
* e.g. the CLI's synchronous module resolution fallback will therefore all have their own cache. | ||
* This allows for the source file cache to be garbage collected once ngcc processing has completed. | ||
*/ | ||
export class SharedFileCache { | ||
private sfCache = new Map<AbsoluteFsPath, ts.SourceFile>(); | ||
|
||
constructor(private fs: FileSystem) {} | ||
|
||
/** | ||
* Loads a `ts.SourceFile` if the provided `fileName` is deemed appropriate to be cached. To | ||
* optimize for memory usage, only files that are generally used in all entry-points are cached. | ||
* If `fileName` is not considered to benefit from caching or the requested file does not exist, | ||
* then `undefined` is returned. | ||
*/ | ||
getCachedSourceFile(fileName: string): ts.SourceFile|undefined { | ||
const absPath = this.fs.resolve(fileName); | ||
if (isDefaultLibrary(absPath, this.fs)) { | ||
return this.getStableCachedFile(absPath); | ||
} else if (isAngularDts(absPath, this.fs)) { | ||
return this.getVolatileCachedFile(absPath); | ||
} else { | ||
return undefined; | ||
} | ||
} | ||
|
||
/** | ||
* Attempts to load the source file from the cache, or parses the file into a `ts.SourceFile` if | ||
* it's not yet cached. This method assumes that the file will not be modified for the duration | ||
* that this cache is valid for. If that assumption does not hold, the `getVolatileCachedFile` | ||
* method is to be used instead. | ||
*/ | ||
private getStableCachedFile(absPath: AbsoluteFsPath): ts.SourceFile|undefined { | ||
if (!this.sfCache.has(absPath)) { | ||
const content = readFile(absPath, this.fs); | ||
if (content === undefined) { | ||
return undefined; | ||
} | ||
const sf = ts.createSourceFile(absPath, content, ts.ScriptTarget.ES2015); | ||
this.sfCache.set(absPath, sf); | ||
} | ||
return this.sfCache.get(absPath)!; | ||
} | ||
|
||
/** | ||
* In contrast to `getStableCachedFile`, this method always verifies that the cached source file | ||
* is the same as what's stored on disk. This is done for files that are expected to change during | ||
* ngcc's processing, such as @angular scoped packages for which the .d.ts files are overwritten | ||
* by ngcc. If the contents on disk have changed compared to a previously cached source file, the | ||
* content from disk is re-parsed and the cache entry is replaced. | ||
*/ | ||
private getVolatileCachedFile(absPath: AbsoluteFsPath): ts.SourceFile|undefined { | ||
const content = readFile(absPath, this.fs); | ||
if (content === undefined) { | ||
return undefined; | ||
} | ||
if (!this.sfCache.has(absPath) || this.sfCache.get(absPath)!.text !== content) { | ||
const sf = ts.createSourceFile(absPath, content, ts.ScriptTarget.ES2015); | ||
this.sfCache.set(absPath, sf); | ||
} | ||
return this.sfCache.get(absPath)!; | ||
} | ||
} | ||
|
||
const DEFAULT_LIB_PATTERN = ['node_modules', 'typescript', 'lib', /^lib\..+\.d\.ts$/]; | ||
|
||
/** | ||
* Determines whether the provided path corresponds with a default library file inside of the | ||
* typescript package. | ||
* | ||
* @param absPath The path for which to determine if it corresponds with a default library file. | ||
* @param fs The filesystem to use for inspecting the path. | ||
*/ | ||
export function isDefaultLibrary(absPath: AbsoluteFsPath, fs: FileSystem): boolean { | ||
return isFile(absPath, DEFAULT_LIB_PATTERN, fs); | ||
} | ||
|
||
const ANGULAR_DTS_PATTERN = ['node_modules', '@angular', /./, /\.d\.ts$/]; | ||
|
||
/** | ||
* Determines whether the provided path corresponds with a .d.ts file inside of an @angular | ||
* scoped package. This logic only accounts for the .d.ts files in the root, which is sufficient | ||
* to find the large, flattened entry-point files that benefit from caching. | ||
* | ||
* @param absPath The path for which to determine if it corresponds with an @angular .d.ts file. | ||
* @param fs The filesystem to use for inspecting the path. | ||
*/ | ||
export function isAngularDts(absPath: AbsoluteFsPath, fs: FileSystem): boolean { | ||
return isFile(absPath, ANGULAR_DTS_PATTERN, fs); | ||
} | ||
|
||
/** | ||
* Helper function to determine whether a file corresponds with a given pattern of segments. | ||
* | ||
* @param path The path for which to determine if it corresponds with the provided segments. | ||
* @param segments Array of segments; the `path` must have ending segments that match the | ||
* patterns in this array. | ||
* @param fs The filesystem to use for inspecting the path. | ||
*/ | ||
function isFile( | ||
path: AbsoluteFsPath, segments: ReadonlyArray<string|RegExp>, fs: FileSystem): boolean { | ||
for (let i = segments.length - 1; i >= 0; i--) { | ||
const pattern = segments[i]; | ||
const segment = fs.basename(path); | ||
if (typeof pattern === 'string') { | ||
if (pattern !== segment) { | ||
return false; | ||
} | ||
} else { | ||
if (!pattern.test(segment)) { | ||
return false; | ||
} | ||
} | ||
path = fs.dirname(path); | ||
} | ||
return true; | ||
} | ||
|
||
/** | ||
* A cache for processing a single entry-point. This exists to share `ts.SourceFile`s between the | ||
* source and typing programs that are created for a single program. | ||
*/ | ||
export class EntryPointFileCache { | ||
private readonly sfCache = new Map<AbsoluteFsPath, ts.SourceFile>(); | ||
|
||
constructor(private fs: FileSystem, private sharedFileCache: SharedFileCache) {} | ||
|
||
/** | ||
* Returns and caches a parsed `ts.SourceFile` for the provided `fileName`. If the `fileName` is | ||
* cached in the shared file cache, that result is used. Otherwise, the source file is cached | ||
* internally. This method returns `undefined` if the requested file does not exist. | ||
* | ||
* @param fileName The path of the file to retrieve a source file for. | ||
* @param languageVersion The language version to use for parsing the file. | ||
*/ | ||
getCachedSourceFile(fileName: string, languageVersion: ts.ScriptTarget): ts.SourceFile|undefined { | ||
const staticSf = this.sharedFileCache.getCachedSourceFile(fileName); | ||
if (staticSf !== undefined) { | ||
return staticSf; | ||
} | ||
|
||
const absPath = this.fs.resolve(fileName); | ||
if (this.sfCache.has(absPath)) { | ||
return this.sfCache.get(absPath); | ||
} | ||
|
||
const content = readFile(absPath, this.fs); | ||
if (content === undefined) { | ||
return undefined; | ||
} | ||
const sf = ts.createSourceFile(fileName, content, languageVersion); | ||
this.sfCache.set(absPath, sf); | ||
return sf; | ||
} | ||
} | ||
|
||
function readFile(absPath: AbsoluteFsPath, fs: FileSystem): string|undefined { | ||
if (!fs.exists(absPath) || !fs.stat(absPath).isFile()) { | ||
return undefined; | ||
} | ||
return fs.readFile(absPath); | ||
} | ||
|
||
/** | ||
* Creates a `ts.ModuleResolutionCache` that uses the provided filesystem for path operations. | ||
* | ||
* @param fs The filesystem to use for path operations. | ||
*/ | ||
export function createModuleResolutionCache(fs: FileSystem): ts.ModuleResolutionCache { | ||
return ts.createModuleResolutionCache(fs.pwd(), fileName => { | ||
return fs.isCaseSensitive() ? fileName : fileName.toLowerCase(); | ||
}); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.