feat: add meta-data to stream error (#620)

* feat: add meta-data to stream error * fix last commit Co-authored-by: Francois HERSENT <[email protected]>
stephenh · Jul 21, 2022 · b68f301 · b68f301
1 parent 9778fa3
commit b68f301
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/integration/grpc-web/example.ts b/integration/grpc-web/example.ts
@@ -956,13 +956,16 @@ export class GrpcWebImpl {
           metadata: maybeCombinedMetadata,
           debug: this.options.debug,
           onMessage: (next) => observer.next(next),
-          onEnd: (code: grpc.Code, message: string) => {
+          onEnd: (code: grpc.Code, message: string, trailers: grpc.Metadata) => {
             if (code === 0) {
               observer.complete();
             } else if (upStreamCodes.includes(code)) {
               setTimeout(upStream, DEFAULT_TIMEOUT_TIME);
             } else {
-              observer.error(new Error(`Error ${code} ${message}`));
+              const err = new Error(message) as any;
+              err.code = code;
+              err.metadata = trailers;
+              observer.error(err);
             }
           },
         });

diff --git a/src/generate-grpc-web.ts b/src/generate-grpc-web.ts
@@ -338,13 +338,16 @@ function createInvokeMethod(ctx: Context) {
             metadata: maybeCombinedMetadata,
             debug: this.options.debug,
             onMessage: (next) => observer.next(next),
-            onEnd: (code: ${grpc}.Code, message: string) => {
+            onEnd: (code: ${grpc}.Code, message: string, trailers: ${grpc}.Metadata) => {
               if (code === 0) {
                 observer.complete();
               } else if (upStreamCodes.includes(code)) {
                 setTimeout(upStream, DEFAULT_TIMEOUT_TIME);
               } else {
-                observer.error(new Error(\`Error \${code} \${message}\`));
+                const err = new Error(message) as any;
+                err.code = code;
+                err.metadata = trailers;
+                observer.error(err);
               }
             },
           });