debian-mirror-gitlab/db/migrate/20161207231620_fixup_environment_name_uniqueness.rb

54 lines
1.6 KiB
Ruby
Raw Normal View History

2017-08-17 22:00:37 +05:30
class FixupEnvironmentNameUniqueness < ActiveRecord::Migration
include Gitlab::Database::MigrationHelpers
DOWNTIME = true
DOWNTIME_REASON = 'Renaming non-unique environments'
def up
environments = Arel::Table.new(:environments)
# Get all [project_id, name] pairs that occur more than once
2017-09-10 17:25:29 +05:30
finder_sql = environments
.group(environments[:project_id], environments[:name])
.having(Arel.sql("COUNT(1)").gt(1))
.project(environments[:project_id], environments[:name])
.to_sql
2017-08-17 22:00:37 +05:30
conflicting = connection.exec_query(finder_sql)
conflicting.rows.each do |project_id, name|
fix_duplicates(project_id, name)
end
end
def down
# Nothing to do
end
# Rename conflicting environments by appending "-#{id}" to all but the first
def fix_duplicates(project_id, name)
environments = Arel::Table.new(:environments)
2017-09-10 17:25:29 +05:30
finder_sql = environments
.where(environments[:project_id].eq(project_id))
.where(environments[:name].eq(name))
.order(environments[:id].asc)
.project(environments[:id], environments[:name])
.to_sql
2017-08-17 22:00:37 +05:30
# Now we have the data for all the conflicting rows
conflicts = connection.exec_query(finder_sql).rows
conflicts.shift # Leave the first row alone
conflicts.each do |id, name|
update_sql =
2017-09-10 17:25:29 +05:30
Arel::UpdateManager.new(ActiveRecord::Base)
.table(environments)
.set(environments[:name] => name + "-" + id.to_s)
.where(environments[:id].eq(id))
.to_sql
2017-08-17 22:00:37 +05:30
connection.exec_update(update_sql, self.class.name, [])
end
end
end