pg_logical_replicator 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2aab5bea887e74a1be5a02da58eb98b525c7ce3c786c7943136332f1b953c203
-  data.tar.gz: 201bed281909128a3bf8569360eb30f2d8d85601d9696207fd31919df5a45155
+  metadata.gz: d85998326734611fffa7a261b4da9e7aaa23cb0fc8e460c3e2e10206e8c2012d
+  data.tar.gz: 375f48b3a024bdcedb8a10c2752c60ea8d4acc5305dc0123d6ab993f6da5d185
 SHA512:
-  metadata.gz: 0d41c1e2e0e75e51ed0dccaeffd47efdb6fd7900c720ab0203e9ac7a2c0247b3d73071b3ae91fff3d29b4b9fd3b32b1598cf7be3dfa3c59248ee75c4e685f4a7
-  data.tar.gz: 9bfa6c1923a933b4ac6d5c8c094da0de2a2230cea61b4d1957f38ecfc9ea8916afe1a4619e32f8abe4711ecd165dfdc833007546644397440a0656f04c1aba36
+  metadata.gz: 387f5aa4609c1d69f11eb996fe4ee64890178ced6b7fb9a1f53eaad0233f9050c941362dc51c1b73e7d1133bef7058898ad1375b6550553048a011d224a27eb9
+  data.tar.gz: a3a0fe506552ecd598d4a31f7958d24ea6f257fe3e073136143b3252ea24a16c57d7e517577a7221c6e994704a4b4947c58fb7ad6b1de979fd36e6906348cd43

data/lib/pg_logical_replicator/cli.rb

@@ -19,7 +19,7 @@ module PgLogicalReplicator
     method_option :target_rep_password, type: :string
     method_option :num_slots, type: :numeric, default: 10
     method_option :groups, type: :array
-
+    
     def setup
       source_host         = options[:source_host]
       source_port         = options[:source_port]
@@ -35,11 +35,11 @@ module PgLogicalReplicator
       target_rep_password = options[:target_rep_password] || target_password
       num_slots           = options[:num_slots]
       target_groups       = options[:groups].map(&:to_i) if options[:groups]
-
+      
       conn = PG.connect(dbname: source_database, user: source_username, password: source_password, host: source_host, port: source_port)
-
+      
       query = "SELECT tablename FROM pg_tables WHERE schemaname = 'public' order by tablename ASC;"
-
+      
       tables = begin
                  result = conn.exec(query)
                  result.map { |row| row['tablename'] }
@@ -47,33 +47,33 @@ module PgLogicalReplicator
                  conn.close if conn
                end
       puts "Total Tables: #{tables.size}"
-
+      
       tables_per_group = tables.size / num_slots
       puts "Tables per group: #{tables_per_group}"
-
+      
       dump_dir_root = "~/db-dumps-#{Time.now.strftime('%Y%m%d%H%M%S')}"
       system("mkdir -p #{dump_dir_root}")
-
+      
       puts "Target Groups: #{target_groups}"
-
+      
       tables.each_slice(tables_per_group).with_index do |table_group, group_idx|
         puts "Processing group #{group_idx + 1} size: #{table_group.size}"
-
+        
         group_number = group_idx + 1
-
+        
         next unless target_groups.nil? || target_groups.include?(group_number)
-
+        
         table_names = table_group.join(',')
-
+        
         dump_dir = "#{dump_dir_root}/#{group_number}"
-
+        
         puts "Removing directory: #{dump_dir}"
-
+        
         FileUtils.rm_rf(dump_dir)
-
+        
         puts "Creating directory: #{dump_dir}"
         FileUtils.mkdir_p(dump_dir)
-
+        
         LogicalReplicationInitializer.new({
                                             slot_name:           "logical_sub_grp_#{group_number}",
                                             publication_name:    "logical_pub_grp_#{group_number}",
@@ -83,9 +83,13 @@ module PgLogicalReplicator
                                             table_names:         table_names,
                                             dump_dir:            dump_dir
                                           }).start
+      rescue StandardError => e
+        puts "Error: #{e.message}"
+      ensure
+        FileUtils.rm_rf(dump_dir)
       end
     end
-
+    
     desc 'stop_replication', 'Stop all replication'
     method_option :source_host, type: :string, required: true
     method_option :source_port, type: :numeric, default: 5432
@@ -97,7 +101,7 @@ module PgLogicalReplicator
     method_option :source_password, type: :string, required: true
     method_option :target_username, type: :string
     method_option :target_password, type: :string
-
+    
     def stop_replication
       source_config = {
         host:     options[:source_host],
@@ -106,7 +110,7 @@ module PgLogicalReplicator
         user:     options[:source_username],
         password: options[:source_password]
       }
-
+      
       target_config = {
         host:     options[:target_host],
         port:     options[:target_port],
@@ -114,10 +118,10 @@ module PgLogicalReplicator
         user:     options[:target_username] || options[:source_username],
         password: options[:target_password] || options[:source_password]
       }
-
+      
       ReplicationStopper.new(source_config, target_config).stop_replication
     end
-
+    
     desc 'transfer_schema', 'Transfer schema from source to target database'
     method_option :source_host, type: :string, required: true
     method_option :source_port, type: :numeric, default: 5432
@@ -129,7 +133,7 @@ module PgLogicalReplicator
     method_option :source_password, type: :string, required: true
     method_option :target_username, type: :string, required: true
     method_option :target_password, type: :string, required: true
-
+    
     def transfer_schema
       SchemaTransfer.new(
         source_host:     options[:source_host],

data/lib/pg_logical_replicator/version.rb

@@ -1,3 +1,3 @@
 module PgLogicalReplicator
-  VERSION = "0.1.5"
+  VERSION = "0.1.6"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_logical_replicator
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - eni9889
@@ -118,7 +118,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3.1
+rubygems_version: 3.4.10
 signing_key:
 specification_version: 4
 summary: PostgreSQL logical replication setup tool