diff --git a/autodynatrace/wrappers/pymongo/wrapper.py b/autodynatrace/wrappers/pymongo/wrapper.py index 73d9cc7..a5de5e3 100644 --- a/autodynatrace/wrappers/pymongo/wrapper.py +++ b/autodynatrace/wrappers/pymongo/wrapper.py @@ -24,7 +24,7 @@ def started(self, event): tracer = sdk.trace_sql_database_request(db_info, "{}".format(operation)) self._tracer_dict[_get_tracer_dict_key(event)] = tracer tracer.start() - logger.debug("Tracing Mongo call: {}({})@{}:{}".format(operation, db_name, host, port)) + #logger.debug("Tracing Mongo call: {}@{}:{}".format(operation, db_name, host, port)) except Exception as e: logger.debug("Error instrumenting MongoDB: {}".format(e)) @@ -42,7 +42,7 @@ def end(self, failed, event, message=""): logger.debug("Got bad pymongo event: {}".format(event)) tracer.mark_failed("MongoDB Command", message) - logger.debug("Ending Mongo call: {}({})@{}:{}".format(event.command_name, event.database_name, event.connection_id[0], event.connection_id[1])) + #logger.debug("Ending Mongo call: {}({})@{}:{}".format(event.command_name, event.database_name, event.connection_id[0], event.connection_id[1])) tracer.end() self._tracer_dict.pop(_get_tracer_dict_key(event)) diff --git a/autodynatrace/wrappers/urllib/wrapper.py b/autodynatrace/wrappers/urllib/wrapper.py index 82cae87..a902547 100644 --- a/autodynatrace/wrappers/urllib/wrapper.py +++ b/autodynatrace/wrappers/urllib/wrapper.py @@ -12,13 +12,27 @@ def dynatrace_putrequest(wrapped, instance, args, kwargs): method, path = args[:2] scheme = "https" if "HTTPS" in instance.__class__.__name__ else "http" url = "{}://{}{}{}".format( - scheme, instance.host, ":{}".format(instance.port) if str(instance.port) not in ["80", "443"] else "", path + scheme, + instance.host, + ":{}".format(instance.port) + if str(instance.port) not in ["80", "443"] + else "", + path, ) tracer = sdk.trace_outgoing_web_request(url, method) tracer.start() setattr(instance, "__dynatrace_tracer", tracer) ret = wrapped(*args, **kwargs) - tag = tracer.outgoing_dynatrace_string_tag + tag = tracer.outgoing_dynatrace_string_tag # outgoing_dynatrace_byte_tag + + byte_tag = tracer.outgoing_dynatrace_byte_tag + decoded_tag = tracer.outgoing_dynatrace_string_tag.decode("utf-8") + + logger.debug("Tracing urllib, url: '{}', Byte tag: '{}'".format(url, byte_tag)) + logger.debug( + "Tracing urllib, url: '{}', Decoded tag: '{}'".format(url, decoded_tag) + ) + logger.debug("Tracing urllib, url: '{}', tag: '{}'".format(url, tag)) instance.putheader("x-dynatrace", tag) return ret @@ -35,7 +49,15 @@ def dynatrace_getresponse(wrapped, instance, args, kwargs): return response - setattr(httplib.HTTPConnection, "putrequest", wrapt.FunctionWrapper(httplib.HTTPConnection.putrequest, dynatrace_putrequest)) setattr( - httplib.HTTPConnection, "getresponse", wrapt.FunctionWrapper(httplib.HTTPConnection.getresponse, dynatrace_getresponse) + httplib.HTTPConnection, + "putrequest", + wrapt.FunctionWrapper(httplib.HTTPConnection.putrequest, dynatrace_putrequest), + ) + setattr( + httplib.HTTPConnection, + "getresponse", + wrapt.FunctionWrapper( + httplib.HTTPConnection.getresponse, dynatrace_getresponse + ), )