class PactBroker::DB::DataMigrations::SetConsumerIdsForPactPublications

Public Class Methods

call(connection) click to toggle source
# File lib/pact_broker/db/data_migrations/set_consumer_ids_for_pact_publications.rb, line 7
def self.call connection
  if columns_exist?(connection)
    query = "UPDATE pact_publications
            SET consumer_id = (SELECT pacticipant_id
              FROM versions
              WHERE id = pact_publications.consumer_version_id)
            WHERE consumer_id is null"
    connection.run(query)
  end
end
column_exists?(connection, table, column) click to toggle source
# File lib/pact_broker/db/data_migrations/set_consumer_ids_for_pact_publications.rb, line 25
def self.column_exists?(connection, table, column)
  connection.table_exists?(table) && connection.schema(table).find{|col| col.first == column }
end
columns_exist?(connection) click to toggle source
# File lib/pact_broker/db/data_migrations/set_consumer_ids_for_pact_publications.rb, line 18
def self.columns_exist?(connection)
  column_exists?(connection, :pact_publications, :consumer_id) &&
    column_exists?(connection, :pact_publications, :id) &&
    column_exists?(connection, :versions, :id) &&
    column_exists?(connection, :versions, :pacticipant_id)
end