@senzops/apm-node 1.1.8 → 1.1.11

package/src/instrumentation/cron.ts

@@ -0,0 +1,41 @@
+import type { SenzorClient } from '../core/client';
+import { hookRequire } from './hook';
+
+export const instrumentNodeCron = (client: SenzorClient, debug: boolean) => {
+  hookRequire('node-cron', (cronExports) => {
+
+    // Abstracted patcher so we can apply it to both the root and the .default export
+    const patchSchedule = (target: any) => {
+      if (!target || typeof target.schedule !== 'function' || target.__senzorPatched) return;
+
+      const originalSchedule = target.schedule;
+
+      target.schedule = function (expression: string, func: (...args: any[]) => any, options: any) {
+        // Handle node-cron's dynamic options argument (can be string or object)
+        const optsObj = typeof options === 'object' ? options : { timezone: options };
+        const taskName = optsObj?.name || `cron: ${expression}`;
+
+        const wrappedFunc = client.wrapTask(
+          taskName,
+          'cron',
+          { metadata: optsObj },
+          func
+        );
+
+        return originalSchedule.call(this, expression, wrappedFunc, options);
+      };
+
+      // Safely mark as patched to prevent infinite loops
+      Object.defineProperty(target, '__senzorPatched', { value: true, enumerable: false, writable: true });
+      if (debug) console.log('[Senzor] Node-Cron successfully instrumented');
+    };
+
+    // Apply patch to root (for const cron = require('node-cron'))
+    patchSchedule(cronExports);
+
+    // Apply patch to default (for import cron from 'node-cron')
+    if (cronExports.default) {
+      patchSchedule(cronExports.default);
+    }
+  });
+};

package/src/instrumentation/hook.ts

@@ -0,0 +1,24 @@
+import Module from 'module';
+
+export const hookRequire = (moduleName: string, onRequire: (exports: any) => void) => {
+  // 1. If it was already loaded (e.g., imported at the top of the file before init)
+  try {
+    const resolvedPath = require.resolve(moduleName);
+    const cached = require.cache[resolvedPath];
+    if (cached && cached.exports) {
+      onRequire(cached.exports);
+    }
+  } catch (e) {
+    // Silently ignore if module is not installed
+  }
+
+  // 2. Intercept future requires
+  const originalLoad = (Module as any)._load;
+  (Module as any)._load = function (request: string, parent: any, isMain: boolean) {
+    const exports = originalLoad.apply(this, arguments);
+    if (request === moduleName && exports) {
+      onRequire(exports);
+    }
+    return exports;
+  };
+};