@thru/replay 0.2.4 → 0.2.5

package/dist/index.mjs

@@ -587,7 +587,10 @@ var ReplayStream = class {
 };
 async function safeClose(pump) {
   try {
-    await pump.close();
+    await Promise.race([
+      pump.close(),
+      new Promise((resolve) => setTimeout(resolve, 5e3))
+    ]);
   } catch {
   }
 }
@@ -1074,6 +1077,7 @@ async function* createAccountsByOwnerReplay(options) {
   const streamBuffer = [];
   let streamDone = false;
   let streamError = null;
+  let lastActivityTime = Date.now();
   try {
     cleanupTimer = setInterval(() => {
       assembler.cleanup();
@@ -1083,6 +1087,7 @@ async function* createAccountsByOwnerReplay(options) {
     const streamProcessor = (async () => {
       try {
         for await (const response of stream) {
+          lastActivityTime = Date.now();
           const event = processResponseMulti(response, assembler);
           if (event) {
             if (event.type === "account") {
@@ -1172,6 +1177,7 @@ async function* createAccountsByOwnerReplay(options) {
     let retryAttempt = 0;
     let currentStream = stream;
     let currentStreamProcessor = streamProcessor;
+    lastActivityTime = Date.now();
     const createStreamProcessor = () => {
       if (clientFactory) {
         try {
@@ -1187,6 +1193,7 @@ async function* createAccountsByOwnerReplay(options) {
         try {
           for await (const response of newStream) {
             retryAttempt = 0;
+            lastActivityTime = Date.now();
             const event = processResponseMulti(response, assembler);
             if (event) {
               if (event.type === "account") {
@@ -1207,7 +1214,18 @@ async function* createAccountsByOwnerReplay(options) {
       return { stream: newStream, processor: newProcessor };
     };
     while (true) {
+      const hadEvents = streamBuffer.length > 0;
       yield* yieldStreamBuffer();
+      if (hadEvents) {
+        lastActivityTime = Date.now();
+      }
+      if (!streamDone && Date.now() - lastActivityTime > retryConfig.connectionTimeoutMs) {
+        logger.warn(
+          `[account-stream] no activity for ${retryConfig.connectionTimeoutMs}ms; forcing reconnection`
+        );
+        streamDone = true;
+        streamError = new Error(`Operation timed out after ${retryConfig.connectionTimeoutMs}ms`);
+      }
       if (streamDone) {
         if (streamError) {
           const backoffMs = calculateBackoff(retryAttempt, retryConfig);
@@ -1219,6 +1237,7 @@ async function* createAccountsByOwnerReplay(options) {
           streamDone = false;
           streamError = null;
           streamBuffer.length = 0;
+          lastActivityTime = Date.now();
           const { stream: newStream, processor: newProcessor } = createStreamProcessor();
           currentStream = newStream;
           currentStreamProcessor = newProcessor;
@@ -1226,6 +1245,7 @@ async function* createAccountsByOwnerReplay(options) {
         } else {
           logger.warn("[account-stream] stream ended unexpectedly; reconnecting...");
           streamDone = false;
+          lastActivityTime = Date.now();
           const { stream: newStream, processor: newProcessor } = createStreamProcessor();
           currentStream = newStream;
           currentStreamProcessor = newProcessor;