55 lines
1.4 KiB
Ruby
55 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class UpdateGitlabSubscriptionsStartAtPostEoa < ActiveRecord::Migration[6.0]
|
|
UPDATE_BATCH_SIZE = 100
|
|
|
|
disable_ddl_transaction!
|
|
|
|
class Plan < ActiveRecord::Base
|
|
self.table_name = 'plans'
|
|
self.inheritance_column = :_type_disabled
|
|
end
|
|
|
|
class GitlabSubscription < ActiveRecord::Base
|
|
include EachBatch
|
|
|
|
self.table_name = 'gitlab_subscriptions'
|
|
self.inheritance_column = :_type_disabled
|
|
|
|
EOA_ROLLOUT_DATE = '2021-01-26'
|
|
|
|
scope :with_plan, -> (from_plan) do
|
|
where("start_date >= ? AND hosted_plan_id = ?", EOA_ROLLOUT_DATE, from_plan.id)
|
|
end
|
|
end
|
|
|
|
def up
|
|
return unless Gitlab.com?
|
|
|
|
silver_plan = Plan.find_by(name: 'silver')
|
|
gold_plan = Plan.find_by(name: 'gold')
|
|
premium_plan = Plan.find_by(name: 'premium')
|
|
ultimate_plan = Plan.find_by(name: 'ultimate')
|
|
|
|
# Silver to Premium
|
|
update_hosted_plan_for_subscription(from_plan: silver_plan, to_plan: premium_plan)
|
|
|
|
# Gold to Ultimate
|
|
update_hosted_plan_for_subscription(from_plan: gold_plan, to_plan: ultimate_plan)
|
|
end
|
|
|
|
def down
|
|
# no-op
|
|
end
|
|
|
|
private
|
|
|
|
def update_hosted_plan_for_subscription(from_plan:, to_plan:)
|
|
return unless from_plan && to_plan
|
|
|
|
GitlabSubscription.with_plan(from_plan).each_batch(of: UPDATE_BATCH_SIZE) do |batch|
|
|
batch.update_all(hosted_plan_id: to_plan.id)
|
|
end
|
|
end
|
|
end
|