close listener connection on TimedOut and BrokerPipe errors

launchbadge · Dec 20, 2024 · 3c0d561 · 3c0d561
1 parent 9f6ea96
commit 3c0d561
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sqlx-postgres/src/listener.rs b/sqlx-postgres/src/listener.rs
@@ -280,7 +280,10 @@ impl PgListener {
                 // update self state, and loop to try again.
                 Err(Error::Io(err))
                     if (err.kind() == io::ErrorKind::ConnectionAborted
-                        || err.kind() == io::ErrorKind::UnexpectedEof) =>
+                        || err.kind() == io::ErrorKind::UnexpectedEof
+                        // see ERRORS section in tcp(7) man page (https://man7.org/linux/man-pages/man7/tcp.7.html)
+                        || err.kind() == io::ErrorKind::TimedOut
+                        || err.kind() == io::ErrorKind::BrokenPipe) =>
                 {
                     if let Some(mut conn) = self.connection.take() {
                         self.buffer_tx = conn.inner.stream.notifications.take();