@@ -56,11 +56,11 @@ def config(self, app):
56
56
57
57
@app .middleware ('request' )
58
58
def before_request (request ):
59
- request [ ' request_info' ] = json_logging .RequestInfo (request )
59
+ request . ctx . request_info = json_logging .RequestInfo (request )
60
60
61
61
@app .middleware ('response' )
62
62
def after_request (request , response ):
63
- request_info = request [ ' request_info' ]
63
+ request_info = request . ctx . request_info
64
64
request_info .update_response_status (response )
65
65
self .request_logger .info ("" , extra = {'request_info' : request_info , 'type' : 'request' })
66
66
@@ -93,11 +93,14 @@ def get_http_header(self, request, header_name, default=None):
93
93
return request .headers .get (header_name )
94
94
return default
95
95
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
98
98
99
99
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
101
104
102
105
def get_protocol (self , request ):
103
106
return json_logging .EMPTY_VALUE
0 commit comments