From e7a893cd4bfbf9aa519552bbf6351e1d28a608dd Mon Sep 17 00:00:00 2001
From: Alexander Onnikov <Alexander.Onnikov@xored.com>
Date: Tue, 3 Sep 2024 17:07:24 +0700
Subject: [PATCH] fix: retry requests to collaborator in case of failure
 (#6468)

Signed-off-by: Alexander Onnikov <Alexander.Onnikov@xored.com>
---
 packages/collaborator-client/src/client.ts | 36 ++++++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/packages/collaborator-client/src/client.ts b/packages/collaborator-client/src/client.ts
index bca4a8c0b1..c7ffa716a4 100644
--- a/packages/collaborator-client/src/client.ts
+++ b/packages/collaborator-client/src/client.ts
@@ -69,6 +69,10 @@ class CollaboratorClientImpl implements CollaboratorClient {
       body: JSON.stringify({ method, documentId, payload })
     })
 
+    if (!res.ok) {
+      throw new Error('HTTP error ' + res.status)
+    }
+
     const result = await res.json()
 
     if (result.error != null) {
@@ -79,12 +83,24 @@ class CollaboratorClientImpl implements CollaboratorClient {
   }
 
   async getContent (document: CollaborativeDoc): Promise<Record<string, Markup>> {
-    const res = (await this.rpc(document, 'getContent', {})) as GetContentResponse
+    const res = await retry(
+      3,
+      async () => {
+        return (await this.rpc(document, 'getContent', {})) as GetContentResponse
+      },
+      50
+    )
     return res.content ?? {}
   }
 
   async updateContent (document: CollaborativeDoc, content: Record<string, Markup>): Promise<void> {
-    await this.rpc(document, 'updateContent', { content })
+    await retry(
+      3,
+      async () => {
+        await this.rpc(document, 'updateContent', { content })
+      },
+      50
+    )
   }
 
   async copyContent (source: CollaborativeDoc, target: CollaborativeDoc): Promise<void> {
@@ -92,3 +108,19 @@ class CollaboratorClientImpl implements CollaboratorClient {
     await this.updateContent(target, content)
   }
 }
+
+async function retry<T> (retries: number, op: () => Promise<T>, delay: number = 100): Promise<T> {
+  let error: any
+  while (retries > 0) {
+    retries--
+    try {
+      return await op()
+    } catch (err: any) {
+      error = err
+      if (retries !== 0) {
+        await new Promise((resolve) => setTimeout(resolve, delay))
+      }
+    }
+  }
+  throw error
+}