hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-05-01 03:35:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

JS: Extract compiler-restarting into a function

Browse Source
This commit is contained in:
Asger Feldthaus
2020-03-05 11:51:43 +00:00
parent 3ae1aada37
commit e1657b237b
1 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
javascript/extractor/lib/typescript/src/main.ts
View File

@@ -86,6 +86,8 @@ class State {
}
let state = new State();
const reloadMemoryThresholdMb = getEnvironmentVariable("SEMMLE_TYPESCRIPT_MEMORY_THRESHOLD", Number, 1000);
/**
* Debugging method for finding cycles in the TypeScript AST. Should not be used in production.
*
@@ -529,26 +531,36 @@ function getEnvironmentVariable<T>(name: string, parse: (x: string) => T, defaul
return value != null ? parse(value) : defaultValue;
}
/**
* Whether the memory usage was last observed to be above the threshold for restarting the TypeScript compiler.
*/
let isAboveReloadThreshold = false;
/**
* If memory usage has moved above a the threshold, reboot the TypeScript compiler instance.
*/
function checkMemoryUsage() {
let bytesUsed = process.memoryUsage().heapUsed;
let megabytesUsed = bytesUsed / 1000000;
if (!isAboveReloadThreshold && megabytesUsed > reloadMemoryThresholdMb && state.project != null) {
console.warn('Restarting TypeScript compiler due to memory usage');
state.project.reload();
isAboveReloadThreshold = true;
}
else if (isAboveReloadThreshold && megabytesUsed < reloadMemoryThresholdMb) {
isAboveReloadThreshold = false;
}
}
function runReadLineInterface() {
reset();
let reloadMemoryThresholdMb = getEnvironmentVariable("SEMMLE_TYPESCRIPT_MEMORY_THRESHOLD", Number, 1000);
let isAboveReloadThreshold = false;
let rl = readline.createInterface({ input: process.stdin, output: process.stdout });
rl.on("line", (line: string) => {
let req: Command = JSON.parse(line);
switch (req.command) {
case "parse":
handleParseCommand(req);
// If memory usage has moved above the threshold, reboot the TypeScript compiler instance.
let bytesUsed = process.memoryUsage().heapUsed;
let megabytesUsed = bytesUsed / 1000000;
if (!isAboveReloadThreshold && megabytesUsed > reloadMemoryThresholdMb && state.project != null) {
console.warn('Restarting TypeScript compiler due to memory usage');
state.project.reload();
isAboveReloadThreshold = true;
} else if (isAboveReloadThreshold && megabytesUsed < reloadMemoryThresholdMb) {
isAboveReloadThreshold = false;
}
checkMemoryUsage();
break;
case "open-project":
handleOpenProjectCommand(req);