From c036d0702e6f782bf1f43ed63f8e7a2e32b711b5 Mon Sep 17 00:00:00 2001
From: Alec Larson <1925840+aleclarson@users.noreply.github.com>
Date: Wed, 7 Sep 2022 13:23:25 -0400
Subject: [PATCH] fix(pg-query-stream): invoke `this.callback` on cursor
 end/error

Closes #2013
---
 packages/pg-query-stream/src/index.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/packages/pg-query-stream/src/index.ts b/packages/pg-query-stream/src/index.ts
index c942b0441..f3251b919 100644
--- 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)