Class: Judoscale::GoodJob::MetricsCollector

Inherits:
JobMetricsCollector
  • Object
show all
Includes:
ActiveRecordHelper
Defined in:
lib/judoscale/good_job/metrics_collector.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ MetricsCollector

Returns a new instance of MetricsCollector.



28
29
30
31
 
# File 'lib/judoscale/good_job/metrics_collector.rb', line 28

def initialize
  super
  @good_job_base_class = self.class.good_job_base_class
end

Class Method Details

.adapter_config ⇒ Object 



20
21
22
 
# File 'lib/judoscale/good_job/metrics_collector.rb', line 20

def self.adapter_config
  Judoscale::Config.instance.good_job
end

.collect?(config) ⇒ Boolean

Returns:

  • (Boolean) 


24
25
26
 
# File 'lib/judoscale/good_job/metrics_collector.rb', line 24

def self.collect?(config)
  super && ActiveRecordHelper.table_exists_for_model?(good_job_base_class)
end

.good_job_base_class ⇒ Object 



16
17
18
 
# File 'lib/judoscale/good_job/metrics_collector.rb', line 16

def self.good_job_base_class
  good_job_v4? ? ::GoodJob::Job : ::GoodJob::Execution
end

.good_job_v4? ⇒ Boolean

Returns:

  • (Boolean) 


12
13
14
 
# File 'lib/judoscale/good_job/metrics_collector.rb', line 12

def self.good_job_v4?
  Gem::Version.new(::GoodJob::VERSION) >= Gem::Version.new("4")
end

Instance Method Details

#collect ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/judoscale/good_job/metrics_collector.rb', line 33

def collect
  metrics = []
  time = Time.now.utc

  if queues.empty?
    self.queues |= run_silently { @good_job_base_class.distinct.pluck(:queue_name) }
  end

  # logically we don't need the finished_at condition, but it lets postgres use the indexes
  oldest_execution_time_by_queue = run_silently do
    @good_job_base_class
      .where(performed_at: nil, finished_at: nil)
      .group(:queue_name)
      .pluck(:queue_name, Arel.sql("min(coalesce(scheduled_at, created_at))"))
      .to_h
  end
  self.queues |= oldest_execution_time_by_queue.keys

  if track_busy_jobs?
    busy_count_by_queue = run_silently do
      @good_job_base_class.running.group(:queue_name).count
    end
    self.queues |= busy_count_by_queue.keys
  end

  queues.each do |queue|
    run_at = oldest_execution_time_by_queue[queue]
    # DateTime.parse assumes a UTC string
    run_at = DateTime.parse(run_at) if run_at.is_a?(String)
    latency_ms = run_at ? ((time - run_at) * 1000).ceil : 0
    latency_ms = 0 if latency_ms < 0

    metrics.push Metric.new(:qt, latency_ms, time, queue)

    if track_busy_jobs?
      busy_count = busy_count_by_queue[queue] || 0
      metrics.push Metric.new(:busy, busy_count, time, queue)
    end
  end

  log_collection(metrics)
  metrics
end