Async generator dependencies in FastAPI

Introduction

Async generator dependencies let you run setup and cleanup code around your API calls. They help manage resources like database connections efficiently.

When you need to open a database connection before handling a request and close it after.

When you want to start a session or transaction before processing and ensure it ends properly.

When you need to acquire and release external resources like files or network connections.

When you want to run some code before and after each API call automatically.

When you want to clean up resources even if an error happens during request handling.

Syntax

FastAPI

async def dependency_name():
    # setup code here
    yield resource
    # cleanup code here

The function must be async and use yield to separate setup and cleanup.

FastAPI runs the code before yield before the request and the code after yield after the request finishes.

Examples

Open a database connection before the request and close it after.

FastAPI

async def get_db():
    db = connect_to_db()
    try:
        yield db
    finally:
        db.close()

Simple async generator that prints messages before and after the request.

FastAPI

async def lifespan():
    print('Start')
    yield
    print('End')

Sample Program

This FastAPI app uses an async generator dependency get_resource. It prints messages when setting up and cleaning up the resource. The endpoint uses the resource and returns it in JSON.

FastAPI

from fastapi import FastAPI, Depends

app = FastAPI()

async def get_resource():
    print('Setup resource')
    yield 'resource data'
    print('Cleanup resource')

@app.get('/')
async def read_root(resource: str = Depends(get_resource)):
    print('Handling request')
    return {'data': resource}

Important Notes

Always use try/finally or code after yield to ensure cleanup runs.

Async generator dependencies work well with async database clients or other async resources.

FastAPI automatically handles calling the async generator before and after the request.

Summary

Async generator dependencies help manage setup and cleanup around requests.

Use async def with yield to create them.

They are great for managing resources like database connections safely.

Practice

1. What is the main purpose of using an async generator dependency in FastAPI?

easy

A. To manage setup and cleanup actions around a request automatically

B. To speed up synchronous code execution

C. To replace all regular functions with asynchronous ones

D. To create infinite loops inside API endpoints

Solution

1. Step 1: Understand async generator dependencies

   Async generator dependencies use async def with yield to run setup code before yielding and cleanup code after.

2. Step 2: Identify their purpose in FastAPI

   This pattern is mainly used to manage resources safely, like opening and closing database connections around a request.

3. Final Answer:

   To manage setup and cleanup actions around a request automatically -> Option A

4. Quick Check:

   Async generator dependencies = setup + cleanup [OK]

Hint: Think setup before yield, cleanup after yield [OK]

Common Mistakes:

• Confusing async generators with normal async functions
• Thinking they speed up synchronous code
• Assuming they replace all functions

2. Which of the following is the correct syntax to define an async generator dependency in FastAPI?

easy

A. async def resource(): yield 'data'

B. def resource(): yield 'data'

C. async def resource(): return 'data'

D. def resource(): return 'data'

Solution

1. Step 1: Identify async generator syntax

   An async generator uses async def and yield inside the function body.

2. Step 2: Compare options

   async def resource(): yield 'data' uses async def and yield, which is correct. Options A and B are synchronous, and C uses return instead of yield.

3. Final Answer:

   async def resource(): yield 'data' -> Option A

4. Quick Check:

   Async + yield = async generator [OK]

Hint: Async generator = async def + yield inside [OK]

Common Mistakes:

• Using return instead of yield
• Forgetting async keyword
• Using synchronous def for async generator

3. Given this FastAPI async generator dependency, what will be printed when the endpoint is called?

async def resource():
    print('Setup')
    yield 'data'
    print('Cleanup')

@app.get('/')
async def read(resource=Depends(resource)):
    print('Inside endpoint')
    return resource

medium

A. Inside endpoint only

B. Inside endpoint, Setup, Cleanup

C. Setup, Cleanup, Inside endpoint

D. Setup, Inside endpoint, Cleanup

Solution

1. Step 1: Understand async generator flow

   When the endpoint is called, FastAPI runs the async generator up to the yield (prints 'Setup'), then runs the endpoint code (prints 'Inside endpoint'), then after response finishes, runs code after yield (prints 'Cleanup').

2. Step 2: Order of prints

   The prints happen in order: 'Setup', 'Inside endpoint', 'Cleanup'.

3. Final Answer:

   Setup, Inside endpoint, Cleanup -> Option D

4. Quick Check:

   Yield pauses, so Setup before, Cleanup after endpoint [OK]

Hint: Setup prints before yield, cleanup after endpoint runs [OK]

Common Mistakes:

• Assuming cleanup runs before endpoint
• Thinking yield returns immediately
• Ignoring async generator lifecycle

4. Identify the error in this async generator dependency code:

async def resource():
    connection = await connect_db()
    yield connection
    await connection.close()

@app.get('/')
async def read(conn=Depends(resource)):
    return {'status': 'ok'}

medium

A. Missing await before yield

B. Cannot yield inside async def

C. Not using try-finally to ensure cleanup

D. Depends should not be used with async generators

Solution

1. Step 1: Check resource cleanup safety

   Without try-finally, if an exception occurs, await connection.close() might not run, causing resource leaks.

2. Step 2: Proper pattern for async generator cleanup

   Use try-finally around yield to guarantee cleanup runs even on errors.

3. Final Answer:

   Not using try-finally to ensure cleanup -> Option C

4. Quick Check:

   Always use try-finally for cleanup in async generators [OK]

Hint: Use try-finally to guarantee cleanup after yield [OK]

Common Mistakes:

• Forgetting try-finally causes resource leaks
• Thinking yield needs await
• Believing Depends can't use async generators

5. You want to create an async generator dependency that opens a database connection, yields it, and ensures it closes even if the endpoint raises an error. Which code snippet correctly implements this?

hard

A. async def db():
conn = await open_db()
yield conn
await conn.close()

B. async def db():
conn = await open_db()
try:
yield conn
finally:
await conn.close()

C. async def db():
conn = await open_db()
yield conn
conn.close()

D. async def db():
conn = open_db()
yield conn
await conn.close()

Solution

1. Step 1: Ensure proper async resource management

   Opening the connection must be awaited, and closing must happen even if errors occur.

2. Step 2: Use try-finally around yield

   Only async def db(): conn = await open_db() try: yield conn finally: await conn.close() uses try-finally to guarantee await conn.close() runs after yield, even on exceptions.

3. Final Answer:

   async def db(): conn = await open_db() try: yield conn finally: await conn.close() -> Option B

4. Quick Check:

   Try-finally ensures cleanup after yield [OK]

Hint: Wrap yield in try-finally to always close resources [OK]

Common Mistakes:

• Not awaiting open_db()
• Not using try-finally for cleanup
• Calling close() without await