Drop ujson and switch to orjson

Author: rumpelsepp

pylsp/__main__.py

@@ -2,16 +2,12 @@
 # Copyright 2021- Python Language Server Contributors.
 
 import argparse
+import json
 import logging
 import logging.config
 import sys
 import time
 
-try:
-    import ujson as json
-except Exception:
-    import json
-
 from ._version import __version__
 from .python_lsp import (
     PythonLSPServer,

pylsp/plugins/pylint_lint.py

@@ -15,8 +15,8 @@
 from pylsp import hookimpl, lsp
 
 try:
-    import ujson as json
-except Exception:
+    import orjson as json
+except ImportError:
     import json
 
 log = logging.getLogger(__name__)

pylsp/python_lsp.py

@@ -9,15 +9,15 @@
 from functools import partial
 from typing import Any, Dict, List
 
-try:
-    import ujson as json
-except Exception:
-    import json
-
 from pylsp_jsonrpc.dispatchers import MethodDispatcher
 from pylsp_jsonrpc.endpoint import Endpoint
 from pylsp_jsonrpc.streams import JsonRpcStreamReader, JsonRpcStreamWriter
 
+try:
+    import orjson as json
+except ImportError:
+    import json
+
 from . import _utils, lsp, uris
 from ._version import __version__
 from .config import config

pyproject.toml

@@ -18,7 +18,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>=3.10.0",
 ]
 dynamic = ["version"]