Properly end multihost sockets

porsager · porsager · commit d6c15fd66d57 · 2021-03-24T00:06:34.000+01:00
diff --git a/lib/connection.js b/lib/connection.js
@@ -283,6 +283,7 @@ function postgresSocket(options, {
   data
 }) {
   let socket
+  let ended = false
   let closed = true
   let succeeded = false
   let next = null
@@ -291,7 +292,7 @@ function postgresSocket(options, {
 
   function onclose(err) {
     oncleanup()
-    !succeeded && i < options.host.length
+    !ended && !succeeded && i < options.host.length
       ? connect()
       : err instanceof Error
         ? error(err)
@@ -379,6 +380,7 @@ function postgresSocket(options, {
       return Promise.resolve()
     },
     end: () => {
+      ended = true
       return new Promise(r => socket && !closed ? (socket.once('close', r), socket.end()) : r())
     },
     connect
