Contents

# frozen_string_literal: true

module PrometheusExporter::Instrumentation
  class DelayedJob
    JOB_CLASS_REGEXP = %r{job_class: (\w+:{0,2})+}.freeze

    class << self
      def register_plugin(client: nil)
        instrumenter = self.new(client: client)
        return unless defined?(Delayed::Plugin)

        plugin = Class.new(Delayed::Plugin) do
          callbacks do |lifecycle|
            lifecycle.around(:invoke_job) do |job, *args, &block|
              max_attempts = Delayed::Worker.max_attempts
              instrumenter.call(job, max_attempts, *args, &block)
            end
          end
        end

        Delayed::Worker.plugins << plugin
      end
    end

    def initialize(client: nil)
      @client = client || PrometheusExporter::Client.default
    end

    def call(job, max_attempts, *args, &block)
      success = false
      start = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
      attempts = job.attempts + 1 # Increment because we're adding the current attempt
      result = block.call(job, *args)
      success = true
      result
    ensure
      duration = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC) - start

      @client.send_json(
        type: "delayed_job",
        name: job.handler.to_s.match(JOB_CLASS_REGEXP).to_a[1].to_s,
        success: success,
        duration: duration,
        attempts: attempts,
        max_attempts: max_attempts
      )
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
prometheus_exporter-0.4.14	lib/prometheus_exporter/instrumentation/delayed_job.rb
prometheus_exporter-0.4.13	lib/prometheus_exporter/instrumentation/delayed_job.rb
prometheus_exporter-0.4.12	lib/prometheus_exporter/instrumentation/delayed_job.rb
prometheus_exporter-0.4.11	lib/prometheus_exporter/instrumentation/delayed_job.rb