wrap errors in NetworkError

fauna · Apr 17, 2024 · d25d0f3 · d25d0f3
1 parent 3180e45
commit d25d0f3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/src/http-client/fetch-client.ts b/src/http-client/fetch-client.ts
@@ -110,8 +110,15 @@ export class FetchClient implements HTTPClient, HTTPStreamClient {
       }
       const reader = body.getReader();
 
-      for await (const line of readLines(reader)) {
-        yield line;
+      try {
+        for await (const line of readLines(reader)) {
+          yield line;
+        }
+      } catch (error) {
+        throw new NetworkError(
+          "The network connection encountered a problem while streaming events.",
+          { cause: error },
+        );
       }
     }
 

diff --git a/src/http-client/node-http2-client.ts b/src/http-client/node-http2-client.ts
@@ -205,7 +205,12 @@ export class NodeHTTP2Client implements HTTPClient, HTTPStreamClient {
           .request(httpRequestHeaders)
           .setEncoding("utf8")
           .on("error", (error: any) => {
-            rejectPromise(error);
+            rejectPromise(
+              new NetworkError(
+                "The network connection encountered a problem while streaming events.",
+                { cause: error },
+              ),
+            );
           })
           .on("response", onResponse);
 
@@ -218,7 +223,12 @@ export class NodeHTTP2Client implements HTTPClient, HTTPStreamClient {
 
         req.end();
       } catch (error) {
-        rejectPromise(error);
+        rejectPromise(
+          new NetworkError(
+            "The network connection encountered a problem while streaming events.",
+            { cause: error },
+          ),
+        );
       }
     });
   }