45 lines
1.2 KiB
Ruby
45 lines
1.2 KiB
Ruby
|
class MigrateSubscriptionsProjectId < ActiveRecord::Migration
|
||
|
|
include Gitlab::Database::MigrationHelpers
|
||
|
|
|
||
|
|
DOWNTIME = true
|
||
|
|
DOWNTIME_REASON = 'Subscriptions will not work as expected until this migration is complete.'
|
||
|
|
|
||
|
|
def up
|
||
|
|
execute <<-EOF.strip_heredoc
|
||
|
|
UPDATE subscriptions
|
||
|
|
SET project_id = (
|
||
|
|
SELECT issues.project_id
|
||
|
|
FROM issues
|
||
|
|
WHERE issues.id = subscriptions.subscribable_id
|
||
|
|
)
|
||
|
|
WHERE subscriptions.subscribable_type = 'Issue';
|
||
|
|
EOF
|
||
|
|
|
||
|
|
execute <<-EOF.strip_heredoc
|
||
|
|
UPDATE subscriptions
|
||
|
|
SET project_id = (
|
||
|
|
SELECT merge_requests.target_project_id
|
||
|
|
FROM merge_requests
|
||
|
|
WHERE merge_requests.id = subscriptions.subscribable_id
|
||
|
|
)
|
||
|
|
WHERE subscriptions.subscribable_type = 'MergeRequest';
|
||
|
|
EOF
|
||
|
|
|
||
|
|
execute <<-EOF.strip_heredoc
|
||
|
|
UPDATE subscriptions
|
||
|
|
SET project_id = (
|
||
|
|
SELECT projects.id
|
||
|
|
FROM labels INNER JOIN projects ON projects.id = labels.project_id
|
||
|
|
WHERE labels.id = subscriptions.subscribable_id
|
||
|
|
)
|
||
|
|
WHERE subscriptions.subscribable_type = 'Label';
|
||
|
|
EOF
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
execute <<-EOF.strip_heredoc
|
||
|
|
UPDATE subscriptions SET project_id = NULL;
|
||
|
|
EOF
|
||
|
|
end
|
||
|
|
end
|