Refactor logging and resource tracking in index.ts and start_notification_aggregator_process.ts

Author: Kushagra Singh Bisen

src/index.ts

@@ -3,6 +3,7 @@ import * as bunyan from "bunyan";
 import * as fs from 'fs';
 import { program } from "commander";
 const log_file = fs.createWriteStream('./logs/info.log', { flags: 'a' });
+const resource_used_log_file = `./logs/resource_used.csv`;
 
 const logger = bunyan.createLogger({
     name: 'solid-stream-notifications-cache',
@@ -19,7 +20,27 @@ const logger = bunyan.createLogger({
             }
         }
     }
-})
+});
+
+interface MemoryUsage {
+    rss: number;
+    heapTotal: number;
+    heapUsed: number;
+    external: number;
+}
+
+fs.writeFileSync(resource_used_log_file, `timestamp, cpu_user, cpu_system, rss, heapTotal, heapUsed, external\n`);
+
+
+function logCpuMemoryUsage() {
+    const cpuUsage = process.cpuUsage(); // in microseconds
+    const memoryUsage: MemoryUsage = process.memoryUsage(); // in bytes
+    const timestamp = Date.now();
+    const logData = `${timestamp}, ${cpuUsage.user}, ${cpuUsage.system}, ${memoryUsage.rss}, ${memoryUsage.heapTotal}, ${memoryUsage.heapUsed}, ${memoryUsage.external}\n`;
+    fs.appendFileSync(resource_used_log_file, logData);
+}
+
+setInterval(logCpuMemoryUsage, 500);
 
 program
     .version('1.0.0')

start_notification_aggregator_process.ts

@@ -1,46 +1,15 @@
-const { spawn } = require('child_process');
-import * as fs from 'fs';
+import { fork, ChildProcess } from 'child_process';
 
-const child = spawn('npm', ['run', 'start'], {
-    stdio: 'inherit',
-    shell: true
-});
+export function startNotificationAggregatorProcess() {
+    const aggregator_process: ChildProcess = fork('./dist/index.js')
 
+    aggregator_process.on('message', (message: string) => {
+        console.log(`Message from the notification aggregator process: ${message}`);
+    });
 
-function getTimestamp() {
-    const now = new Date();
-    return `${now.getFullYear()}-${(now.getMonth() + 1).toString().padStart(2, '0')}-${now.getDate().toString().padStart(2, '0')}-${now.getHours().toString().padStart(2, '0')}-${now.getMinutes().toString().padStart(2, '0')}-${now.getSeconds().toString().padStart(2, '0')}`;
+    aggregator_process.on('exit', (code: number, signal: string) => {
+        console.log(`Notification aggregator process exited with code ${code} and signal ${signal}`);
+    });
 }
 
-interface MemoryUsage {
-    rss: number;
-    heapTotal: number;
-    heapUsed: number;
-    external: number;
-}
-
-function logCPUMemoryUsage(logFile: string) {
-    const cpuUsage = process.cpuUsage();
-    const memoryUsage: MemoryUsage = process.memoryUsage();
-    const timestamp = Date.now();
-    const logLine = `${timestamp},${cpuUsage.user},${cpuUsage.system},${memoryUsage.rss},${memoryUsage.heapTotal},${memoryUsage.heapUsed},${memoryUsage.external}\n`;
-    fs.appendFileSync(logFile, logLine);
-}
-
-const timestamp = getTimestamp();
-const logFile = `notifications-stream-aggregator-resource-usage-${timestamp}.csv`;
-fs.writeFileSync(logFile, 'timestamp,cpu_user,cpu_system,memory_rss,memory_heapTotal,memory_heapUsed,memory_external\n');
-
-setInterval(() => {
-    logCPUMemoryUsage(logFile);
-}, 500);
-
-child.on('exit', (code: any) => {
-    console.log(`Child process exited with code ${code}`);
-    process.exit(code);
-});
-
-child.on('error', (err: any) => {
-    console.log(`Child process error: ${err}`);
-    process.exit(1);
-});
+startNotificationAggregatorProcess();