from typing import Callable
from contextlib import asynccontextmanager

from fastapi import FastAPI

__on_startup: list[Callable] = []
__on_shutdown: list[Callable] = []

def add_startup(func: Callable):
    """
    注册一个函数，在应用启动时调用。
    
    :param func: 需要注册的函数。它应该是一个异步函数。
    """
    __on_startup.append(func)

def add_shutdown(func: Callable):
    """
    注册一个函数，在应用关闭时调用。
    
    :param func: 需要注册的函数。
    """
    __on_shutdown.append(func)

@asynccontextmanager
async def lifespan(app: FastAPI):
    """
    应用程序的生命周期管理器。
    
    此函数在应用启动时执行所有注册的启动函数，
    并在应用关闭时执行所有注册的关闭函数。
    """
    # Execute all startup functions
    for func in __on_startup:
        await func()
    
    yield
    
    # Execute all shutdown functions
    for func in __on_shutdown:
        await func()