@@ -56,11 +56,11 @@ def config(self, app):
5656
5757 @app .middleware ('request' )
5858 def before_request (request ):
59- request [ ' request_info' ] = json_logging .RequestInfo (request )
59+ request . ctx . request_info = json_logging .RequestInfo (request )
6060
6161 @app .middleware ('response' )
6262 def after_request (request , response ):
63- request_info = request [ ' request_info' ]
63+ request_info = request . ctx . request_info
6464 request_info .update_response_status (response )
6565 self .request_logger .info ("" , extra = {'request_info' : request_info , 'type' : 'request' })
6666
@@ -93,11 +93,14 @@ def get_http_header(self, request, header_name, default=None):
9393 return request .headers .get (header_name )
9494 return default
9595
96- def set_correlation_id (self , request_ , value ):
97- request_ [ ' correlation_id' ] = value
96+ def set_correlation_id (self , request , value ):
97+ request . ctx . correlation_id = value
9898
9999 def get_correlation_id_in_request_context (self , request ):
100- return request .get ('correlation_id' )
100+ try :
101+ return request .ctx .correlation_id
102+ except AttributeError :
103+ return None
101104
102105 def get_protocol (self , request ):
103106 return json_logging .EMPTY_VALUE
0 commit comments