def _wrapped_func(*args, **kwargs):
with trace(
ZIPKIN_API_URL,
service_name=self.__class__.__name__,
span_name="user defined api handler",
):
resp = self._func(*args, **kwargs)
return resp
def handle_batch_request(self, request):
requests = DataLoader.split_requests(request.data)
with trace(
ZIPKIN_API_URL,
service_name=self.__class__.__name__,
span_name=f"call `{self._handler.__class__.__name__}`",
):
responses = self.handler.handle_batch_request(requests, self.func)
return DataLoader.merge_responses(responses)
def _wrapped_api_func():
with trace(ZIPKIN_API_URL,
request.headers,
service_name=self.__class__.__name__):
resp = api_func()
return resp
