pg_easy_replicate 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: efcbf369a4410abd783ea1dba49ead97fa471705842df4a35d28864349b1c70c
-  data.tar.gz: 676d97aa0c551dd0f9d58c3bccb229951bcb2dfc34653995375c359718bc1d7f
+  metadata.gz: 86261e2b60e55c5770bbd1537d5a3a501a514d1eaf8e741c5c81890d864b0a0f
+  data.tar.gz: bf6a710930b4f9533d01c3f66e3b91c02ad5a422df9d17be2bbaf72cdb979a2e
 SHA512:
-  metadata.gz: 774ef530f4be4b409fdf2d4956a44deb09aebf59c8ef480074d426ca4c72006673c7a6915d860e61846fb00df8e14fa9534e920f5802873ed727e37564197b25
-  data.tar.gz: 50055941c3a83351b22ceaeb539b807a7cec5e3130732944c63e8a5440a7707a273a546681b1d581d4f79d9fba616998fc48cffc009e401bfc570e76e27064f0
+  metadata.gz: 5d7aab6e89948a45248fcb19fb6d3e9a6982adbc23cba094fe413b2a376cbffb1bbfdb011833f16c4c3aeb3c4d3e1192792699b4c360a20555fc05972e2ae111
+  data.tar.gz: a4c72947b466664dfc716a2ed21cab159d0f6768b3ae74c9c95db105bafb22da57244236a1f9677e1936af8140989dae9a0c5b1729c2f4bc62dfcbbad0d6a007

data/CHANGELOG.md

@@ -1,4 +1,8 @@
-## [0.2.1] - 2023-12-29
+## [0.2.1] - 2024-01-21
+
+- Extend config check to assert for REPLICA IDENTITY on tables and drop index bug - #88
+
+## [0.2.1] - 2024-01-20
 
 - Don't attempt to drop and recreate unique indices - #88
 - Dependency updates

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    pg_easy_replicate (0.2.2)
+    pg_easy_replicate (0.2.3)
       ougai (~> 2.0.0)
       pg (~> 1.5.3)
       sequel (>= 5.69, < 5.77)

data/README.md

@@ -4,7 +4,7 @@
 [![Smoke spec](https://github.com/shayonj/pg_easy_replicate/actions/workflows/smoke.yaml/badge.svg?branch=main)](https://github.com/shayonj/pg_easy_replicate/actions/workflows/ci.yaml)
 [![Gem Version](https://badge.fury.io/rb/pg_easy_replicate.svg?2)](https://badge.fury.io/rb/pg_easy_replicate)
 
-`pg_easy_replicate` is a CLI orchestrator tool that simplifies the process of setting up [logical replication](https://www.postgresql.org/docs/current/logical-replication.html) between two PostgreSQL databases. `pg_easy_replicate` also supports switchover. After the source (primary database) is fully replicated, `pg_easy_replicate` puts it into read-only mode and via logical replication flushes all data to the new target database. This ensures zero data loss and minimal downtime for the application. This method can be useful for performing minimal downtime (up to <1min, depending) major version upgrades between two PostgreSQL databases, load testing with blue/green database setup and other similar use cases.
+`pg_easy_replicate` is a CLI orchestrator tool that simplifies the process of setting up [logical replication](https://www.postgresql.org/docs/current/logical-replication.html) between two PostgreSQL databases. `pg_easy_replicate` also supports switchover. After the source (primary database) is fully replicated, `pg_easy_replicate` puts it into read-only mode and via logical replication flushes all data to the new target database. This ensures zero data loss and minimal downtime for the application. This method can be useful for performing minimal downtime (up to <1min, depending) major version upgrades between a Blue/Green PostgreSQL database setup, load testing and other similar use cases.
 
 Battle tested in production at [Tines](https://www.tines.com/) 🚀
 
@@ -19,7 +19,7 @@ Battle tested in production at [Tines](https://www.tines.com/) 🚀
   - [Config check](#config-check)
   - [Bootstrap](#bootstrap)
   - [Bootstrap and Config Check with special user role in AWS or GCP](#bootstrap-and-config-check-with-special-user-role-in-aws-or-gcp)
-    - [Config Check](#config-check)
+    - [Config Check](#config-check-1)
     - [Bootstrap](#bootstrap-1)
   - [Start sync](#start-sync)
   - [Stats](#stats)
@@ -29,7 +29,7 @@ Battle tested in production at [Tines](https://www.tines.com/) 🚀
   - [Rolling restart strategy](#rolling-restart-strategy)
   - [DNS Failover strategy](#dns-failover-strategy)
 - [FAQ](#faq)
-  - [Adding internal user to pgBouncer `userlist`](#adding-internal-user-to-pgbouncer-userlist)
+  - [Adding internal user to `pg_hba` or pgBouncer `userlist`](#adding-internal-user-to-pg_hba-or-pgbouncer-userlist)
 - [Contributing](#contributing)
 
 ## Installation
@@ -61,6 +61,7 @@ https://hub.docker.com/r/shayonj/pg_easy_replicate
 - PostgreSQL 10 and later
 - Ruby 3.0 and later
 - Database users should have `SUPERUSER` permissions, or pass in a special user with privileges to create the needed role, schema, publication and subscription on both databases. More on `--special-user-role` section below.
+- See more on [FAQ](#faq) below
 
 ## Limits
 
@@ -256,9 +257,9 @@ Next, you can set up a program that watches the `stats` and waits until `switcho
 
 ## FAQ
 
-### Adding internal user to pgBouncer `userlist`
+### Adding internal user to `pg_hba` or pgBouncer `userlist`
 
-`pg_easy_replicate` creates a special user to orchestrate the replication. If you use pgBouncer, you may need to allow `pger_su_h1a4fb` as a user that can perform login by adding it to the `userlist`.
+`pg_easy_replicate` sets up a designated user for managing the replication process. In case you handle user permissions through `pg_hba`, it's necessary to modify this list to permit sessions from `pger_su_h1a4fb`. Similarly, with pgBouncer, you'll need to authorize `pger_su_h1a4fb` for login access by including it in the `userlist`.
 
 ## Contributing
 

data/lib/pg_easy_replicate/cli.rb

@@ -18,6 +18,7 @@ module PgEasyReplicate
                   desc: "Copy schema to the new database"
     method_option :tables,
                   aliases: "-t",
+                  default: "",
                   desc:
                     "Comma separated list of table names. Default: All tables"
     method_option :schema_name,
@@ -87,6 +88,7 @@ module PgEasyReplicate
                     "Name of the schema tables are in, only required if passing list of tables"
     method_option :tables,
                   aliases: "-t",
+                  default: "",
                   desc:
                     "Comma separated list of table names. Default: All tables"
     method_option :recreate_indices_post_copy,
@@ -117,8 +119,12 @@ module PgEasyReplicate
                   desc: "Name of the group previously provisioned"
     method_option :lag_delta_size,
                   aliases: "-l",
-                  desc:
-                    "The size of the lag to watch for before switchover. Default 200KB."
+                  desc: "The size of the lag to watch for before switchover. Default 200KB."
+    method_option :skip_vacuum_analyze,
+                  type: :boolean,
+                  default: false,
+                  aliases: "-s",
+                  desc: "Skip vacuum analyzing tables before switchover."
     # method_option :bi_directional,
     #               aliases: "-b",
     #               desc:
@@ -127,6 +133,7 @@ module PgEasyReplicate
       PgEasyReplicate::Orchestrate.switchover(
         group_name: options[:group_name],
         lag_delta_size: options[:lag_delta_size],
+        skip_vacuum_analyze: options[:skip_vacuum_analyze]
       )
     end
 

data/lib/pg_easy_replicate/orchestrate.rb

@@ -204,16 +204,19 @@ module PgEasyReplicate
         group_name:,
         source_conn_string: source_db_url,
         target_conn_string: target_db_url,
-        lag_delta_size: nil
+        lag_delta_size: nil,
+        skip_vacuum_analyze: false
       )
         group = Group.find(group_name)
         tables_list = group[:table_names].split(",")
 
-        run_vacuum_analyze(
-          conn_string: target_conn_string,
-          tables: tables_list,
-          schema: group[:schema_name],
-        )
+        unless skip_vacuum_analyze
+          run_vacuum_analyze(
+            conn_string: target_conn_string,
+            tables: tables_list,
+            schema: group[:schema_name],
+          )
+        end
 
         watch_lag(group_name: group_name, lag: lag_delta_size || DEFAULT_LAG)
 
@@ -238,11 +241,13 @@ module PgEasyReplicate
         )
         mark_switchover_complete(group_name)
         # Run vacuum analyze to refresh the planner post switchover
-        run_vacuum_analyze(
-          conn_string: target_conn_string,
-          tables: tables_list,
-          schema: group[:schema_name],
-        )
+        unless skip_vacuum_analyze
+          run_vacuum_analyze(
+            conn_string: target_conn_string,
+            tables: tables_list,
+            schema: group[:schema_name],
+          )
+        end
         drop_subscription(
           group_name: group_name,
           target_conn_string: target_conn_string,
@@ -358,11 +363,13 @@ module PgEasyReplicate
             schema: schema,
             table: t,
           )
+
           Query.run(
-            query: "VACUUM VERBOSE ANALYZE #{t}",
+            query: "VACUUM VERBOSE ANALYZE #{t};",
             connection_url: conn_string,
             schema: schema,
             transaction: false,
+            using_vacuum_analyze: true,
           )
         end
       rescue => e

data/lib/pg_easy_replicate/query.rb

@@ -10,20 +10,26 @@ module PgEasyReplicate
         connection_url:,
         user: internal_user_name,
         schema: nil,
-        transaction: true
+        transaction: true,
+        using_vacuum_analyze: false
       )
         conn =
           connect(connection_url: connection_url, schema: schema, user: user)
+        timeout ||= "5s"
         if transaction
           r =
             conn.transaction do
               conn.run("SET search_path to #{quote_ident(schema)}") if schema
-              conn.run("SET statement_timeout to '5s'")
+              conn.run("SET statement_timeout to '#{timeout}'")
               conn.fetch(query).to_a
             end
         else
           conn.run("SET search_path to #{quote_ident(schema)}") if schema
-          conn.run("SET statement_timeout to '5s'")
+          if using_vacuum_analyze
+            conn.run("SET statement_timeout=0")
+          else
+            conn.run("SET statement_timeout to '5s'")
+          end
           r = conn.fetch(query).to_a
         end
         conn.disconnect

data/lib/pg_easy_replicate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module PgEasyReplicate
-  VERSION = "0.2.2"
+  VERSION = "0.2.3"
 end

data/scripts/e2e-start.sh

@@ -10,7 +10,7 @@ export SOURCE_DB_URL="postgres://james-bond:james-bond123%407%21%273aaR@localhos
 export TARGET_DB_URL="postgres://james-bond:james-bond123%407%21%273aaR@localhost:5433/postgres-db"
 export PGPASSWORD='james-bond123@7!'"'"''"'"'3aaR'
 
-# Bootstrap and cleanup
+# Config check, Bootstrap and cleanup
 echo "===== Performing Bootstrap and cleanup"
 bundle exec bin/pg_easy_replicate bootstrap -g cluster-1 --copy-schema
 bundle exec bin/pg_easy_replicate start_sync -g cluster-1 -s public --recreate-indices-post-copy

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_easy_replicate
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Shayon Mukherjee
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-01-21 00:00:00.000000000 Z
+date: 2024-01-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ougai