diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2020-02-12 21:10:43 +0000 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2020-02-12 21:10:43 +0000 |
commit | 3c0d198d0c0fff5a7949f676bb6cb414908bca02 (patch) | |
tree | 3101b330803e94e2110fc71ecc6b8324a71fb66a /db | |
parent | b0fe37ef58e19bc6f5d1db99f2ddc809f180a6f5 (diff) | |
download | gitlab-ce-3c0d198d0c0fff5a7949f676bb6cb414908bca02.tar.gz |
Add latest changes from gitlab-org/security/gitlab@12-7-stable-ee
Diffstat (limited to 'db')
-rw-r--r-- | db/post_migrate/20200204113223_schedule_recalculate_project_authorizations.rb | 43 | ||||
-rw-r--r-- | db/schema.rb | 2 |
2 files changed, 44 insertions, 1 deletions
diff --git a/db/post_migrate/20200204113223_schedule_recalculate_project_authorizations.rb b/db/post_migrate/20200204113223_schedule_recalculate_project_authorizations.rb new file mode 100644 index 00000000000..83b58300115 --- /dev/null +++ b/db/post_migrate/20200204113223_schedule_recalculate_project_authorizations.rb @@ -0,0 +1,43 @@ +# frozen_string_literal: true + +class ScheduleRecalculateProjectAuthorizations < ActiveRecord::Migration[5.1] + include Gitlab::Database::MigrationHelpers + + DOWNTIME = false + + MIGRATION = 'RecalculateProjectAuthorizations' + BATCH_SIZE = 2_500 + DELAY_INTERVAL = 2.minutes.to_i + + disable_ddl_transaction! + + class Namespace < ActiveRecord::Base + include ::EachBatch + + self.table_name = 'namespaces' + end + + class ProjectAuthorization < ActiveRecord::Base + include ::EachBatch + + self.table_name = 'project_authorizations' + end + + def up + say "Scheduling #{MIGRATION} jobs" + + max_group_id = Namespace.where(type: 'Group').maximum(:id) + project_authorizations = ProjectAuthorization.where('project_id <= ?', max_group_id) + .select(:user_id) + .distinct + + project_authorizations.each_batch(of: BATCH_SIZE, column: :user_id) do |authorizations, index| + delay = index * DELAY_INTERVAL + user_ids = authorizations.map(&:user_id) + BackgroundMigrationWorker.perform_in(delay, MIGRATION, [user_ids]) + end + end + + def down + end +end diff --git a/db/schema.rb b/db/schema.rb index 528631c4b36..41bebdb8eac 100644 --- a/db/schema.rb +++ b/db/schema.rb @@ -10,7 +10,7 @@ # # It's strongly recommended that you check this file into your version control system. -ActiveRecord::Schema.define(version: 2020_01_27_090233) do +ActiveRecord::Schema.define(version: 2020_02_04_113223) do # These are extensions that must be enabled in order to support this database enable_extension "pg_trgm" |