@zintrust/trace 1.6.5 → 1.6.7

package/dist/watchers/HttpWatcher.js

@@ -1,3 +1,4 @@
+import { Logger } from '@zintrust/core';
 import { TraceContext } from '../context.js';
 import { EntryType } from '../types.js';
 import { AuthTag } from '../utils/authTag.js';
@@ -35,8 +36,9 @@ const resolveRequestPayload = (req, config) => {
 };
 const registerCompletionHandler = (response, onComplete) => {
     const raw = response.getRaw();
-    if (typeof raw.once !== 'function')
-        return () => undefined;
+    if (typeof raw.once !== 'function') {
+        return { attached: false, cleanup: () => undefined };
+    }
     let completed = false;
     const cleanup = () => {
         if (typeof raw.off === 'function') {
@@ -53,7 +55,11 @@ const registerCompletionHandler = (response, onComplete) => {
     };
     raw.once('finish', markCompleted);
     raw.once('close', markCompleted);
-    return cleanup;
+    return { attached: true, cleanup };
+};
+const isResponseComplete = (response) => {
+    const raw = response.getRaw();
+    return raw.writableEnded === true || raw.finished === true;
 };
 const captureResponse = (response, config) => {
     const headers = {};
@@ -147,8 +153,7 @@ export const HttpWatcher = Object.freeze({
                 if (content.responseStatus >= 500)
                     tags.push('failed');
                 responseCapture.restore();
-                storage
-                    .writeEntry({
+                const entry = {
                     uuid: crypto.randomUUID(),
                     batchId,
                     type: EntryType.REQUEST,
@@ -156,11 +161,21 @@ export const HttpWatcher = Object.freeze({
                     tags,
                     isLatest: true,
                     createdAt: TraceContext.now(),
-                })
-                    .catch(() => undefined); // fire-and-forget
+                };
+                storage.writeEntry(entry).catch((error) => {
+                    Logger.warn('[trace] HttpWatcher writeEntry failed', {
+                        method: content.method,
+                        uri: content.uri,
+                        entryUuid: entry.uuid,
+                        error: error instanceof Error ? error.message : String(error),
+                    });
+                }); // fire-and-forget
             };
-            registerCompletionHandler(response, persistEntry);
+            const completionHandler = registerCompletionHandler(response, persistEntry);
             await next();
+            if (!completionHandler.attached || isResponseComplete(response)) {
+                persistEntry();
+            }
         };
         registerMiddleware(middleware);
         return () => undefined;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zintrust/trace",
-  "version": "1.6.5",
+  "version": "1.6.7",
   "description": "Trace assistant for ZinTrust: logs requests, queries, exceptions, jobs, and more.",
   "private": false,
   "type": "module",