主页 / 问题 / 1583050
Accepted
mrBars1k
Asked:2024-06-07 16:35:39 +0000 UTC2024-06-07 16:35:39 +0000 UTC

无法异步连接到机器人和数据库

  • 772

我正在尝试将Aiogram3和asyncpg结合起来，但我对异步缺乏理解。

我尝试以这种方式启动数据库和机器人：

dp = Dispatcher()
adb = DB(DSN)


@dp.message(CommandStart)
async def start(message: Message):
    async with adb.pool.acquire() as conn: # взять подключение из пула и потом вернуть обратно;
        x = await conn.fetch('SELECT * FROM manga;')
    print(x)
    await message.answer('Работает.')


async def main():
    bot = Bot(token=TOKEN)

    await adb.connect()
    await dp.start_polling(bot)

if __name__ == '__main__':
    asyncio.run(main())

class DB:
    def __init__(self, dsn):
        self.dsn = dsn
        self.pool = None

    async def connect(self):
        self.pool = await asyncpg.create_pool(self.dsn) # создать пулл подключений;

    async def close(self):
        await self.pool.close()

但我意识到这里没有异步的味道，两个协程相互阻塞并按顺序工作，而不是异步。

原则上，我还没有找到任何有价值的使用示例asyncpg，尽管该库在所有类似库中的速度似乎是无与伦比的，因为它不使用文本指令，而是通过 C 访问字节代码。

如果有任何最佳实践，我会很高兴阅读它。

python

2 个回答

  1. 或许这个方法能帮助解决问题。我以你的代码为基础，所以类是相同的。

    class DB:

    def __init__(self, dsn):
        self.dsn = dsn
        self.pool = None

    async def get_connection(self):
        return self.pool

    async def connect(self):
        self.pool = await asyncpg.create_pool(self.dsn)
            
    async def close(self):
        await self.pool.close()

    我们指示连接参数，在我们的例子中是 dns postgres://user:password@host:port/database。

    db = DB("postgres://user:password@host:port/database")

    从变量 中，db我们调用创建池的方法connect()，然后调用get_connection()获取池。然后我们就可以使用数据库了。

    async def main():
    await db.connect()

    pool = await db.get_connection()

    async with pool.acquire() as conn:
        print(await conn.fetchrow("""SELECT * FROM users"""))


asyncio.run(main())
    • 1
  2. Best Answer

    通过创建任务解决了这个问题：

    async def main():
    bot = Bot(token=TOKEN)

    task1 = asyncio.create_task(adb.connect())
    task2 = asyncio.create_task(dp.start_polling(bot))

    await task1 
    await task2

if __name__ == '__main__':
    asyncio.run(main())

    另外，使用gather，我设法做了同样的事情，但更短更简单。

    async def main():
    bot = Bot(token=TOKEN)

    task1 = asyncio.create_task(adb.connect())
    task2 = asyncio.create_task(dp.start_polling(bot))

    await asyncio.gather(task1, task2)

if __name__ == '__main__':
    asyncio.run(main())

    但即使使用这种方法，有时也会出现没有这种方法的错误.acquire()，因为当机器人已经运行时，数据库没有时间创建连接池。或类似的东西。

    • 0

相关问题