Use orjson to de-/serialize json-rpc messages (#2513)

Author: deathaxe

.github/workflows/main.yml

@@ -41,7 +41,7 @@ jobs:
           python-version: '3.8'
       - run: sudo apt update
       - run: sudo apt install --no-install-recommends -y x11-xserver-utils
-      - run: pip3 install mypy==1.7.1 flake8==5.0.4 pyright==1.1.381 --user
+      - run: pip3 install mypy==1.7.1 flake8==5.0.4 pyright==1.1.381 orjson==3.10.7 --user
       - run: echo "$HOME/.local/bin" >> $GITHUB_PATH
       - run: mypy stubs
       - run: flake8 plugin tests

dependencies.json

@@ -3,6 +3,7 @@
         ">=4096": [
             "bracex",
             "mdpopups",
+            "orjson",
             "typing_extensions",
             "wcmatch"
         ]

plugin/core/transports.py

@@ -17,6 +17,10 @@
 import time
 import weakref
 
+try:
+    import orjson
+except ImportError:
+    orjson = None
 
 T = TypeVar('T')
 T_contra = TypeVar('T_contra', contravariant=True)
@@ -80,6 +84,8 @@ def read_data(self, reader: IO[bytes]) -> dict[str, Any] | None:
 
     @staticmethod
     def _encode(data: dict[str, Any]) -> bytes:
+        if orjson:
+            return orjson.dumps(data)
         return json.dumps(
             data,
             ensure_ascii=False,
@@ -90,6 +96,8 @@ def _encode(data: dict[str, Any]) -> bytes:
 
     @staticmethod
     def _decode(message: bytes) -> dict[str, Any]:
+        if orjson:
+            return orjson.loads(message)
         return json.loads(message.decode('utf-8'))
 
 

tox.ini

@@ -23,6 +23,7 @@ deps =
     mypy==1.7.1
     flake8==5.0.4
     pyright==1.1.381
+    orjson==3.10.7
 commands =
     # mypy disabled for main code as it doesn't currently support cyclic definitions - https://github.com/python/mypy/issues/731
     mypy stubs