83 lines
2.2 KiB
Ruby
83 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Database
|
|
module Reindexing
|
|
class Coordinator
|
|
include ExclusiveLeaseGuard
|
|
|
|
# Maximum lease time for the global Redis lease
|
|
# This should be higher than the maximum time for any
|
|
# long running step in the reindexing process (compare with
|
|
# statement timeouts).
|
|
TIMEOUT_PER_ACTION = 1.day
|
|
|
|
attr_reader :index, :notifier
|
|
|
|
def initialize(index, notifier = GrafanaNotifier.new)
|
|
@index = index
|
|
@notifier = notifier
|
|
end
|
|
|
|
def perform
|
|
# This obtains a global lease such that there's
|
|
# only one live reindexing process at a time.
|
|
try_obtain_lease do
|
|
action = ReindexAction.create_for(index)
|
|
|
|
with_notifications(action) do
|
|
perform_for(index, action)
|
|
end
|
|
end
|
|
end
|
|
|
|
def drop
|
|
try_obtain_lease do
|
|
Gitlab::AppLogger.info("Removing index #{index.identifier} which is a leftover, temporary index from previous reindexing activity")
|
|
|
|
retries = Gitlab::Database::WithLockRetriesOutsideTransaction.new(
|
|
connection: index.connection,
|
|
timing_configuration: REMOVE_INDEX_RETRY_CONFIG,
|
|
klass: self.class,
|
|
logger: Gitlab::AppLogger
|
|
)
|
|
|
|
retries.run(raise_on_exhaustion: false) do
|
|
index.connection.tap do |conn|
|
|
conn.execute("DROP INDEX CONCURRENTLY IF EXISTS #{conn.quote_table_name(index.schema)}.#{conn.quote_table_name(index.name)}")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def with_notifications(action)
|
|
notifier.notify_start(action)
|
|
yield
|
|
ensure
|
|
notifier.notify_end(action)
|
|
end
|
|
|
|
def perform_for(index, action)
|
|
ReindexConcurrently.new(index).perform
|
|
rescue StandardError
|
|
action.state = :failed
|
|
|
|
raise
|
|
ensure
|
|
action.finish
|
|
end
|
|
|
|
def lease_timeout
|
|
TIMEOUT_PER_ACTION
|
|
end
|
|
|
|
def lease_key
|
|
[super, index.connection_db_config.name].join('/')
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|