Fixes statement_timeout unit handling on int->bigint migration. Refs #20074

[arvados.git] / services / api / db / migrate / 20221230155924_bigint_id.rb

diff --git a/services/api/db/migrate/20221230155924_bigint_id.rb b/services/api/db/migrate/20221230155924_bigint_id.rb
index 20791ae017e41069396fb127b86102644e6ac7b0..932cb025dc337c6e4bd8ddae7f23949f85ad6a21 100644 (file)
--- a/services/api/db/migrate/20221230155924_bigint_id.rb
+++ b/services/api/db/migrate/20221230155924_bigint_id.rb
@@ -5,6 +5,9 @@
 class BigintId < ActiveRecord::Migration[5.2]
   disable_ddl_transaction!
   def up
+    old_value = query_value('SHOW statement_timeout')
+    execute "SET statement_timeout TO '0'"
+
     change_column :api_client_authorizations, :id, :bigint
     change_column :api_client_authorizations, :api_client_id, :bigint
     change_column :api_client_authorizations, :user_id, :bigint
@@ -30,6 +33,8 @@ class BigintId < ActiveRecord::Migration[5.2]
     change_column :traits, :id, :bigint
     change_column :virtual_machines, :id, :bigint
     change_column :workflows, :id, :bigint
+
+    execute "SET statement_timeout TO #{quote(old_value)}"
   end
 
   def down