fix(tracer,dbapi): adding debug messages

Author: maorlx

epsagon/events/dbapi.py

@@ -23,7 +23,7 @@ def parse_dsn(dsn):
 
 from ..trace import trace_factory
 from ..event import BaseEvent
-from ..utils import database_connection_type
+from ..utils import database_connection_type, print_debug
 
 MAX_QUERY_SIZE = 2048
 
@@ -68,33 +68,47 @@ def __init__(
         """
 
         super(DBAPIEvent, self).__init__(start_time)
-
         self.event_id = 'dbapi-{}'.format(str(uuid4()))
 
         # in case of pg instrumentation we extract data from the dsn property
         if hasattr(connection, 'dsn'):
+            print_debug('Parsing using dsn')
             dsn = parse_dsn(connection.dsn)
-            db_name = dsn['dbname']
+            print_debug('Parsed dsn: {}'.format(dsn))
+            db_name = dsn.get('dbname', '')
             host = dsn.get('host', 'local')
             query = cursor.query
         else:
             query = _args[0]
             host = connection.extract_hostname
             db_name = connection.extract_dbname
+        print_debug(
+            'Extracted db name - {}, host - {}, query - {}'.format(
+                db_name, host, query
+            )
+        )
 
         self.resource['name'] = db_name if db_name else host
 
         # NOTE: The operation might not be identified properly when
         # using 'WITH' clause
-        operation = query.split()[0].lower()
+        splitted_query = query.split()
+        if not splitted_query:
+            print_debug('Cannot extract operation from query {}'.format(query))
+            operation = ''
+        else:
+            operation = splitted_query[0].lower()
         self.resource['operation'] = operation
-
         # override event type with the specific DB type
         self.resource['type'] = database_connection_type(
             host,
             self.RESOURCE_TYPE
         )
-
+        print_debug(
+            '{}: resource name - {}, operation - {}'.format(
+                self.event_id, self.resource['name'], self.resource['operation']
+            )
+        )
         self.resource['metadata'] = {
             'Host': host,
             'Driver': connection.__class__.__module__.split('.')[0],
@@ -107,6 +121,11 @@ def __init__(
                 (not trace_factory.metadata_only)
         ):
             self.resource['metadata']['Query'] = query[:MAX_QUERY_SIZE]
+            print_debug(
+                '{}: collected query {}'.format(
+                    self.event_id, self.resource['metadata']['Query']
+                    )
+            )
 
         if exception is None:
             # Update response data
@@ -155,6 +174,7 @@ def create_event(wrapped, cursor_wrapper, args, kwargs, start_time,
         :param exception:
         :return:
         """
+        print_debug('Creating DBAPI event')
         event = DBAPIEvent(
             cursor_wrapper.connection_wrapper,
             cursor_wrapper,
@@ -163,4 +183,5 @@ def create_event(wrapped, cursor_wrapper, args, kwargs, start_time,
             start_time,
             exception,
         )
+        print_debug('Adding DBAPI event to trace')
         trace_factory.add_event(event)

epsagon/trace.py

@@ -1182,19 +1182,29 @@ def _send_traces(self):
             else create_transport(self.collector_url, self.token)
         )
 
-        # Update events resource metadata.
-        for event in self.events:
-            # Remove ignored keys.
-            event.resource['metadata'] = self.remove_ignored_keys(
-                event.resource['metadata'])
-            # Keep allowed keys.
-            if self.keys_to_allow:
-                event.resource['metadata'] = self.get_dict_with_allow_keys(
+        try:
+            # Update events resource metadata.
+            for event in self.events:
+                # Remove ignored keys.
+                event.resource['metadata'] = self.remove_ignored_keys(
                     event.resource['metadata'])
-            type(self)._trim_dict_values(
-                event.resource['metadata'],
-                MAX_METADATA_FIELD_SIZE_LIMIT
-            )
+                # Keep allowed keys.
+                if self.keys_to_allow:
+                    event.resource['metadata'] = self.get_dict_with_allow_keys(
+                        event.resource['metadata'])
+                type(self)._trim_dict_values(
+                    event.resource['metadata'],
+                    MAX_METADATA_FIELD_SIZE_LIMIT
+                )
+        except Exception as exception:
+            if self.debug:
+                print(
+                    'Failed to send trace: '
+                    'updating events resource metadata failed: {}'.format(
+                        exception
+                    )
+                )
+            return
 
         try:
             if self.runner:

epsagon/triggers/aws_lambda.py

@@ -7,7 +7,7 @@
 from importlib import import_module
 import hashlib
 import json
-from epsagon.utils import add_data_if_needed, parse_json
+from epsagon.utils import add_data_if_needed, parse_json, print_debug
 from ..event import BaseEvent
 from ..constants import EPSAGON_HEADER
 
@@ -220,6 +220,7 @@ def __init__(self, start_time, event, context):
             'Notification Message',
             message
         )
+        print_debug('Initialized SNS Lambda trigger')
 
 
 class SQSLambdaTrigger(BaseLambdaTrigger):

epsagon/wrappers/aws_lambda.py

@@ -142,7 +142,7 @@ def _lambda_wrapper(*args, **kwargs):
                 epsagon.trace.trace_factory.send_traces()
             # pylint: disable=W0703
             except Exception:
-                pass
+                epsagon.utils.print_debug('Failed to send Lambda trace')
 
     _lambda_wrapper.__instrumented__ = True
     return _lambda_wrapper