主页 / 问题 / 1590739
Accepted
Edmindo
Asked:2024-08-15 07:27:38 +0000 UTC2024-08-15 07:27:38 +0000 UTC

如何在异步模式下正确使用psql数据库 |电报机器人 |爱图

  • 772

我开始使用 psql 数据库开发 tg 机器人，出现了一个问题：

如何在异步模式下正确使用psql数据库，从而不做大量昂贵的操作并合理使用资源。

大家好

我最近开始在aiogram 3中编写一个电报机器人。需要使用psql数据库。首先，我决定弄清楚如何在同步模式下工作，这里一切看起来都很简单 - 我们在起点创建一个永久连接，并为每个请求创建一个游标，在请求结束时关闭它。

在编写测试处理程序后，我决定尝试让机器人通过此处理程序，在短时间内通过对处理程序的 7-8 次调用以及对数据库的请求成功完成了这一任务。好吧，我并没有指望异步机器人中有同步解决方案。我开始研究异步模式工作的问题，然后问题出现了：

如何在异步模式下正确使用psql数据库，从而不做大量昂贵的操作并合理使用资源。

我阅读了在互联网上找到的一些资源、文档，并询问了 GPT。在整个示例中，为每个请求创建一个单独的连接。看来这并不是很理性。

我在一个处理程序中的典型情况是从消息中获取用户 ID，使用该 id 查询数据库并从表中提取具有该 id 的记录。大约有 10-12 个这样的处理程序。

我很高兴看到代码示例

postgresql

1 个回答

  1. Best Answer

    对于此任务，使用asyncpg更正确：

    为了避免为每个请求创建单独的连接，
    您可以从池中获取连接。
    创建一个类并Database通过方法fetch_user_data发出异步请求。

    基本示例：

    import asyncio
from aiogram import Bot, Dispatcher, types
from aiogram.contrib.fsm_storage.memory import MemoryStorage
import asyncpg

API_TOKEN = 'YOUR_API_TOKEN'
DSN = 'postgresql://user:password@localhost/dbname'

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

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

    async def fetch_user_data(self, user_id):
        async with self.pool.acquire() as connection:
            result = await connection.fetch('SELECT * FROM users WHERE id = $1', user_id)
            return result

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


async def on_startup(dp):
    # Инициализация базы данных
    await db.init()


async def on_shutdown(dp):
    # Закрытие базы данных
    await db.close()


async def handle_start(message: types.Message):
    user_id = message.from_user.id
    user_data = await db.fetch_user_data(user_id)
    await message.answer(f"Данные пользователя: {user_data}")


async def main():
    # Создание экземпляра бота и диспетчера
    bot = Bot(token=API_TOKEN)
    storage = MemoryStorage()
    dp = Dispatcher(bot, storage=storage)

    # Регистрация обработчиков
    dp.register_message_handler(handle_start, commands=["start"])

    # Запуск бота
    await dp.start_polling(on_startup=on_startup, on_shutdown=on_shutdown)

if __name__ == '__main__':
    db = Database(DSN)
    asyncio.run(main())
    • 2

相关问题