Migrate from built-in json to faster orjson

It is way faster than the current json parser: https://github.com/ijl/orjson?tab=readme-ov-file#performance

Author: wendigo

setup.py

@@ -84,6 +84,7 @@
     python_requires=">=3.9",
     install_requires=[
         "lz4",
+        "orjson >= 3.11.0 ; platform_python_implementation != 'PyPy'",
         "python-dateutil",
         "pytz",
         # CVE-2024-47081

tests/unit/test_client.py

@@ -9,7 +9,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import json
 import threading
 import time
 import urllib
@@ -26,6 +25,10 @@
 import gssapi
 import httpretty
 import keyring
+try:
+    import orjson as json
+except ImportError:
+    import json
 import pytest
 import requests
 from httpretty import httprettified
@@ -61,8 +64,7 @@
 
 @mock.patch("trino.client.TrinoRequest.http")
 def test_trino_initial_request(mock_requests, sample_post_response_data):
-    mock_requests.Response.return_value.json.return_value = sample_post_response_data
-
+    mock_requests.Response.return_value.text = json.dumps(sample_post_response_data)
     req = TrinoRequest(
         host="coordinator",
         port=8080,
@@ -692,7 +694,7 @@ def run(self) -> None:
 
 @mock.patch("trino.client.TrinoRequest.http")
 def test_trino_fetch_request(mock_requests, sample_get_response_data):
-    mock_requests.Response.return_value.json.return_value = sample_get_response_data
+    mock_requests.Response.return_value.text = json.dumps(sample_get_response_data)
 
     req = TrinoRequest(
         host="coordinator",
@@ -718,7 +720,7 @@ def test_trino_fetch_request(mock_requests, sample_get_response_data):
 
 @mock.patch("trino.client.TrinoRequest.http")
 def test_trino_fetch_request_data_none(mock_requests, sample_get_response_data_none):
-    mock_requests.Response.return_value.json.return_value = sample_get_response_data_none
+    mock_requests.Response.return_value.text = json.dumps(sample_get_response_data_none)
 
     req = TrinoRequest(
         host="coordinator",
@@ -744,7 +746,7 @@ def test_trino_fetch_request_data_none(mock_requests, sample_get_response_data_n
 
 @mock.patch("trino.client.TrinoRequest.http")
 def test_trino_fetch_error(mock_requests, sample_get_error_response_data):
-    mock_requests.Response.return_value.json.return_value = sample_get_error_response_data
+    mock_requests.Response.return_value.text = json.dumps(sample_get_error_response_data)
 
     req = TrinoRequest(
         host="coordinator",
@@ -1154,8 +1156,9 @@ def headers(self):
                 'X-Trino-Fake-2': 'two',
             }
 
-        def json(self):
-            return sample_get_response_data
+        @property
+        def text(self):
+            return json.dumps(sample_get_response_data)
 
     req = TrinoRequest(
         host="coordinator",

trino/client.py

@@ -39,7 +39,6 @@
 import base64
 import copy
 import functools
-import json
 import os
 import random
 import re
@@ -66,6 +65,11 @@
 from zoneinfo import ZoneInfo
 
 import lz4.block
+try:
+    import orjson as json
+except ImportError:
+    import json
+
 import requests
 import zstandard
 from requests import Response
@@ -689,7 +693,7 @@ def process(self, http_response: Response) -> TrinoStatus:
             self.raise_response_error(http_response)
 
         http_response.encoding = "utf-8"
-        response = http_response.json()
+        response = json.loads(http_response.text)
         if "error" in response and response["error"]:
             raise self._process_error(response["error"], response.get("id"))