Jtm migration for bigints

app/models/study_type_answer.rb

@@ -20,4 +20,5 @@
 
 class StudyTypeAnswer < ApplicationRecord
   belongs_to :study_type_question
+  belongs_to :protocol
 end

app/models/subsidy.rb

@@ -22,6 +22,7 @@ class Subsidy < ApplicationRecord
   audited
 
   belongs_to :sub_service_request
+  belongs_to :identity, foreign_key: 'approved_by'
   has_many :notes, as: :notable
 
   delegate :organization, :direct_cost_total, to: :sub_service_request, allow_nil: true

db/migrate/20171106161235_change_ids_from_int_to_bigint.rb

@@ -0,0 +1,14 @@
+class ChangeIdsFromIntToBigint < ActiveRecord::Migration[5.1]
+  def change
+    # moved this to take lib/tasks/migrate_ids_to_bigint.rake since it breaks when model changes are made
+    # might also be nice to be able to run this again
+    say "#"*50
+    say "#"*50
+    say ""
+    say "This migration has been moved to lib/tasks/migrate_ids_to_bigint.rake"
+    say "Usage: rake migrate_ids_to_bigint"
+    say ""
+    say "#"*50
+    say "#"*50
+  end
+end

lib/tasks/migrate_ids_to_bigint.rake

@@ -0,0 +1,97 @@
+# Copyright © 2011-2018 MUSC Foundation for Research Development~
+# All rights reserved.~
+
+# Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:~
+
+# 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.~
+
+# 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following~
+# disclaimer in the documentation and/or other materials provided with the distribution.~
+
+# 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products~
+# derived from this software without specific prior written permission.~
+
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,~
+# BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT~
+# SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL~
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS~
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR~
+# TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.~
+
+desc "convert ID column to bigint"
+task :migrate_ids_to_bigint => :environment do
+  db_models, db_habtms = map_models_to_tablenames
+  non_ar_tables = (ActiveRecord::Base.connection.tables - db_models.keys - db_habtms)
+  references = Hash.new{ |h, k| h[k] = [] }
+  foreign_keys = Hash.new{ |h, k| h[k] = [] }
+  ActiveRecord::Base.transaction do
+
+    db_models.except("documents").each do |table_name, model|
+      fks = ActiveRecord::Base.connection.foreign_keys(table_name)
+      foreign_keys[table_name] = fks if fks.present?
+      references[table_name] = get_references(model) unless get_references(model).empty?
+    end
+
+    foreign_keys.each do |table_name, fks|
+      fks.each do |foreign_key|
+        ActiveRecord::Base.connection.remove_foreign_key table_name, name: foreign_key.name
+      end
+    end
+
+    db_models.each do |table_name, model|
+      ActiveRecord::Base.connection.change_column table_name, model.primary_key, :bigint, auto_increment: true if column_is_integer? model, model.primary_key
+    end
+
+    references.each do |table_name, references|
+      references.each do |column_name|
+        ActiveRecord::Base.connection.change_column table_name, column_name, :bigint
+      end
+    end
+
+    foreign_keys.each do |table_name, fks|
+      fks.each do |foreign_key|
+        ActiveRecord::Base.connection.add_foreign_key table_name, foreign_key.to_table, column: foreign_key.column, primary_key: foreign_key.primary_key
+      end
+    end
+
+    db_habtms.each do |table_name|
+      columns = ActiveRecord::Base.connection.columns(table_name)
+      columns.each do |column|
+        ActiveRecord::Base.connection.change_column table_name, column.name, :bigint if column.type == :integer
+      end
+    end
+
+    non_ar_tables.each do |table_name|
+      sql_result = ActiveRecord::Base.connection.exec_query("SHOW COLUMNS FROM #{table_name}")
+      key_index = sql_result.columns.find_index("Key")
+      type_index = sql_result.columns.find_index("Type")
+      name_index = sql_result.columns.find_index("Field")
+      reference_columns = sql_result.rows.select{ |row| row[key_index].present? }
+      reference_columns.each do |column|
+        ActiveRecord::Base.connection.change_column table_name, column[name_index], :bigint if column[type_index] == "int(11)"
+      end
+    end
+  end
+end
+
+def map_models_to_tablenames
+  Rails.application.eager_load!
+
+  tables = ActiveRecord::Base.connection.tables
+  models = ActiveRecord::Base.descendants
+  db_habtms = models.map{ |model| model.reflect_on_all_associations(:has_and_belongs_to_many) }.compact.flatten.map{ |reflection| reflection.join_table }.uniq
+  db_models = models.group_by(&:table_name).slice(*tables).except(*db_habtms)
+  db_models.each{ |table_name, models| db_models[table_name] = models.first }
+  return db_models, db_habtms
+end
+
+def get_references model
+  bt_associations = model.reflect_on_all_associations.select do |association|
+    association.foreign_key.present? && model.columns_hash[association.foreign_key.to_s].present? && column_is_integer?(model, association.foreign_key.to_s)
+  end
+  bt_associations.map{ |association| association.foreign_key.to_s }
+end
+
+def column_is_integer? model, name
+  model.columns_hash[name].type == :integer
+end