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 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)