diff options
Diffstat (limited to 'db/post_migrate/20181010133639_backfill_store_project_full_path_in_repo.rb')
-rw-r--r-- | db/post_migrate/20181010133639_backfill_store_project_full_path_in_repo.rb | 125 |
1 files changed, 8 insertions, 117 deletions
diff --git a/db/post_migrate/20181010133639_backfill_store_project_full_path_in_repo.rb b/db/post_migrate/20181010133639_backfill_store_project_full_path_in_repo.rb index 1e2e4527d2b..e9ab45ae9a1 100644 --- a/db/post_migrate/20181010133639_backfill_store_project_full_path_in_repo.rb +++ b/db/post_migrate/20181010133639_backfill_store_project_full_path_in_repo.rb @@ -3,134 +3,25 @@ class BackfillStoreProjectFullPathInRepo < ActiveRecord::Migration include Gitlab::Database::MigrationHelpers - DOWNTIME = false + DOWNTIME = false + BATCH_SIZE = 1_000 + DELAY_INTERVAL = 5.minutes + UP_MIGRATION = 'BackfillProjectFullpathInRepoConfig::Up' + DOWN_MIGRATION = 'BackfillProjectFullpathInRepoConfig::Down' - class Repository - attr_reader :storage - - def initialize(storage, relative_path) - @storage = storage - @relative_path = relative_path - end - - def gitaly_repository - Gitaly::Repository.new(storage_name: @storage, relative_path: @relative_path) - end - end - - module Storage - class HashedProject - attr_accessor :project - - ROOT_PATH_PREFIX = '@hashed'.freeze - - def initialize(project) - @project = project - end - - def disk_path - "#{ROOT_PATH_PREFIX}/#{disk_hash[0..1]}/#{disk_hash[2..3]}/#{disk_hash}" - end - - def disk_hash - @disk_hash ||= Digest::SHA2.hexdigest(project.id.to_s) if project.id - end - end - - class LegacyProject - attr_accessor :project - - def initialize(project) - @project = project - end - - def disk_path - project.full_path - end - end - end - - module Routable - extend ActiveSupport::Concern - - def full_path - @full_path ||= build_full_path - end - - def build_full_path - if parent && path - parent.full_path + '/' + path - else - path - end - end - end - - class Namespace < ActiveRecord::Base - self.table_name = 'namespaces' - - include Routable - - belongs_to :parent, class_name: "Namespace" - end + disable_ddl_transaction! class Project < ActiveRecord::Base self.table_name = 'projects' - include Routable include EachBatch - - FULLPATH_CONFIG_KEY = 'gitlab.fullpath' - - belongs_to :namespace - delegate :disk_path, to: :storage - alias_method :parent, :namespace - - def add_fullpath_config - entries = { FULLPATH_CONFIG_KEY => full_path } - - repository_service.set_config(entries) - end - - def remove_fullpath_config - repository_service.delete_config([FULLPATH_CONFIG_KEY]) - end - - def storage - @storage ||= - if hashed_storage? - Storage::HashedProject.new(self) - else - Storage::LegacyProject.new(self) - end - end - - def hashed_storage? - self.storage_version && self.storage_version >= 1 - end - - def repository - @repository ||= Repository.new(repository_storage, disk_path + '.git') - end - - def repository_service - @repository_service ||= Gitlab::GitalyClient::RepositoryService.new(repository) - end end def up - Project.each_batch do |batch| - batch.each do |project| - project.add_fullpath_config - end - end + queue_background_migration_jobs_by_range_at_intervals(Project, UP_MIGRATION, DELAY_INTERVAL) end def down - Project.each_batch do |batch| - batch.each do |project| - project.remove_fullpath_config - end - end + queue_background_migration_jobs_by_range_at_intervals(Project, DOWN_MIGRATION, DELAY_INTERVAL) end end |