RequestId

This is a gem with a collection of middleware for easily cascading Heroku request id's Throughout the system. It includes:

  • Rack middleware, which adds the request_id to Thread.current[:request_id].
  • Sidekiq Client middleware, which adds the request_id to the message payload.
  • Sidekiq Server middleware, which adds the request_id to Thread.current[:request_id] from the request_id in the message payload.
  • Faraday middleware, which adds the request_id as a response header X-Request-Id.

Installation

Add this line to your application's Gemfile:

gem 'request_id'

Usage

Add the rack middleware:

use Rack::RequestId

If you're using Sidekiq

Add the client middleware.

Sidekiq.configure_client do |config|
  config.client_middleware do |chain|
    chain.add Sidekiq::Middleware::Client::RequestId
  end
end

Add the server middleware.

Sidekiq.configure_server do |config|
  config.client_middleware do |chain|
    chain.add Sidekiq::Middleware::Client::RequestId
  end

  config.server_middleware do |chain|
    chain.add Sidekiq::Middleware::Client::RequestId
  end
end

If you're using Faraday

Add the middleware.

  builder.use Faraday::RequestId

Customization

You can customize each middleware to store the value of any header you like in the same fashion. For instance, if you wanted to track a X-Request-Id header as well as a X-Session-Id header, you could do so like this:

# Rack
use Rack::RequestId
use Rack::RequestId, key: :request_id, value: -> (env) { env['HTTP_X_SESSION_ID'], response_header: 'X-Session-Id' }

# Sidekiq
Sidekiq.configure_client do |config|
  config.client_middleware do |chain|
    chain.add Sidekiq::Middleware::Client::RequestId
    chain.add Sidekiq::Middleware::Client::RequestId, key: :session_id, value: -> { ::RequestId.get(:session_id) }
  end
end

Sidekiq.configure_server do |config|
  config.client_middleware do |chain|
    chain.add Sidekiq::Middleware::Client::RequestId
    chain.add Sidekiq::Middleware::Client::RequestId, key: :session_id, value: -> { ::RequestId.get(:session_id) }
  end

  config.server_middleware do |chain|
    chain.add Sidekiq::Middleware::Client::RequestId, key: :session_id, value: lambda { |item| item['session_id'] }
  end
end

# Faraday
builder.use Faraday::RequestId, key: :session_id, header: 'X-Session-Id'

Contributing

  1. Fork it
  2. Create your feature branch (git checkout -b my-new-feature)
  3. Commit your changes (git commit -am 'Add some feature')
  4. Push to the branch (git push origin my-new-feature)
  5. Create new Pull Request