44 lines
1.3 KiB
Ruby
44 lines
1.3 KiB
Ruby
#
|
|
# Copyright (C) 2011 Instructure, Inc.
|
|
#
|
|
# This file is part of Canvas.
|
|
#
|
|
# Canvas is free software: you can redistribute it and/or modify it under
|
|
# the terms of the GNU Affero General Public License as published by the Free
|
|
# Software Foundation, version 3 of the License.
|
|
#
|
|
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License along
|
|
# with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
|
|
class RequestContextGenerator
|
|
def initialize(app)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
# This is a crummy way to plumb this data through to the logger
|
|
request_id = UUIDSingleton.instance.generate
|
|
@@session_key ||= ActionController::Base.session_options[:key]
|
|
session_id = CGI::Cookie::parse(env['HTTP_COOKIE'])[@@session_key][0] rescue nil
|
|
Thread.current[:context] = {
|
|
:request_id => request_id,
|
|
:session_id => session_id
|
|
}
|
|
|
|
status, headers, body = @app.call(env)
|
|
headers['X-Request-Context-Id'] = request_id
|
|
[ status, headers, body ]
|
|
end
|
|
|
|
def self.request_id
|
|
Thread.current[:context].try(:[], :request_id)
|
|
end
|
|
end
|