add handler

porsager · Feb 17, 2024 · 3e3d5e8 · 3e3d5e8
1 parent cd02af8
commit 3e3d5e8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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 }
     })
   }