feat: emit connection event for a StreamConnection (#533)

Fixes https://togithub.com/googleapis/nodejs-bigquery-storage/issues/529 🦕 Supersedes https://togithub.com/googleapis/nodejs-bigquery-storage/pull/530
googleapis · Jan 9, 2025 · 6688ea3 · 6688ea3
1 parent f125792
commit 6688ea3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/managedwriter/stream_connection.ts b/src/managedwriter/stream_connection.ts
@@ -107,12 +107,15 @@ export class StreamConnection extends EventEmitter {
     });
     this._connection.on('pause', () => {
       this.trace('connection paused');
+      this.emit('pause');
     });
     this._connection.on('resume', () => {
       this.trace('connection resumed');
+      this.emit('resume');
     });
     this._connection.on('end', () => {
       this.trace('connection ended');
+      this.emit('end');
     });
   }
 
@@ -364,6 +367,7 @@ export class StreamConnection extends EventEmitter {
     );
     this.close();
     this.open();
+    this.emit('reconnect');
   }
 
   /**
@@ -375,6 +379,7 @@ export class StreamConnection extends EventEmitter {
     }
     this._connection.end();
     this._connection.removeAllListeners();
+    this.emit('close');
     this._connection.destroy();
     this._connection = null;
   }

diff --git a/system-test/managed_writer_client_test.ts b/system-test/managed_writer_client_test.ts
@@ -1655,7 +1655,7 @@ describe('managedwriter.WriterClient', () => {
       });
 
       let reconnectedCalled = false;
-      sandbox.stub(connection, 'reconnect').callsFake(() => {
+      connection.on('reconnect', () => {
         reconnectedCalled = true;
       });