@ai-matrx/api 0.0.0-stage → 0.1.0

package/dist/chunk-YTEOFGFC.js

@@ -0,0 +1,57 @@
+import { TotalTimeoutError, HeartbeatTimeoutError } from './chunk-KIEAFMKB.js';
+
+// src/net/stream-monitor.ts
+var DEFAULTS = {
+  heartbeatTimeoutMs: 3e4,
+  maxLifetimeMs: 6e5
+};
+async function* monitorStream(source, opts = {}) {
+  const heartbeatTimeoutMs = opts.heartbeatTimeoutMs ?? DEFAULTS.heartbeatTimeoutMs;
+  const maxLifetimeMs = opts.maxLifetimeMs ?? DEFAULTS.maxLifetimeMs;
+  const abort = opts.abortController;
+  const startedAt = Date.now();
+  const iter = source[Symbol.asyncIterator]();
+  while (true) {
+    const elapsed = Date.now() - startedAt;
+    const remainingLifetime = maxLifetimeMs - elapsed;
+    if (remainingLifetime <= 0) {
+      abort?.abort("total-timeout");
+      throw new TotalTimeoutError(maxLifetimeMs);
+    }
+    const deadline = Math.min(heartbeatTimeoutMs, remainingLifetime);
+    let timer = null;
+    let timedOutKind = null;
+    const timeoutPromise = new Promise((_, reject) => {
+      timer = setTimeout(() => {
+        timedOutKind = deadline === remainingLifetime ? "total" : "heartbeat";
+        abort?.abort(
+          timedOutKind === "total" ? "total-timeout" : "heartbeat-timeout"
+        );
+        reject(
+          timedOutKind === "total" ? new TotalTimeoutError(maxLifetimeMs) : new HeartbeatTimeoutError(heartbeatTimeoutMs)
+        );
+      }, deadline);
+    });
+    let result;
+    try {
+      result = await Promise.race([iter.next(), timeoutPromise]);
+    } catch (err) {
+      if (timer !== null) clearTimeout(timer);
+      try {
+        const cleanup = iter.return?.(void 0);
+        if (cleanup) {
+          void Promise.resolve(cleanup).catch(() => {
+          });
+        }
+      } catch {
+      }
+      throw err;
+    }
+    if (timer !== null) clearTimeout(timer);
+    if (result.done) return;
+    opts.onHeartbeat?.(Date.now());
+    yield result.value;
+  }
+}
+
+export { monitorStream };