From f2df86c946c0418655e2b60c3f3177afad48076b Mon Sep 17 00:00:00 2001
From: Zachary Hancock <zhancock@edx.org>
Date: Tue, 23 Jan 2024 11:36:58 -0500
Subject: [PATCH] feat: additional performance tracing (#443)

---
 CHANGELOG.rst                        | 4 ++++
 lti_consumer/__init__.py             | 2 +-
 lti_consumer/lti_1p3/consumer.py     | 1 +
 lti_consumer/lti_1p3/key_handlers.py | 3 +++
 4 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
index f6fe13f0..37d1839b 100644
--- a/CHANGELOG.rst
+++ b/CHANGELOG.rst
@@ -15,6 +15,10 @@ Please See the `releases tab <https://github.com/openedx/xblock-lti-consumer/rel
 
 Unreleased
 ~~~~~~~~~~
+9.8.3 - 2024-01-23
+------------------
+* Additional NewRelic traces to functions suspected of causing performance issues.
+
 9.8.2 - 2024-01-19
 ------------------
 * Add NewRelic traces to functions suspected of causing performance issues.
diff --git a/lti_consumer/__init__.py b/lti_consumer/__init__.py
index 02aa0744..cd9ee742 100644
--- a/lti_consumer/__init__.py
+++ b/lti_consumer/__init__.py
@@ -4,4 +4,4 @@
 from .apps import LTIConsumerApp
 from .lti_xblock import LtiConsumerXBlock
 
-__version__ = '9.8.2'
+__version__ = '9.8.3'
diff --git a/lti_consumer/lti_1p3/consumer.py b/lti_consumer/lti_1p3/consumer.py
index 9d4a099e..32a618fc 100644
--- a/lti_consumer/lti_1p3/consumer.py
+++ b/lti_consumer/lti_1p3/consumer.py
@@ -34,6 +34,7 @@ class LtiConsumer1p3:
     LTI 1.3 Consumer Implementation
     """
 
+    @function_trace('lti_consumer.lti_1p3.consumer.LtiConsumer1p3.__init__')
     def __init__(
             self,
             iss,
diff --git a/lti_consumer/lti_1p3/key_handlers.py b/lti_consumer/lti_1p3/key_handlers.py
index 78601e7f..0d530df7 100644
--- a/lti_consumer/lti_1p3/key_handlers.py
+++ b/lti_consumer/lti_1p3/key_handlers.py
@@ -11,6 +11,7 @@
 import logging
 
 from Cryptodome.PublicKey import RSA
+from edx_django_utils.monitoring import function_trace
 from jwkest import BadSignature, BadSyntax, WrongNumberOfParts, jwk
 from jwkest.jwk import RSAKey, load_jwks_from_url
 from jwkest.jws import JWS, NoSuitableSigningKeys, UnknownAlgorithm
@@ -32,6 +33,7 @@ class ToolKeyHandler:
     in order to validate the JWT Signature of messages
     signed with the tools signature.
     """
+    @function_trace('lti_consumer.key_handlers.ToolKeyHandler.__init__')
     def __init__(self, public_key=None, keyset_url=None):
         """
         Instance message validator
@@ -162,6 +164,7 @@ class PlatformKeyHandler:
     This class loads the platform key and is responsible for
     encoding JWT messages and exporting public keys.
     """
+    @function_trace('lti_consumer.key_handlers.PlatformKeyHandler.__init__')
     def __init__(self, key_pem, kid=None):
         """
         Import Key when instancing class if a key is present.