[FASTAPI] Bump fastapi to 0.93.0 (#7991)

Introduce lifespan events

Author: Micael Malta

frameworks/Python/fastapi/app.py

@@ -1,3 +1,6 @@
+import multiprocessing
+from contextlib import asynccontextmanager
+
 import asyncpg
 import os
 from fastapi import FastAPI, Request
@@ -15,6 +18,8 @@
 READ_ROW_SQL = 'SELECT "id", "randomnumber" FROM "world" WHERE id = $1'
 WRITE_ROW_SQL = 'UPDATE "world" SET "randomnumber"=$1 WHERE id=$2'
 ADDITIONAL_ROW = [0, "Additional fortune added at request time."]
+MAX_POOL_SIZE = 1000//multiprocessing.cpu_count()
+MIN_POOL_SIZE = max(int(MAX_POOL_SIZE / 2), 1)
 
 
 def get_num_queries(queries):
@@ -34,8 +39,6 @@ def get_num_queries(queries):
 
 templates = Jinja2Templates(directory="templates")
 
-app = FastAPI()
-
 
 async def setup_database():
     return await asyncpg.create_pool(
@@ -44,17 +47,21 @@ async def setup_database():
         database="hello_world",
         host="tfb-database",
         port=5432,
+        min_size=MIN_POOL_SIZE,
+        max_size=MAX_POOL_SIZE,
     )
 
 
-@app.on_event("startup")
-async def startup_event():
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Setup the database connection pool
     app.state.connection_pool = await setup_database()
+    yield
+    # Close the database connection pool
+    await app.state.connection_pool.close()
 
 
-@app.on_event("shutdown")
-async def shutdown_event():
-    await app.state.connection_pool.close()
+app = FastAPI(lifespan=lifespan)
 
 
 @app.get("/json")

frameworks/Python/fastapi/app_orm.py

@@ -1,6 +1,7 @@
 import logging
 import multiprocessing
 import os
+from contextlib import asynccontextmanager
 from operator import attrgetter
 from random import randint, sample
 
@@ -42,6 +43,7 @@ class Fortune(Base):
 ADDITIONAL_FORTUNE = Fortune(
     id=0, message="Additional fortune added at request time."
 )
+MAX_POOL_SIZE = 1000//multiprocessing.cpu_count()
 
 sort_fortunes_key = attrgetter("message")
 
@@ -50,8 +52,6 @@ class Fortune(Base):
 )
 templates = Jinja2Templates(directory=template_path)
 
-app = FastAPI()
-
 
 async def setup_database():
     dsn = "postgresql+asyncpg://%s:%s@tfb-database:5432/hello_world" % (
@@ -62,13 +62,26 @@ async def setup_database():
     engine = create_async_engine(
         dsn,
         future=True,
+        pool_size=MAX_POOL_SIZE,
         connect_args={
             "ssl": False  # NEEDED FOR NGINX-UNIT OTHERWISE IT FAILS
         },
     )
     return sessionmaker(engine, class_=AsyncSession)
 
 
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Setup the database connection pool
+    app.state.db_session = await setup_database()
+    yield
+    # Close the database connection pool
+    await app.state.db_session.close()
+
+
+app = FastAPI(lifespan=lifespan)
+
+
 def get_num_queries(queries):
     try:
         query_count = int(queries)
@@ -82,16 +95,6 @@ def get_num_queries(queries):
     return query_count
 
 
-@app.on_event("startup")
-async def startup_event():
-    app.state.db_session = await setup_database()
-
-
-@app.on_event("shutdown")
-async def shutdown_event():
-    await app.state.db_session.close()
-
-
 @app.get("/json")
 async def json_serialization():
     return UJSONResponse({"message": "Hello, world!"})

frameworks/Python/fastapi/requirements.txt

@@ -1,4 +1,4 @@
 asyncpg==0.27.0
-fastapi==0.92.0
+fastapi==0.93.0
 Jinja2==3.1.2
 ujson==5.7.0