pg_online_schema_change 0.7.2 → 0.7.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d9b2ebcfd1944ac2ddba65466ca474751637d6fdb2db7abcf3a5d587e716c617
-  data.tar.gz: 3a6ac8b49967e9483f75359006097865de640c3ce414dad5bd0e07e766d4a3c3
+  metadata.gz: 70f01145ab63bcb06e4aea9e79b8dc5ac1bf5cc37749964589d2dc940214d8c4
+  data.tar.gz: 52f2b7e5fe1b793f1f8a414972d282a41661eb0045b65d095b65b4c9ab2d6a2a
 SHA512:
-  metadata.gz: a0a7d0bf9bf030859d6b7cb36b1d69770d3af9c7721c5e5042732887959bc1664e1c8bd4f09f076cdafcb1b272a57f49ddf1fbb5d86aad60a4b0b81d79378804
-  data.tar.gz: bb12999f436c8d93591c17b97f5f508d7234355ef25780834565a8222da4c43dd3dfd52450210007bc3f0a32cd0231958a3e7645af538d1af67cc3e620582f14
+  metadata.gz: d4d0c36d71d4aa8020063c63bba2358ffd2d088ad9da21ce3419a8916acbd74c80dfefd5a134423e708ef6b72f444dcbf0ebfd6ba4ff26956d562c258a94fbcc
+  data.tar.gz: c73369d1a69a0824474d5a930411b2f2eee0e7ac61986fb7b8df47d84e010da2cb35fec6e054e5e8986e6877f1992f256b420404a41f1cdd85620f8f1b9a3bb1

data/.rubocop.yml

@@ -24,7 +24,7 @@ Layout/SpaceAroundEqualsInParameterDefault:
 
 Metrics/AbcSize:
   Enabled: true
-  Max: 40
+  Max: 45
   Exclude:
     - "spec/**/*"
 

data/.rubocop_todo.yml

@@ -1,6 +1,6 @@
 # This configuration was generated by
 # `rubocop --auto-gen-config`
-# on 2022-03-13 19:35:49 UTC using RuboCop version 1.23.0.
+# on 2022-09-24 14:22:44 UTC using RuboCop version 1.23.0.
 # The point is for the user to remove these configuration records
 # one by one as the offenses are removed from the code base.
 # Note that changes in the inspected code, or installation of new
@@ -9,7 +9,7 @@
 # Offense count: 2
 # Configuration parameters: CountComments, CountAsOne.
 Metrics/ClassLength:
-  Max: 250
+  Max: 266
 
 # Offense count: 2
 # Configuration parameters: IgnoredMethods.
@@ -26,7 +26,7 @@ Packaging/GemspecGit:
   Exclude:
     - 'pg_online_schema_change.gemspec'
 
-# Offense count: 67
+# Offense count: 68
 # Configuration parameters: CountAsOne.
 RSpec/ExampleLength:
   Max: 55

data/CHANGELOG.md

@@ -1,4 +1,8 @@
 
+## [0.7.2] - 2022-09-17
+* Update primary key sequence on shadow table https://github.com/shayonj/pg-osc/pull/72
+  - Thanks to @brycethornton for the report
+
 ## [0.7.1] - 2022-03-13
 * Bump pg to `1.3.4` https://github.com/shayonj/pg-osc/commit/d086c19d4f273dc960491cbedf9e0602d812896b
 

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    pg_online_schema_change (0.7.2)
+    pg_online_schema_change (0.7.3)
       ougai (~> 2.0.0)
       pg (~> 1.3.2)
       pg_query (~> 2.1.3)

data/README.md

@@ -260,6 +260,16 @@ rvm use 3.0.0
 
 To install this gem onto your local machine, run `bundle exec rake install`. 
 
+### Local testing 
+```
+docker compose up
+
+pgbench --initialize -s 10 --foreign-keys --host localhost -U jamesbond -d postgres
+pgbench -T 60000 -c 5 --host localhost -U jamesbond -d postgres
+
+bundle exec bin/pg-online-schema-change perform -a 'ALTER TABLE pgbench_accounts ALTER COLUMN aid TYPE BIGINT' -d "postgres" -h "localhost" -u "jamesbond" -w "password"
+```
+
 ## Releasing
 
 - Bump version in `version.rb`

data/lib/pg_online_schema_change/orchestrate.rb

@@ -238,7 +238,10 @@ module PgOnlineSchemaChange
         rows = Replay.rows_to_play(opened)
         Replay.play!(rows, opened)
 
+        query_for_primary_key_refresh = Query.query_for_primary_key_refresh(shadow_table, primary_key, client.table, opened)
+
         sql = <<~SQL
+          #{query_for_primary_key_refresh};
           ALTER TABLE #{client.table} RENAME to #{old_primary_table};
           ALTER TABLE #{shadow_table} RENAME to #{client.table};
           #{referential_foreign_key_statements}

data/lib/pg_online_schema_change/query.rb

@@ -319,6 +319,28 @@ module PgOnlineSchemaChange
           FROM ONLY #{client.table}
         SQL
       end
+
+      def primary_key_sequence(shadow_table, primary_key, opened)
+        query = <<~SQL
+          SELECT pg_get_serial_sequence(\'#{shadow_table}\', \'#{primary_key}\') as sequence_name
+        SQL
+
+        result = run(client.connection, query, opened)
+
+        result.map do |row|
+          row["sequence_name"]
+        end&.first
+      end
+
+      def query_for_primary_key_refresh(shadow_table, primary_key, table, opened)
+        sequence_name = primary_key_sequence(shadow_table, primary_key, opened)
+
+        return "" if sequence_name.nil?
+
+        <<~SQL
+          SELECT setval((select pg_get_serial_sequence(\'#{shadow_table}\', \'#{primary_key}\')), (SELECT max(#{primary_key}) FROM #{table})+1);
+        SQL
+      end
     end
   end
 end

data/lib/pg_online_schema_change/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module PgOnlineSchemaChange
-  VERSION = "0.7.2"
+  VERSION = "0.7.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_online_schema_change
 version: !ruby/object:Gem::Version
-  version: 0.7.2
+  version: 0.7.3
 platform: ruby
 authors:
 - Shayon Mukherjee
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-09-17 00:00:00.000000000 Z
+date: 2022-09-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ougai