@thru/replay 0.2.3 → 0.2.5

package/dist/index.cjs

@@ -54,7 +54,11 @@ var ChainClient = class {
     return connectNode.createGrpcTransport({
       baseUrl: this.options.baseUrl,
       useBinaryFormat: this.options.useBinaryFormat ?? true,
-      interceptors: mergedInterceptors.length ? mergedInterceptors : void 0
+      interceptors: mergedInterceptors.length ? mergedInterceptors : void 0,
+      pingIntervalMs: 3e4,
+      pingIdleConnection: true,
+      pingTimeoutMs: 1e4,
+      idleConnectionTimeoutMs: 0
     });
   }
   createHeaderInterceptor() {
@@ -313,7 +317,7 @@ var LivePump = class {
 var DEFAULT_RETRY_CONFIG = {
   initialDelayMs: 1e3,
   maxDelayMs: 3e4,
-  connectionTimeoutMs: 3e4
+  connectionTimeoutMs: 3e5
 };
 function calculateBackoff(attempt, config) {
   const delay2 = config.initialDelayMs * Math.pow(2, attempt);
@@ -584,7 +588,10 @@ var ReplayStream = class {
 };
 async function safeClose(pump) {
   try {
-    await pump.close();
+    await Promise.race([
+      pump.close(),
+      new Promise((resolve) => setTimeout(resolve, 5e3))
+    ]);
   } catch {
   }
 }
@@ -1071,6 +1078,7 @@ async function* createAccountsByOwnerReplay(options) {
   const streamBuffer = [];
   let streamDone = false;
   let streamError = null;
+  let lastActivityTime = Date.now();
   try {
     cleanupTimer = setInterval(() => {
       assembler.cleanup();
@@ -1080,6 +1088,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") {
@@ -1169,6 +1178,7 @@ async function* createAccountsByOwnerReplay(options) {
     let retryAttempt = 0;
     let currentStream = stream;
     let currentStreamProcessor = streamProcessor;
+    lastActivityTime = Date.now();
     const createStreamProcessor = () => {
       if (clientFactory) {
         try {
@@ -1184,6 +1194,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") {
@@ -1204,7 +1215,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);
@@ -1216,6 +1238,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;
@@ -1223,6 +1246,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;