def __exit__(self, type, value, traceback):
end_time = time()
duration = end_time - self.start_time
if should_sample(duration):
sample(self.sample_type, self.sample_key, duration, self.params)
return False
def task_postrun_handler(task_id, task, args, kwargs, retval, **kwds):
duration = time() - task_start_times[task_id]
del task_start_times[task_id]
if not should_sample(duration):
return
sample('celery', str(task), duration, [args, kwargs])
def process_response(self, request, response):
duration = time() - self.request_start_times.pop(request)
view_fallback = {"function": request.path, "args": [], "kwargs": {}}
view_call = self.request_view_calls.pop(request, view_fallback)
if should_sample(duration):
sample("request", view_call["function"], duration, [view_call["args"], view_call["kwargs"]])
return response
def sampler(*args, **kwargs):
start = time.time()
try:
if pre_invoke:
pre_invoke(*args, **kwargs)
return method(*args, **kwargs)
finally:
stop = time.time()
if should_sample(stop - start):
query, query_type = sampling_method(*args, **kwargs)
if query:
sample(query_type, query, stop - start, [args, kwargs])
def sampler(*args, **kwargs):
start = time.time()
try:
if pre_invoke:
pre_invoke(*args, **kwargs)
return method(*args, **kwargs)
finally:
stop = time.time()
if should_sample(stop - start):
query, query_type = sampling_method(*args, **kwargs)
if query:
sample(query_type, query, stop - start, [args, kwargs])
def process_response(self, request, response):
duration = time() - self.request_start_times.pop(request)
view_fallback = {
'function': request.path,
'args': [],
'kwargs': {},
}
view_call = self.request_view_calls.pop(request, view_fallback)
if should_sample(duration):
sample('request', view_call['function'], duration,
[view_call['args'], view_call['kwargs']])
return response
def process_response(self, request, response):
duration = time() - self.request_start_times.pop(request)
view_fallback = {
'function': request.path,
'args': [],
'kwargs': {},
}
view_call = self.request_view_calls.pop(request, view_fallback)
if should_sample(duration):
sample('request',
view_call['function'],
duration,
[view_call['args'], view_call['kwargs']])
return response
def log_sql(self, sql, time, params):
if not should_sample(time):
return
sample('sql', sql, time, params)
def log_sql(self, sql, time, params):
if not should_sample(time):
return
sample('sql', sql, time, params)
