Replace ujson with orjson for JSON serialization

- Update pyproject.toml dependencies to require orjson instead of ujson.
- Update import fallbacks in pylsp/python_lsp.py, pylsp/__main__.py, and
  pylsp/plugins/pylint_lint.py.
- Handle orjson-specific behavior (bytes output in dumps, absence of
  load method).

Fixes: #673
Signed-off-by: Matěj Cepl <[email protected]>

Author: mcepl

pylsp/__main__.py

@@ -8,7 +8,7 @@
 import time
 
 try:
-    import ujson as json
+    import orjson as json
 except Exception:
     import json
 
@@ -99,7 +99,10 @@ def _configure_logger(verbose=0, log_config=None, log_file=None) -> None:
 
     if log_config:
         with open(log_config, encoding="utf-8") as f:
-            logging.config.dictConfig(json.load(f))
+            if json.__name__ == "orjson":
+                logging.config.dictConfig(json.loads(f.read()))
+            else:
+                logging.config.dictConfig(json.load(f))
     else:
         formatter = logging.Formatter(LOG_FORMAT)
         if log_file:

pylsp/plugins/pylint_lint.py

@@ -15,7 +15,7 @@
 from pylsp import hookimpl, lsp
 
 try:
-    import ujson as json
+    import orjson as json
 except Exception:
     import json
 

pylsp/python_lsp.py

@@ -11,7 +11,7 @@
 from typing import Any
 
 try:
-    import ujson as json
+    import orjson as json
 except Exception:
     import json
 
@@ -152,7 +152,10 @@ async def pylsp_ws(websocket):
         def send_message(message, websocket):
             """Handler to send responses of  processed requests to respective web socket clients"""
             try:
-                payload = json.dumps(message, ensure_ascii=False)
+                if json.__name__ == "orjson":
+                    payload = json.dumps(message).decode("utf-8")
+                else:
+                    payload = json.dumps(message, ensure_ascii=False)
                 loop.call_soon_threadsafe(send_queue.put_nowait, (payload, websocket))
             except Exception as e:
                 log.exception("Failed to write message %s, %s", message, str(e))

pyproject.toml

@@ -19,7 +19,7 @@ dependencies = [
     "jedi>=0.17.2,<0.20.0",
     "pluggy>=1.0.0",
     "python-lsp-jsonrpc>=1.1.0,<2.0.0",
-    "ujson>=3.0.0",
+    "orjson",
     "black"
 ]
 dynamic = ["version"]