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

JS: simplify cache interaction

Browse Source
This commit is contained in:
Esben Sparre Andreasen
2019-09-05 08:16:18 +01:00
parent 6dbe827dd3
commit 03d38ca54b
5 changed files with 56 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
javascript/extractor/src/com/semmle/js/extractor/ExtractionMetrics.java
View File

@@ -1,19 +1,11 @@
package com.semmle.js.extractor;
import com.semmle.util.exception.Exceptions;
import com.semmle.util.files.FileUtil;
import com.semmle.util.trap.TrapWriter;
import com.semmle.util.trap.TrapWriter.Label;
import com.semmle.util.trap.pathtransformers.PathTransformer;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.lang.management.ManagementFactory;
import java.lang.management.ThreadMXBean;
import java.nio.charset.Charset;
import java.util.Stack;
import java.util.zip.GZIPOutputStream;
/** Metrics for the (single-threaded) extraction of a single file. */
public class ExtractionMetrics {
@@ -79,55 +71,39 @@ public class ExtractionMetrics {
private boolean timingsFailed;
/**
* Appends these metrics to a trap file. Note that this makes the resulting trap file content
* Writes the data metrics to a trap file. Note that this makes the resulting trap file content
* non-deterministic.
*/
public void appendToTrapFile(File trapFileToAppendTo) {
if (trapFileToAppendTo == null) {
return;
public void writeDataToTrap(TrapWriter trapwriter) {
trapwriter.addTuple(
"extraction_data",
fileLabel,
cacheFile != null ? cacheFile.getAbsolutePath() : "",
canReuseCacheFile,
length);
}
/**
* Writes the timing metrics to a trap file. Note that this makes the resulting trap file content
* non-deterministic.
*/
public void writeTimingsToTrap(TrapWriter trapwriter) {
if (!stack.isEmpty()) {
failTimings(
String.format(
"Could not properly record extraction times for %s. (stack = %s)%n",
fileLabel, stack.toString()));
}
BufferedWriter out = null;
FileOutputStream fos = null;
GZIPOutputStream gzip = null;
TrapWriter trapwriter = null;
try {
fos = new FileOutputStream(trapFileToAppendTo, true);
gzip = new GZIPOutputStream(fos);
out = new BufferedWriter(new OutputStreamWriter(gzip, Charset.forName("UTF-8")));
trapwriter = new TrapWriter(out, PathTransformer.std());
trapwriter.addTuple(
"extraction_data",
fileLabel,
cacheFile != null ? cacheFile.getAbsolutePath() : "",
canReuseCacheFile,
length);
if (!stack.isEmpty()) {
failTimings(
String.format(
"Could not properly record extraction times for %s. (stack = %s)%n",
fileLabel, stack.toString()));
if (!timingsFailed) {
for (int i = 0; i < ExtractionPhase.values().length; i++) {
trapwriter.addTuple("extraction_time", fileLabel, i, 0, (float) cpuTimes[i]);
trapwriter.addTuple("extraction_time", fileLabel, i, 1, (float) wallclockTimes[i]);
}
if (!timingsFailed) {
for (int i = 0; i < ExtractionPhase.values().length; i++) {
trapwriter.addTuple("extraction_time", fileLabel, i, 0, (float) cpuTimes[i]);
trapwriter.addTuple("extraction_time", fileLabel, i, 1, (float) wallclockTimes[i]);
}
}
FileUtil.close(trapwriter);
} catch (Exception e) {
FileUtil.close(fos);
FileUtil.close(gzip);
FileUtil.close(out);
FileUtil.close(trapwriter);
Exceptions.ignore(e, "Ignoring exception for extraction metrics writing");
}
}
private void failTimings(String msg) {
System.err.printf(msg);
System.err.println(msg);
System.err.flush();
this.timingsFailed = true;
}

8
javascript/extractor/src/com/semmle/js/extractor/FileExtractor.java
View File

@@ -450,14 +450,12 @@ public class FileExtractor {
metrics.setCacheFile(cacheFile);
metrics.setCanReuseCacheFile(canReuseCacheFile);
metrics.writeDataToTrap(trapwriter);
if (canUseCacheFile) {
FileUtil.close(trapwriter);
if (canReuseCacheFile) {
FileUtil.append(cacheFile, resultFile);
metrics.stopPhase(ExtractionPhase.FileExtractor_extractContents);
metrics.appendToTrapFile(resultFile);
return null;
}
@@ -481,14 +479,14 @@ public class FileExtractor {
int linesOfCode = loc.getLinesOfCode(), linesOfComments = loc.getLinesOfComments();
trapwriter.addTuple("numlines", fileLabel, numLines, linesOfCode, linesOfComments);
trapwriter.addTuple("filetype", fileLabel, fileType.toString());
metrics.stopPhase(ExtractionPhase.FileExtractor_extractContents);
metrics.writeTimingsToTrap(trapwriter);
successful = true;
return linesOfCode;
} finally {
if (!successful && trapwriter instanceof CachingTrapWriter)
((CachingTrapWriter) trapwriter).discard();
FileUtil.close(trapwriter);
metrics.stopPhase(ExtractionPhase.FileExtractor_extractContents);
metrics.appendToTrapFile(resultFile);
}
}

2
javascript/extractor/src/com/semmle/js/extractor/Main.java
View File

@@ -37,7 +37,7 @@ public class Main {
* A version identifier that should be updated every time the extractor changes in such a way that
* it may produce different tuples for the same file under the same {@link ExtractorConfig}.
*/
public static final String EXTRACTOR_VERSION = "2019-09-03";
public static final String EXTRACTOR_VERSION = "2019-09-04";
public static final Pattern NEWLINE = Pattern.compile("\n");

10
javascript/extractor/src/com/semmle/js/extractor/test/TrapTests.java
View File

@@ -166,6 +166,16 @@ public class TrapTests {
String expected = new WholeIO().strictreadText(trap);
expectedVsActual.add(Pair.make(expected, actual));
}
@Override
public void addTuple(String tableName, Object... values) {
if ("extraction_data".equals(tableName)
|| "extraction_time".equals(tableName)) {
// ignore non-deterministic tables
return;
}
super.addTuple(tableName, values);
}
};
}