Donal McBreen
18 lines
381 B
Ruby
18 lines
381 B
Ruby
require_relative "extensions"
|
|
|
|
class DatabaseRoleLogger
|
|
def initialize(app)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
Rails.logger.tagged(ActiveRecord::Base.current_role.to_s) do
|
|
@app.call(env)
|
|
end
|
|
end
|
|
end
|
|
|
|
if ActiveRecord::Base.replica_configured?
|
|
Rails.application.config.middleware.insert_after ActiveRecord::Middleware::DatabaseSelector, DatabaseRoleLogger
|
|
end
|