AISupplyAgent/applications/resource/resource_manager.py

from utils.mysql import DatabaseManager


class ResourceManager:
    def __init__(self, env):
        self.env = env
        self.mysql_client: DatabaseManager | None = None

    async def startup(self):
        # 初始化 MySQL
        self.mysql_client = DatabaseManager(self.env)
        await self.mysql_client.init_pools()
        print("MySQL connected")

    async def shutdown(self):
        # 关闭 MySQL
        if self.mysql_client:
            await self.mysql_client.close_pools()
            print("Mysql closed")


_resource_manager: ResourceManager | None = None


def init_resource_manager(env):
    global _resource_manager
    if _resource_manager is None:
        _resource_manager = ResourceManager(env)
    return _resource_manager


def get_resource_manager() -> ResourceManager:
    return _resource_manager