From 3e3d5e894a86b03b5e6edac9f52bd7ca4abd2ce5 Mon Sep 17 00:00:00 2001
From: "louis.tian" <louis.tian@aquamonix.com.au>
Date: Wed, 24 Jan 2024 11:02:11 +1100
Subject: [PATCH] add handler

---
 src/subscribe.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/subscribe.js b/src/subscribe.js
index 7a70842e..decb42c6 100644
--- a/src/subscribe.js
+++ b/src/subscribe.js
@@ -47,7 +47,7 @@ export default function Subscribe(postgres, options) {
 
   return subscribe
 
-  async function subscribe(event, fn, onsubscribe = noop) {
+  async function subscribe(event, fn, onsubscribe = noop, onerror = noop) {
     event = parseEvent(event)
 
     if (!connection)
@@ -66,6 +66,7 @@ export default function Subscribe(postgres, options) {
     return connection.then(x => {
       connected(x)
       onsubscribe()
+      stream && stream.on('error', onerror)
       return { unsubscribe, state, sql }
     })
   }