43 lines
1.4 KiB
Java
43 lines
1.4 KiB
Java
package ex11;
|
|
|
|
import java.io.IOException;
|
|
import java.io.OutputStream;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.StandardOpenOption;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
public final class Profiler {
|
|
private static final Map<CallInfo, long[]> count = new HashMap<>();
|
|
|
|
private static final Path outFile = Path.of("ex11.csv");
|
|
|
|
private static boolean stop = false;
|
|
|
|
static {
|
|
Runtime.getRuntime().addShutdownHook(new Thread(() -> {
|
|
stop = true;
|
|
try (final OutputStream path = Files.newOutputStream(outFile, StandardOpenOption.WRITE)) {
|
|
final CallInfoCSVWriter w = new CallInfoCSVWriter(path);
|
|
for (final var entry : count.entrySet()) {
|
|
System.out.println(entry.getKey().toString() + "\n#invokes " + entry.getValue()[0] + "\n");
|
|
w.write(entry.getKey(), entry.getValue()[0]);
|
|
}
|
|
System.out.println("Invocations have been written to: " + outFile.toAbsolutePath());
|
|
} catch (final IOException e) {
|
|
throw new RuntimeException("Writing CSV file with call info failed", e);
|
|
}
|
|
}));
|
|
}
|
|
|
|
private Profiler() {
|
|
}
|
|
|
|
public static void registerCall(final CallInfo info) {
|
|
if (stop) return;
|
|
count.computeIfAbsent(info, ignored -> new long[1])[0]++;
|
|
}
|
|
|
|
}
|