pg_easy_replicate 0.3.5 → 0.3.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4edfae292a9e881e90c5a845c46cb830e2335dcf4bc3f98cffeb8518a8db8326
-  data.tar.gz: 0b4c27643bd2d7e215e6cf0377475f6c2f02f0eb64686107ce921bf5e499a04d
+  metadata.gz: e7291bb160de0db280d2e11a7b5e46d89bdd6b259ea2acf070ab302a80b27579
+  data.tar.gz: 7b5f1df7f16a833ce6f9fc2bdac7765e0d8835b6b271d029834a06a0f21bd3b7
 SHA512:
-  metadata.gz: 184982dc8ebf187b5a130ddd1ec142e6c569382a092966bf45cd2c83df619cb0bf718ad64b53c97cd5ec38cbf3a7526fe4b4b60ef8ab1a9b65438b305e6b8751
-  data.tar.gz: dc8faa56515d1c43eab8b7f220c6e64e368c99a1ae83da6bb7dcbc01609337ab1e6d9ae142668e2b39231013cb42609ea1d39ad5b904f7935593cca539d89f7f
+  metadata.gz: a066f7421b3be12ad1aa28017c9c680652f18477a37390fdaaf58efe2dc31aca28abc40715f15e33afbf628c973c2d8786d6d2818e22ea59e9ee6ca9c74dccd8
+  data.tar.gz: 45b5cfa989e4209ceddacc2582088ff712e45a17d34bfff423c30d3d350635cc894cc0f6c84dc5a68cb96ec4f0280d16cc8a2d4b108296543cde095b752312d6

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    pg_easy_replicate (0.3.5)
+    pg_easy_replicate (0.3.6)
       ougai (~> 2.0.0)
       pg (~> 1.5.3)
       pg_query (~> 5.1.0)

data/lib/pg_easy_replicate/cli.rb

@@ -78,6 +78,11 @@ module PgEasyReplicate
                   aliases: "-s",
                   desc:
                     "Cleans up the publication and subscription for the respective group"
+    method_option :restore_connection_on_source_db,
+                  aliases: "-r",
+                  type: :boolean,
+                  default: false,
+                  desc: "Restore connection on source db after switchover"
     def cleanup
       PgEasyReplicate.cleanup(options)
     end
@@ -163,14 +168,34 @@ module PgEasyReplicate
       end
     end
 
-    desc "notify", "Sends a notification with replication status to a specified url"
-    method_option :group_name, aliases: "-g", required: true, desc: "Name of the group previously provisioned"
-    method_option :url, aliases: "-u", required: true, desc: "URL for notification"
-    method_option :frequency, aliases: "-f", type: :numeric, default: 10, desc: "Frequency for sending stats to the endpoint provided"
-    method_option :timeout, aliases: "-t", type: :numeric, default: 10, desc: "Timeout for the notify request"
+    desc "notify",
+         "Sends a notification with replication status to a specified url"
+    method_option :group_name,
+                  aliases: "-g",
+                  required: true,
+                  desc: "Name of the group previously provisioned"
+    method_option :url,
+                  aliases: "-u",
+                  required: true,
+                  desc: "URL for notification"
+    method_option :frequency,
+                  aliases: "-f",
+                  type: :numeric,
+                  default: 10,
+                  desc: "Frequency for sending stats to the endpoint provided"
+    method_option :timeout,
+                  aliases: "-t",
+                  type: :numeric,
+                  default: 10,
+                  desc: "Timeout for the notify request"
 
     def notify
-      PgEasyReplicate::Stats.notify(options[:group_name], options[:url], options[:frequency], options[:timeout])
+      PgEasyReplicate::Stats.notify(
+        options[:group_name],
+        options[:url],
+        options[:frequency],
+        options[:timeout],
+      )
     end
 
     desc "list_ddl_changes", "Lists recent DDL changes in the source database"

data/lib/pg_easy_replicate/group.rb

@@ -20,7 +20,7 @@ module PgEasyReplicate
           column(:updated_at, Time, default: Sequel::CURRENT_TIMESTAMP)
           column(:started_at, Time)
           column(:failed_at, Time)
-          column(:recreate_indices_post_copy, TrueClass, default: true)
+          column(:recreate_indices_post_copy, TrueClass, default: false)
           column(:switchover_completed_at, Time)
         end
       ensure
@@ -44,6 +44,7 @@ module PgEasyReplicate
           schema_name: options[:schema_name],
           started_at: options[:started_at],
           failed_at: options[:failed_at],
+          recreate_indices_post_copy: options[:recreate_indices_post_copy],
         )
       rescue => e
         abort_with("Adding group entry failed: #{e.message}")

data/lib/pg_easy_replicate/helper.rb

@@ -75,7 +75,7 @@ module PgEasyReplicate
 
     def determine_tables(conn_string:, list: "", exclude_list: "", schema: nil)
       schema ||= "public"
-      
+
       tables = convert_to_array(list)
       exclude_tables = convert_to_array(exclude_list)
       validate_table_lists(tables, exclude_tables, schema)
@@ -115,14 +115,25 @@ module PgEasyReplicate
       exclude_table_list = convert_to_array(exclude_tables)
 
       if !table_list.empty? && !exclude_table_list.empty?
-        abort_with("Options --tables(-t) and --exclude-tables(-e) cannot be used together.")
+        abort_with(
+          "Options --tables(-t) and --exclude-tables(-e) cannot be used together.",
+        )
       elsif !table_list.empty?
         if table_list.size > 0 && (schema_name.nil? || schema_name == "")
           abort_with("Schema name is required if tables are passed")
         end
-      elsif exclude_table_list.size > 0 && (schema_name.nil? || schema_name == "")
+      elsif exclude_table_list.size > 0 &&
+            (schema_name.nil? || schema_name == "")
         abort_with("Schema name is required if exclude tables are passed")
       end
     end
+
+    def restore_connections_on_source_db
+      logger.info("Restoring connections")
+
+      alter_sql =
+        "ALTER USER #{quote_ident(db_user(source_db_url))} set default_transaction_read_only = false"
+      Query.run(query: alter_sql, connection_url: source_db_url)
+    end
   end
 end

data/lib/pg_easy_replicate/orchestrate.rb

@@ -254,7 +254,7 @@ module PgEasyReplicate
           )
         end
       rescue => e
-        restore_connections_on_source_db(group_name)
+        restore_connections_on_source_db
         abort_with("Switchover failed: #{e.message}")
       end
 
@@ -314,14 +314,6 @@ module PgEasyReplicate
         raise "Unable to revoke connections on source db: #{e.message}"
       end
 
-      def restore_connections_on_source_db(group_name)
-        logger.info("Restoring connections")
-
-        alter_sql =
-          "ALTER USER #{quote_ident(db_user(source_db_url))} set default_transaction_read_only = false"
-        Query.run(query: alter_sql, connection_url: source_db_url)
-      end
-
       def refresh_sequences(conn_string:, schema: nil)
         logger.info("Refreshing sequences")
         sql = <<~SQL

data/lib/pg_easy_replicate/version.rb

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

data/lib/pg_easy_replicate.rb

@@ -213,6 +213,11 @@ module PgEasyReplicate
             end
           end
         end,
+        -> do
+          if options[:restore_connection_on_source_db]
+            restore_connections_on_source_db
+          end
+        end,
       ]
 
       cleanup_steps.each do |step|

data/scripts/release.sh

@@ -14,9 +14,9 @@ gem push pg_easy_replicate-"$VERSION".gem
 echo "=== Sleeping for 15s ===="
 sleep 15
 
-# echo "=== Pushing tags to github ===="
-# git tag v"$VERSION"
-# git push origin --tags
+echo "=== Pushing tags to github ===="
+git tag v"$VERSION"
+git push origin --tags
 
 echo "=== Cleaning up ===="
 rm pg_easy_replicate-"$VERSION".gem

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_easy_replicate
 version: !ruby/object:Gem::Version
-  version: 0.3.5
+  version: 0.3.6
 platform: ruby
 authors:
 - Shayon Mukherjee
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-11-18 00:00:00.000000000 Z
+date: 2024-11-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ougai