require 'sidekiq/instrument/mixin'

module Sidekiq::Instrument
  class ServerMiddleware
    include Sidekiq::Instrument::MetricNames

    def call(worker, job, queue, &block)
      Statter.statsd.increment(metric_name(worker, 'dequeue'))
      Statter.dogstatsd&.increment('sidekiq.dequeue', worker_dog_options(worker))

      start_time = Time.now
      yield block
      execution_time_ms = (Time.now - start_time) * 1000
      Statter.statsd.measure(metric_name(worker, 'runtime'), execution_time_ms)
      Statter.dogstatsd&.timing('sidekiq.runtime', execution_time_ms, worker_dog_options(worker))
    rescue StandardError => e
      Statter.statsd.increment(metric_name(worker, 'error'))
      Statter.dogstatsd&.increment('sidekiq.error', worker_dog_options(worker))
      raise e
    ensure
      Statter.dogstatsd&.flush(sync: true)
    end
  end
end