improve mongo logs

Author: neSpecc

src/metrics/mongodb.ts

@@ -132,11 +132,74 @@ function formatParams(params: any): string {
 }
 
 /**
- * Log MongoDB command details to console
- * Format: [requestId] db.collection.command(params)
+ * Colorize duration based on performance thresholds
+ * @param duration - Duration in milliseconds
+ * @returns Colorized duration string
+ */
+function colorizeDuration(duration: number): string {
+  let color: Effect;
+
+  if (duration < 50) {
+    color = Effect.ForegroundGreen;
+  } else if (duration < 100) {
+    color = Effect.ForegroundYellow;
+  } else {
+    color = Effect.ForegroundRed;
+  }
+
+  return sgr(`${duration}ms`, color);
+}
+
+/**
+ * Interface for storing command information with timestamp
+ */
+interface StoredCommandInfo {
+  formattedCommand: string;
+  timestamp: number;
+}
+
+/**
+ * Map to store formatted command information by requestId
+ */
+const commandInfoMap = new Map<number, StoredCommandInfo>();
+
+/**
+ * Timeout for cleaning up stale command info (30 seconds)
+ */
+const COMMAND_INFO_TIMEOUT_MS = 30000;
+
+/**
+ * Cleanup stale command info to prevent memory leaks
+ * Removes entries older than COMMAND_INFO_TIMEOUT_MS
+ */
+function cleanupStaleCommandInfo(): void {
+  const now = Date.now();
+  const keysToDelete: number[] = [];
+
+  for (const [requestId, info] of commandInfoMap.entries()) {
+    if (now - info.timestamp > COMMAND_INFO_TIMEOUT_MS) {
+      keysToDelete.push(requestId);
+    }
+  }
+
+  if (keysToDelete.length > 0) {
+    console.warn(`Cleaning up ${keysToDelete.length} stale MongoDB command info entries (possible memory leak)`);
+    for (const key of keysToDelete) {
+      commandInfoMap.delete(key);
+    }
+  }
+}
+
+/**
+ * Periodic cleanup interval
+ */
+setInterval(cleanupStaleCommandInfo, COMMAND_INFO_TIMEOUT_MS);
+
+/**
+ * Store MongoDB command details for later logging
  * @param event - MongoDB command event
  */
-function logCommandStarted(event: any): void {
+function storeCommandInfo(event: any): void {
   const collectionRaw = extractCollectionFromCommand(event.command, event.commandName);
   const collection = sgr(normalizeCollectionName(collectionRaw), Effect.ForegroundGreen);
   const db = event.databaseName || 'unknown db';
@@ -147,28 +210,49 @@ function logCommandStarted(event: any): void {
   const projection = event.command.projection;
   const params = filter || update || pipeline;
   const paramsStr = formatParams(params);
+  const projectionStr = projection ? ` projection: ${formatParams(projection)}` : '';
+
+  const formattedCommand = `[${event.requestId}] ${db}.${collection}.${commandName}(${paramsStr})${projectionStr}`;
 
-  console.log(`[${event.requestId}] ${db}.${collection}.${commandName}(${paramsStr}) ${projection ? `projection: ${formatParams(projection)}` : ''}`);
+  commandInfoMap.set(event.requestId, {
+    formattedCommand,
+    timestamp: Date.now(),
+  });
 }
 
 /**
  * Log MongoDB command success to console
- * Format: [requestId] commandName ✓ duration
+ * Format: [requestId] db.collection.command(params) ✓ duration
  * @param event - MongoDB command event
  */
 function logCommandSucceeded(event: any): void {
-  console.log(`[${event.requestId}] ${event.commandName} ✓ ${event.duration}ms`);
+  const info = commandInfoMap.get(event.requestId);
+  const durationStr = colorizeDuration(event.duration);
+
+  if (info) {
+    console.log(`${info.formattedCommand} ✓ ${durationStr}`);
+    commandInfoMap.delete(event.requestId);
+  } else {
+    console.log(`[${event.requestId}] ${event.commandName} ✓ ${durationStr}`);
+  }
 }
 
 /**
  * Log MongoDB command failure to console
- * Format: [requestId] ✗ error
+ * Format: [requestId] db.collection.command(params) ✗ error duration
  * @param event - MongoDB command event
  */
 function logCommandFailed(event: any): void {
   const errorMsg = event.failure?.message || event.failure?.errmsg || 'Unknown error';
-
-  console.error(`[${event.requestId}] ${event.commandName} ✗ ${errorMsg} (${event.duration}ms)`);
+  const info = commandInfoMap.get(event.requestId);
+  const durationStr = colorizeDuration(event.duration);
+
+  if (info) {
+    console.error(`${info.formattedCommand} ✗ ${errorMsg} ${durationStr}`);
+    commandInfoMap.delete(event.requestId);
+  } else {
+    console.error(`[${event.requestId}] ${event.commandName} ✗ ${errorMsg} ${durationStr}`);
+  }
 }
 
 /**
@@ -177,7 +261,7 @@ function logCommandFailed(event: any): void {
  */
 export function setupMongoMetrics(client: MongoClient): void {
   client.on('commandStarted', (event) => {
-    logCommandStarted(event);
+    storeCommandInfo(event);
 
     // Store start time and metadata for this command
     const metadataKey = `${event.requestId}`;