def patched_query_request(original_func, instance, args, kwargs):
pin = Pin.get_from(instance)
if not pin or not pin.enabled():
return original_func(*args, **kwargs)
endpoint_name = getattr(instance, "host").split(".")[0]
with pin.tracer.trace(
"{}.command".format(endpoint_name),
service="{}.{}".format(pin.service, endpoint_name),
span_type=SpanTypes.HTTP,
) as span:
span.set_tag(SPAN_MEASURED_KEY)
operation_name = None
if args:
operation_name = get_argument_value(args, kwargs, 0, "action")
span.resource = "%s.%s" % (endpoint_name, operation_name.lower())
else:
span.resource = endpoint_name
aws.add_span_arg_tags(span, endpoint_name, args, AWS_QUERY_ARGS_NAME,
AWS_QUERY_TRACED_ARGS)
# Obtaining region name
region_name = _get_instance_region_name(instance)
meta = {
aws.AGENT: "boto",
aws.OPERATION: operation_name,
}
if region_name:
meta[aws.REGION] = region_name
span.set_tags(meta)
# Original func returns a boto.connection.HTTPResponse object
result = original_func(*args, **kwargs)
span.set_tag(http.STATUS_CODE, getattr(result, "status"))
span.set_tag(http.METHOD, getattr(result, "_method"))
# set analytics sample rate
span.set_tag(ANALYTICS_SAMPLE_RATE_KEY,
config.boto.get_analytics_sample_rate())
return result
def patched_auth_request(original_func, instance, args, kwargs):
# Catching the name of the operation that called make_request()
operation_name = None
# Go up the stack until we get the first non-ddtrace module
# DEV: For `lambda.list_functions()` this should be:
# - ddtrace.contrib.boto.patch
# - ddtrace.vendor.wrapt.wrappers
# - boto.awslambda.layer1 (make_request)
# - boto.awslambda.layer1 (list_functions)
# But can vary depending on Python versions; that's why we use an heuristic
frame = inspect.currentframe().f_back
operation_name = None
while frame:
if frame.f_code.co_name == "make_request":
operation_name = frame.f_back.f_code.co_name
break
frame = frame.f_back
pin = Pin.get_from(instance)
if not pin or not pin.enabled():
return original_func(*args, **kwargs)
endpoint_name = getattr(instance, "host").split(".")[0]
with pin.tracer.trace(
"{}.command".format(endpoint_name),
service="{}.{}".format(pin.service, endpoint_name),
span_type=SpanTypes.HTTP,
) as span:
span.set_tag(SPAN_MEASURED_KEY)
if args:
http_method = get_argument_value(args, kwargs, 0, "method")
span.resource = "%s.%s" % (endpoint_name, http_method.lower())
else:
span.resource = endpoint_name
aws.add_span_arg_tags(span, endpoint_name, args, AWS_AUTH_ARGS_NAME,
AWS_AUTH_TRACED_ARGS)
# Obtaining region name
region_name = _get_instance_region_name(instance)
meta = {
aws.AGENT: "boto",
aws.OPERATION: operation_name,
}
if region_name:
meta[aws.REGION] = region_name
span.set_tags(meta)
# Original func returns a boto.connection.HTTPResponse object
result = original_func(*args, **kwargs)
span.set_tag(http.STATUS_CODE, getattr(result, "status"))
span.set_tag(http.METHOD, getattr(result, "_method"))
# set analytics sample rate
span.set_tag(ANALYTICS_SAMPLE_RATE_KEY,
config.boto.get_analytics_sample_rate())
return result
