scout_reporter.rb

# frozen_string_literal: true

# Instruments a block of code and reports it to scout as a background job. Do NOT use this to instrument a small
# amount of code. It is meant to be used for ruby processes that we start and scout doesn't automatically instrument.
class ScoutReporter
  @@agent = nil

  def self.initialize_agent
    if @@agent.nil?
      @@agent = ::ScoutApm::Agent.instance
      @@agent.start
    end
  end

  def self.instrument(job_name, &block)
    initialize_agent

    return if @@agent.nil?

    req = ScoutApm::RequestManager.lookup

    begin
      req.start_layer(ScoutApm::Layer.new('Queue', 'default')) #UNKNOWN_QUEUE_PLACEHOLDER
      started_queue = true
      req.start_layer(ScoutApm::Layer.new('Job', job_name))
      started_job = true

      yield

    rescue Exception => e
      req.error!
      raise
    ensure
      req.stop_layer if started_job
      req.stop_layer if started_queue
    end
  end
end