如何使用 FastAPI + uvicorn 在工作人员之间共享数据库连接？

Question

我正在尝试使用 FastAPI + uvicorn 创建一个应用程序。

此应用程序必须能够处理同时连接。我不能保证所有代码都可以以异步/等待方式执行。

然后，我想使用--workers Xuvicorn 中的选项来处理同时连接，但我需要在所有工作人员之间共享相同的数据库连接。

我尝试了以下示例：

import time
from pymongo.database import Database
from fastapi import Depends, FastAPI
from dynaconf import settings
from pymongo import MongoClient

print("-------> Creating a new MongoDB connection")
db_conn = MongoClient(settings.MONGODB_URI)
db = db_conn.get_database('mydb')

app = FastAPI()

def get_db():
    return db

@app.get("/{id}")
async def main(id: str, db: Database = Depends(get_db)):
    print("Recebido id: " + id)
    time.sleep(10)
    return { 'id': id }

$uvicorn main:app --workers 2
INFO:     Uvicorn running on http://127.0.0.1:8000 (Press CTRL+C to quit)
INFO:     Started parent process [24730]
-------> Creating a new MongoDB connection
-------> Creating a new MongoDB connection
INFO:     Started server process [24733]
INFO:     Waiting for application startup.
INFO:     Application startup complete.
INFO:     Started server process [24732]
INFO:     Waiting for application startup.
INFO:     Application startup complete.

但是我得到了两个 mongodb 连接。

如何共享 MongoDB 连接并避免为每个工作人员创建连接？

Answer 1

您不得共享连接，因为它是有状态的。两个或多个进程无法成功使用单个套接字连接。