Use a faster JSON library

twistlock/datadog_checks/twistlock/twistlock.py

@@ -14,6 +14,11 @@
 from .config import Config
 from .utils import normalize_api_data_inplace
 
+try:
+    import orjson as json
+except ImportError:
+    import json
+
 DOCKERIO_PREFIX = "docker.io/"
 
 # min_severity tag allows to query number of vulns for a severity target easily
@@ -322,7 +327,7 @@ def _retrieve_json(self, path):
         try:
             # it's possible to get a null response from the server
             # {} is a bit easier to deal with
-            j = response.json() or {}
+            j = json.loads(response.content) or {}
             if 'err' in j:
                 err_msg = "Error in response: {}".format(j.get("err"))
                 self.log.error(err_msg)

twistlock/requirements.in

@@ -1 +1,2 @@
-python-dateutil==2.8.0
+python-dateutil==2.8.0
+orjson==2.6.1; python_version > '3.0'

twistlock/tests/test_twistlock.py

@@ -8,6 +8,7 @@
 import mock
 import pytest
 
+from datadog_checks.base import ensure_bytes
 from datadog_checks.dev import get_here
 from datadog_checks.twistlock import TwistlockCheck
 
@@ -47,6 +48,10 @@ def __init__(self, j):
         self._json = j
         self.status_code = 200
 
+    @property
+    def content(self):
+        return ensure_bytes(self._json)
+
     def json(self):
         return json.loads(self._json)