tracing.rb 1.16 KB
Newer Older
1 2 3 4 5
# frozen_string_literal: true

if Gitlab::Tracing.enabled?
  require 'opentracing'

6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
  Rails.application.configure do |config|
    config.middleware.insert_after Gitlab::Middleware::CorrelationId, ::Gitlab::Tracing::RackMiddleware
  end

  # Instrument the Sidekiq client
  Sidekiq.configure_client do |config|
    config.client_middleware do |chain|
      chain.add Gitlab::Tracing::Sidekiq::ClientMiddleware
    end
  end

  # Instrument Sidekiq server calls when running Sidekiq server
  if Sidekiq.server?
    Sidekiq.configure_server do |config|
      config.server_middleware do |chain|
        chain.add Gitlab::Tracing::Sidekiq::ServerMiddleware
      end
    end
  end

26 27 28
  # Instrument Rails
  Gitlab::Tracing::Rails::ActiveRecordSubscriber.instrument

29 30 31 32 33 34 35 36
  # In multi-processed clustered architectures (puma, unicorn) don't
  # start tracing until the worker processes are spawned. This works
  # around issues when the opentracing implementation spawns threads
  Gitlab::Cluster::LifecycleEvents.on_worker_start do
    tracer = Gitlab::Tracing::Factory.create_tracer(Gitlab.process_name, Gitlab::Tracing.connection_string)
    OpenTracing.global_tracer = tracer if tracer
  end
end