Add CustomUvicornWorker to support dynamic root path configuration

Author: sreenathnair

app/app.py

@@ -1,6 +1,8 @@
 import time
 from contextlib import asynccontextmanager
+import os
 
+from uvicorn.workers import UvicornWorker
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.gzip import GZipMiddleware
@@ -94,6 +96,13 @@ async def add_extra_headers(request: Request, call_next):
     return response
 
 
+class CustomUvicornWorker(UvicornWorker):
+    CONFIG_KWARGS = {}
+    root_path = os.environ.get("ROOT_PATH", "")
+    if root_path:
+        CONFIG_KWARGS.update({"root_path": root_path})
+
+
 def custom_openapi():
     if not app.openapi_schema:
         app.openapi_schema = get_openapi(

app/gunicorn_conf.py

@@ -8,4 +8,4 @@
 keepalive = 5
 threads = 10
 workers = 3
-worker_class = "uvicorn.workers.UvicornWorker"
+worker_class = "app.app.CustomUvicornWorker"