From 68e8cb5bc3d378084e787f31c2b5a1814be37cd5 Mon Sep 17 00:00:00 2001
From: Louis Pieterse <45560107+louis-pie@users.noreply.github.com>
Date: Tue, 10 Mar 2020 10:01:54 +0000
Subject: [PATCH] Session wal_sender_timeout for PG12 (#43)

---
 setup.py                                          |  2 +-
 .../sync_strategies/logical_replication.py        | 15 ++-------------
 2 files changed, 3 insertions(+), 14 deletions(-)

diff --git a/setup.py b/setup.py
index f4e18f0d..4edf2fd2 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
     long_description = f.read()
 
 setup(name='pipelinewise-tap-postgres',
-      version='1.5.0',
+      version='1.5.1',
       description='Singer.io tap for extracting data from PostgresSQL - PipelineWise compatible',
       long_description=long_description,
       long_description_content_type='text/markdown',
diff --git a/tap_postgres/sync_strategies/logical_replication.py b/tap_postgres/sync_strategies/logical_replication.py
index 1213d56e..53004715 100644
--- a/tap_postgres/sync_strategies/logical_replication.py
+++ b/tap_postgres/sync_strategies/logical_replication.py
@@ -395,19 +395,8 @@ def sync_tables(conn_info, logical_streams, state, end_lsn, state_file):
     # Set session wal_sender_timeout for PG12 and above
     version = get_pg_version(cur)
     if (version >= 120000):
-        wal_sender_timeout = 10800 #3 hours
-
-        # Detect if the source is amazon-rds and convert to milliseconds
-        cur.execute("SELECT count(*) FROM pg_settings where name ilike '%rds%'")
-        is_rds = cur.fetchone()[0]
-
-        if is_rds == 0:
-            wal_sender_timeout = wal_sender_timeout
-            LOGGER.info("Set session wal_sender_timeout = {} seconds".format(wal_sender_timeout))
-        else:
-            wal_sender_timeout = wal_sender_timeout * 1000
-            LOGGER.info("Set session wal_sender_timeout = {} milliseconds".format(wal_sender_timeout))
-
+        wal_sender_timeout = 10800000 #10800000ms = 3 hours
+        LOGGER.info("Set session wal_sender_timeout = {} milliseconds".format(wal_sender_timeout))
         cur.execute("SET SESSION wal_sender_timeout = {}".format(wal_sender_timeout))
 
     try: