fastapi使用上下文管理db连接,不用来回传递db实例

  1. 问题
    之前接口会通过depends依赖注入一个db连接实例,由于接口中调用了其他函数,用到了这个db,所以需要将db作为参数进行传递,但不够优雅,大体如下
from fastapi import APIRouter, HTTPException, Path, Depends
from sqlalchemy.orm import Session
from services.mysql.bar import get_bar_by_id
from services.mysql.process_x import bar_process
from databases.mysql import get_db

router = APIRouter(prefix='/foo')


@router.get('/bar/{bar_id}')
def process_bar(bar_id: int = Path(..., title='The ID of the bar to process', ge=1),
                          mysql_session: Session = Depends(get_db)):
    # From the crontroller, to a service which only runs a query. This is fine.
    bar = get_bar_by_id(bar_id, mysql_session)

    if bar is None:
        raise HTTPException(status_code=404,
                            detail='Bar not found for id: {bar_id}'.format(bar_id=bar_id))

    # This one calls a function in a service which has a lot of business logic but no queries
    processed_bar = bar_process(bar, mysql_session)

    return processed_bar
  1. 改造,使用上下文管理器管理db
from sqlalchemy.orm import sessionmaker,Session
from sqlalchemy import create_engine
from contextvars import ContextVar
SQLALCHEMY_DATABASE_URL = "sqlite:///db_test_3.db"
# SQLALCHEMY_DATABASE_URL = "postgresql://user:password@postgresserver/db"

engine = create_engine(
    SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})
# 仅适用于SQLite。其他数据库不需要。 链接参数:检查同一条线? 即需要可多线程

# 通过sessionmaker得到一个类,一个能产生session的工厂。
SessionLocal = sessionmaker(autocommit=False, autoflush=True, bind=engine) 
             # 会话生成器   自动提交            自动刷新,这个很重要,对于自增主键,在add后还没commit,只有将这个设置为True,才会拿到主键

def get_db():
    db: Session = SessionLocal()
    try:
        yield db
        db.commit()
    except Exception:
        db.rollback()
    finally:
        db.close()

db_session: ContextVar[Session] = ContextVar('db_session')

  1. 注入db
from fastapi import FastAPI,Depends, Path
from sqlalchemy.orm import Session
from connection import get_db
from models import Student
from commonmethod import getvalue
from connection import db_session
import uvicorn
app = FastAPI()


@app.get('/test/{id}')
async def test(id:int=Path(...),db:Session=Depends(get_db)):
    db_session.set(db)
    stu = db.query(Student).filter(Student.id==id).first()
    addres = getvalue(stu)
    return stu
  1. 在函数中使用db
from sqlalchemy.orm import Session
from models import Address,Student
from connection import db_session
async def getvalue(stu:Student):
    context_aware_session = db_session.get() #主要是这句话
    value = context_aware_session.query(Address).filter(Address.student_id==stu.id).first()
    return value

https://stackoverflow.com/questions/66464098/using-a-db-dependency-in-fastapi-without-having-to-pass-it-through-a-function-tr

https://python-dependency-injector.ets-labs.org/examples/fastapi-sqlalchemy.html

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
`databaseQueueWithPath` 是 FMDB 库提供的一个创建并返回一个 `FMDatabaseQueue` 对象的方法,该对象用于执行数据库查询和更新操作,支持多线程并发访问。 以下是一个使用 `databaseQueueWithPath` 的示例: ```objective-c NSString *dbPath = [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) objectAtIndex:0]; dbPath = [dbPath stringByAppendingPathComponent:@"myDatabase.sqlite"]; FMDatabaseQueue *dbQueue = [FMDatabaseQueue databaseQueueWithPath:dbPath]; [dbQueue inDatabase:^(FMDatabase *db) { // 在这里执行数据库查询和更新操作 // 例如:[db executeUpdate:@"CREATE TABLE IF NOT EXISTS myTable(id integer primary key autoincrement, name text, age integer)"]; // 例如:FMResultSet *result = [db executeQuery:@"SELECT * FROM myTable"]; }]; ``` 在上面的示例中,我们首先获取了应用程序沙盒中的 Documents 目录,然后在该目录下创建了一个名为 `myDatabase.sqlite` 的 SQLite 数据库文件,并将其作为参数传递给 `databaseQueueWithPath` 方法来创建一个 `FMDatabaseQueue` 对象。然后,我们通过调用 `inDatabase` 方法来执行一些数据库查询和更新操作,这些操作将在 `FMDatabase` 对象的上下文中执行,以确保线程安全。在 `inDatabase` 方法中,我们可以像使用单个 `FMDatabase` 对象一样使用 `db` 对象来执行 SQL 语句。 希望这个示例能帮助您理解如何使用 `databaseQueueWithPath` 方法。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值