Added new relic metrics to bearerAuthentication

openedx · Feb 3, 2020 · 5ffd36f · 5ffd36f
1 parent e7f562f
commit 5ffd36f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/edx_rest_framework_extensions/auth/bearer/authentication.py b/edx_rest_framework_extensions/auth/bearer/authentication.py
@@ -4,6 +4,7 @@
 
 import requests
 from django.contrib.auth import get_user_model
+from edx_django_utils.monitoring import set_custom_metric
 from rest_framework import exceptions
 from rest_framework.authentication import BaseAuthentication, get_authorization_header
 
@@ -36,21 +37,26 @@ def get_user_info_url(self):
         return get_setting('OAUTH2_USER_INFO_URL')
 
     def authenticate(self, request):
+        set_custom_metric("BearerAuthentication", "Failed")  # default value
         if not self.get_user_info_url():
             logger.warning('The setting OAUTH2_USER_INFO_URL is invalid!')
+            set_custom_metric("BearerAuthentication", "NoURL")
             return None
-
+        set_custom_metric("BearerAuthentication_user_info_url", self.get_user_info_url())
         auth = get_authorization_header(request).split()
 
         if not auth or auth[0].lower() != b'bearer':
+            set_custom_metric("BearerAuthentication", "None")
             return None
 
         if len(auth) == 1:
             raise exceptions.AuthenticationFailed('Invalid token header. No credentials provided.')
         elif len(auth) > 2:
             raise exceptions.AuthenticationFailed('Invalid token header. Token string should not contain spaces.')
 
-        return self.authenticate_credentials(auth[1].decode('utf8'))
+        output = self.authenticate_credentials(auth[1].decode('utf8'))
+        set_custom_metric("BearerAuthentication", "Success")
+        return output
 
     def authenticate_credentials(self, token):
         """

diff --git a/requirements/dev.txt b/requirements/dev.txt
@@ -59,7 +59,7 @@ pyjwt==1.7.1
 pylint-celery==0.3
 pylint-django==0.11.1
 pylint-plugin-utils==0.6
-pylint==1.9.5
+pylint==1.9.4
 pymongo==3.9.0
 pyparsing==2.4.5
 pytest-cov==2.8.1

diff --git a/requirements/test.txt b/requirements/test.txt
@@ -51,7 +51,7 @@ pyjwt==1.7.1
 pylint-celery==0.3        # via edx-lint
 pylint-django==0.11.1     # via edx-lint
 pylint-plugin-utils==0.6  # via pylint-celery, pylint-django
-pylint==1.9.5             # via edx-lint, pylint-celery, pylint-django, pylint-plugin-utils
+pylint==1.9.4             # via edx-lint, pylint-celery, pylint-django, pylint-plugin-utils
 pymongo==3.9.0
 pyparsing==2.4.5          # via packaging
 pytest-cov==2.8.1