close listener connection on TimedOut and BrokenPipe errors (#3648)

* close listener connection on TimedOut and BrokerPipe errors * use matches macro to generate jump table instead of chain of conditions
launchbadge · Dec 25, 2024 · 4590b9c · 4590b9c
1 parent fdd4663
commit 4590b9c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/sqlx-postgres/src/listener.rs b/sqlx-postgres/src/listener.rs
@@ -279,8 +279,14 @@ impl PgListener {
                 // The connection is dead, ensure that it is dropped,
                 // update self state, and loop to try again.
                 Err(Error::Io(err))
-                    if (err.kind() == io::ErrorKind::ConnectionAborted
-                        || err.kind() == io::ErrorKind::UnexpectedEof) =>
+                    if matches!(
+                        err.kind(),
+                        io::ErrorKind::ConnectionAborted |
+                        io::ErrorKind::UnexpectedEof |
+                        // see ERRORS section in tcp(7) man page (https://man7.org/linux/man-pages/man7/tcp.7.html)
+                        io::ErrorKind::TimedOut |
+                        io::ErrorKind::BrokenPipe
+                    ) =>
                 {
                     if let Some(mut conn) = self.connection.take() {
                         self.buffer_tx = conn.inner.stream.notifications.take();