fix(ddc): Throw RpcError if the upload stream closed by the node (#270)

Cerebellum-Network · Oct 2, 2024 · b6ccce1 · b6ccce1
1 parent 1b5e810
commit b6ccce1
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/packages/ddc/src/FileApi/FileApi.ts b/packages/ddc/src/FileApi/FileApi.ts
@@ -161,7 +161,10 @@ export class FileApi {
 
     const call = this.api.putRawPiece({ meta });
 
-    await call.requests.send({
+    /**
+     * Send none-blocking request message.
+     */
+    call.requests.send({
       body: {
         oneofKind: 'metadata',
         metadata: { ...metadata, size },
@@ -185,9 +188,12 @@ export class FileApi {
         break;
       }
 
-      await call.requests.send({
-        body: { oneofKind: 'content', content: { data: value } },
-      });
+      await Promise.race([
+        call.status,
+        call.requests.send({
+          body: { oneofKind: 'content', content: { data: value } },
+        }),
+      ]);
 
       bytesSent += value.byteLength;
     }
@@ -250,9 +256,9 @@ export class FileApi {
     const call = this.api.getFile({ meta });
 
     /**
-     * Send request message.
+     * Send none-blocking request message.
      */
-    await call.requests.send({
+    call.requests.send({
       body: {
         oneofKind: 'request',
         request: { ...request, authenticate },