patching register type from internal

srprash · srprash · commit b9f63271e1d5 · 2022-07-26T16:09:20.000-07:00
diff --git a/aws_xray_sdk/ext/psycopg2/patch.py b/aws_xray_sdk/ext/psycopg2/patch.py
@@ -23,6 +23,11 @@ def patch():
         'quote_ident',
         _xray_register_type_fix
     )
+    wrapt.wrap_function_wrapper(
+        'psycopg2._json',
+        'register_type',
+        _xray_register_type_fix
+    )
 
 
 def _xray_traced_connect(wrapped, instance, args, kwargs):
