aicodeswitch 5.2.10 → 5.2.12

package/dist/server/transformers/stream-timing-transform.js

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StreamTimingTransform = void 0;
+const stream_1 = require("stream");
+/**
+ * StreamTimingTransform - 流式打点 Transform（服务性能统计专用）
+ *
+ * 透传所有数据（对象模式 / Buffer / string 均可），仅记录：
+ *   - firstEventAt：首个被解析出的 SSE 事件流经本 Transform 的时刻（≈ 首 Token 返回时刻）
+ *   - lastEventAt：最后一个事件流经的时刻（≈ 整个返回结束时刻）
+ *
+ * 由调用方在转发开始前注入 startTime（请求发起时刻），即可派生：
+ *   - ttftMs       = firstEventAt - startTime
+ *   - generationMs = lastEventAt - firstEventAt
+ *
+ * 该 Transform 仅做时间记录，不修改任何数据内容，对转发链路零影响。
+ */
+class StreamTimingTransform extends stream_1.Transform {
+    constructor(startTime) {
+        super({ writableObjectMode: true, readableObjectMode: true });
+        /** 请求发起时刻（由外部注入） */
+        Object.defineProperty(this, "startTime", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        /** 首个事件到达时刻（未收到则为 0） */
+        Object.defineProperty(this, "firstEventAt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0
+        });
+        /** 最后一个事件到达时刻 */
+        Object.defineProperty(this, "lastEventAt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0
+        });
+        this.startTime = startTime;
+    }
+    _transform(chunk, _encoding, callback) {
+        try {
+            const now = Date.now();
+            if (this.firstEventAt === 0) {
+                this.firstEventAt = now;
+            }
+            this.lastEventAt = now;
+            this.push(chunk);
+        }
+        catch (error) {
+            console.error('[StreamTimingTransform] Error in _transform:', error);
+        }
+        callback();
+    }
+    /** 是否采集到至少一个事件（用于判定精确口径可用性） */
+    hasTiming() {
+        return this.firstEventAt > 0 && this.lastEventAt > 0;
+    }
+}
+exports.StreamTimingTransform = StreamTimingTransform;