Session wal_sender_timeout for PG12 (#43)

transferwise · Mar 10, 2020 · 68e8cb5 · 68e8cb5
1 parent ecf04e5
commit 68e8cb5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 14 deletions.
diff --git 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
@@ -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: