Performance improvements

pyproject.toml

@@ -13,6 +13,7 @@ dependencies = [
     "packaging>=24.1",
     "typer>=0.12.5",
     "uvicorn>=0.32.0",
+    "uvloop>=0.21.0",
     "werkzeug>=3.0.4",
     "whenever~=0.6.9",
 ]
@@ -31,4 +32,4 @@ dev-dependencies = [
 ]
 
 [tool.ruff.lint]
-extend-select = ["I"]
+extend-select = ["I"]

src/pip_timemachine/main.py

@@ -78,11 +78,13 @@ def filter_files_by_moment(files: list[dict]) -> tuple[list, dict]:
         modified_versions[version] = None
     return modified_files, modified_versions
 
+
 def initiate_shutdown():
     logger.info("Gracefully shutting down")
     if UNICORN_SERVER:
         UNICORN_SERVER.should_exit = True
 
+
 @app.get("/shutdown-pip-timemachine-server")
 async def shutdown_pip_timemachine_server(background_tasks: BackgroundTasks):
     """
@@ -145,7 +147,9 @@ def run_server(moment: dt.datetime, index: str = INDEX, port: int = 8040):
     INDEX = index.rstrip("/")
 
     UNICORN_SERVER = uvicorn.Server(
-        uvicorn.Config("pip_timemachine.main:app", port=port)
+        uvicorn.Config(
+            "pip_timemachine.main:app", port=port, workers=4, timeout_keep_alive=15
+        )
     )
     UNICORN_SERVER.run()