fix(pg-query-stream): invoke this.callback on cursor end/error

Closes brianc#2013
aleclarson · Dec 5, 2023 · e55da72 · e55da72
1 parent 6cd0aeb
commit e55da72
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/pg-query-stream/src/index.ts b/packages/pg-query-stream/src/index.ts
@@ -13,6 +13,7 @@ class QueryStream extends Readable implements Submittable {
   cursor: any
   _result: any
 
+  callback: Function
   handleRowDescription: Function
   handleDataRow: Function
   handlePortalSuspended: Function
@@ -26,6 +27,11 @@ class QueryStream extends Readable implements Submittable {
 
     super({ objectMode: true, autoDestroy: true, highWaterMark: batchSize || highWaterMark })
     this.cursor = new Cursor(text, values, config)
+    this.cursor.on('end', (result) => {
+      this.callback && this.callback(null, result)
+    }).on('error', (err) => {
+      this.callback && this.callback(err)
+    })
 
     // delegate Submittable callbacks to cursor
     this.handleRowDescription = this.cursor.handleRowDescription.bind(this.cursor)