More metrics

Author: whu-lyft

kmsauth/__init__.py

@@ -291,15 +291,29 @@ def decrypt_token(self, username, token):
                 if len(self.TOKENS) >= self.token_cache_size:
                     self.stats.incr('token_cache_eviction')
 
+                # Checkpoint 3.5: After stats calls in cache miss
+                checkpoint_3_5 = (datetime.datetime.utcnow() - time_start).total_seconds() * 1000  # noqa: E501
+                self.stats.timing('checkpoint_3_5_after_cache_miss_stats', checkpoint_3_5)  # noqa: E501
+
             try:
                 token = base64.b64decode(token)
+                if self.stats:
+                    # Checkpoint 3.7: After base64 decode
+                    checkpoint_3_7 = (datetime.datetime.utcnow() - time_start).total_seconds() * 1000  # noqa: E501
+                    self.stats.timing('checkpoint_3_7_after_base64_decode', checkpoint_3_7)  # noqa: E501
+
                 # Ensure normal context fields override whatever is in
                 # extra_context.
                 context = copy.deepcopy(self.extra_context)
                 context['to'] = self.to_auth_context
                 context['from'] = _from
                 if version > 1:
                     context['user_type'] = user_type
+
+                if self.stats:
+                    # Checkpoint 3.9: After context setup
+                    checkpoint_3_9 = (datetime.datetime.utcnow() - time_start).total_seconds() * 1000  # noqa: E501
+                    self.stats.timing('checkpoint_3_9_after_context_setup', checkpoint_3_9)  # noqa: E501
                 if self.stats:
                     with self.stats.timer('kms_decrypt_token'):
                         data = self.kms_client.decrypt(

setup.py

@@ -13,7 +13,7 @@
 
 from setuptools import setup, find_packages
 
-VERSION = "0.6.4.dev5"
+VERSION = "0.6.4.dev6"
 
 requirements = [
     # Boto3 is the Amazon Web Services (AWS) Software Development Kit (SDK)