summaryrefslogtreecommitdiff
path: root/app/workers/gitlab_performance_bar_stats_worker.rb
blob: 6d637ad15867ec055a426c347e12f29075eef9b8 (plain)
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# frozen_string_literal: true

class GitlabPerformanceBarStatsWorker
  include ApplicationWorker

  data_consistency :always
  worker_resource_boundary :cpu

  sidekiq_options retry: 3

  LEASE_KEY = 'gitlab:performance_bar_stats'
  LEASE_TIMEOUT = 600
  WORKER_DELAY = 120
  STATS_KEY = 'performance_bar_stats:pending_request_ids'
  STATS_KEY_EXPIRE = 30.minutes.to_i

  feature_category :metrics
  idempotent!

  def perform(lease_uuid)
    Gitlab::Redis::Cache.with do |redis|
      request_ids = fetch_request_ids(redis, lease_uuid)
      stats = Gitlab::PerformanceBar::Stats.new(redis)

      request_ids.each do |id|
        stats.process(id)
      end
    end
  end

  private

  def fetch_request_ids(redis, lease_uuid)
    ids = redis.smembers(STATS_KEY)
    redis.del(STATS_KEY)
    Gitlab::ExclusiveLease.cancel(LEASE_KEY, lease_uuid)

    ids
  end
end