Class: MergeRequests::ProcessScheduledMergeWorker

Inherits:
Object
  • Object
show all
Includes:
ApplicationWorker, CronjobQueue, Gitlab::ExclusiveLeaseHelpers
Defined in:
app/workers/merge_requests/process_scheduled_merge_worker.rb

Overview

rubocop:disable Scalability/IdempotentWorker – time dependent queries can’t be idempotent

Constant Summary collapse

LOCK_RETRY =
3
LOCK_TTL =
5.minutes
DELAY =
7.seconds
BATCH_SIZE =
500

Constants included from Gitlab::ExclusiveLeaseHelpers

Gitlab::ExclusiveLeaseHelpers::FailedToObtainLockError

Constants included from ApplicationWorker

ApplicationWorker::LOGGING_EXTRA_KEY, ApplicationWorker::SAFE_PUSH_BULK_LIMIT

Constants included from Gitlab::Loggable

Gitlab::Loggable::ANONYMOUS

Constants included from WorkerAttributes

WorkerAttributes::DEFAULT_CONCURRENCY_LIMIT_PERCENTAGE_BY_URGENCY, WorkerAttributes::DEFAULT_DATA_CONSISTENCY, WorkerAttributes::DEFAULT_DATA_CONSISTENCY_PER_DB, WorkerAttributes::DEFAULT_DEFER_DELAY, WorkerAttributes::LOAD_BALANCED_DATA_CONSISTENCIES, WorkerAttributes::NAMESPACE_WEIGHTS, WorkerAttributes::VALID_DATA_CONSISTENCIES, WorkerAttributes::VALID_RESOURCE_BOUNDARIES, WorkerAttributes::VALID_URGENCIES

Instance Method Summary collapse

Methods included from Gitlab::ExclusiveLeaseHelpers

#in_lock

Methods included from Gitlab::Loggable

#build_structured_payload

Methods included from Gitlab::SidekiqVersioning::Worker

#job_version

Methods included from WorkerContext

#with_context

Instance Method Details

#performObject



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
# File 'app/workers/merge_requests/process_scheduled_merge_worker.rb', line 20

def perform
  in_lock(lock_key, **lock_params) do
    # rubocop:disable CodeReuse/ActiveRecord -- using keyset pagination with custom order for better performance
    order = Gitlab::Pagination::Keyset::Order.build([
      Gitlab::Pagination::Keyset::ColumnOrderDefinition.new(
        attribute_name: 'merge_after',
        order_expression: MergeRequests::MergeSchedule.arel_table[:merge_after].asc,
        nullable: :not_nullable
      ),
      Gitlab::Pagination::Keyset::ColumnOrderDefinition.new(
        attribute_name: 'merge_request_id',
        order_expression: MergeRequests::MergeSchedule.arel_table[:merge_request_id].asc,
        nullable: :not_nullable
      )
    ])

    scope = MergeRequests::MergeSchedule
      .where(merge_after: ..Time.zone.now)
      .order(order)
      .select(:merge_after, :merge_request_id)

    iterator = Gitlab::Pagination::Keyset::Iterator.new(scope: scope)

    iteration = 0
    iterator.each_batch(of: BATCH_SIZE) do |records|
      loaded_records = MergeRequest
        .with_auto_merge_enabled
        .where(id: records.to_a.pluck(:merge_request_id))

      enqueue_auto_merge_process_worker(loaded_records, iteration)
      iteration += 1
    end
    # rubocop:enable CodeReuse/ActiveRecord
  end
end