使用Python、FastAPI和Psycopg 3异步Postgres

原文:使用Python、FastAPI和Psycopg 3异步Postgres - 桑鸟网

在写这篇文章的时候,由于Psycopg 3是一个相当新的版本,还没有大量的示例使用,所以我想我将演示一些东西。

pip install fastapi[all]
pip install psycopg[binary,pool]
  
import os
from contextlib import asynccontextmanager

from fastapi import FastAPI, Request
from psycopg_pool import AsyncConnectionPool

def get_conn_str():
    return f"""
    dbname={os.getenv('POSTGRES_DB')}
    user={os.getenv('POSTGRES_USER')}
    password={os.getenv('POSTGRES_PASSWORD')}
    host={os.getenv('POSTGRES_HOST')}
    port={os.getenv('POSTGRES_PORT')}
    """

@asynccontextmanager
async def lifespan(app: FastAPI):
    app.async_pool = AsyncConnectionPool(conninfo=get_conn_str())
    yield
    await app.async_pool.close()

app = FastAPI(lifespan=lifespan)

@app.get("/dogs")
async def get_red_dogs(request: Request):
    async with request.app.async_pool.connection() as conn:
        async with conn.cursor() as cur:
            await cur.execute("""
                SELECT * 
                FROM dogs
            """)
            results = await cur.fetchall()
            return results

@app.post("/dogs")
async def insert_dog(request: Request):
    try:
        async with request.app.async_pool.connection() as conn:
            async with conn.cursor() as cur:
                await cur.execute("""
                        INSERT INTO dogs (name,age,colour)
                        VALUES (%s, %s, %s); 
                    """, ("bonzo", 10, "red",))
                await conn.commit()
    except Exception:
        await conn.rollback()

应用程序启动时,初始化连接池并将其分配给应用程序对象上名为async_pool的属性。生命周期事件是当前在FastAPI中服务器启动后执行代码的推荐方式。

这样就可以在请求过程中从连接池中获取连接,而不是初始化一个新连接,初始化新连接有时可能需要很长时间。

使用上述语法对数据库执行异步操作。

参考资料:Lifespan Events - FastAPI

  • 8
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值