From def1ae1e9f0096ac6f194c829ba62f10d41d93f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Edgar=20Ram=C3=ADrez=20Mondrag=C3=B3n?=
 <edgarrm358@gmail.com>
Date: Mon, 25 Nov 2024 22:41:47 -0600
Subject: [PATCH] chore: Flush logger hanlders at exit

---
 tap_postgres/tap.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tap_postgres/tap.py b/tap_postgres/tap.py
index c3fc6ce3..de241859 100644
--- a/tap_postgres/tap.py
+++ b/tap_postgres/tap.py
@@ -504,6 +504,8 @@ def ssh_tunnel_connect(self, *, ssh_config: dict[str, Any], url: URL) -> URL:
     def clean_up(self) -> None:
         """Stop the SSH Tunnel."""
         self.logger.info("Shutting down SSH Tunnel")
+        for handler in self.logger.handlers:
+            handler.close()
         self.ssh_tunnel.stop()
 
     def catch_signal(self, signum, frame) -> None: