lib/tasks/gitlab/db/lock_writes.rake


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

# frozen_string_literal: true

namespace :gitlab do
  namespace :db do
    desc "GitLab | DB | Install prevent write triggers on all databases"
    task lock_writes: [:environment, 'gitlab:db:validate_config'] do
      logger = Logger.new($stdout)
      logger.level = Gitlab::Utils.to_boolean(ENV['VERBOSE']) ? Logger::INFO : Logger::WARN
      Gitlab::Database::TablesLocker.new(
        logger: logger,
        dry_run: Gitlab::Utils.to_boolean(ENV['DRY_RUN'], default: false)
      ).lock_writes
    end

    desc "GitLab | DB | Remove all triggers that prevents writes from all databases"
    task unlock_writes: :environment do
      logger = Logger.new($stdout)
      logger.level = Gitlab::Utils.to_boolean(ENV['VERBOSE']) ? Logger::INFO : Logger::WARN
      Gitlab::Database::TablesLocker.new(
        logger: logger,
        dry_run: Gitlab::Utils.to_boolean(ENV['DRY_RUN'], default: false)
      ).unlock_writes
    end
  end
end