blob: 69d997044699251580e20042f5fc506dfed73ce4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
# 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
|