如何在FastAPI中實現資料庫連接和事務處理
引言：
隨著網路應用程式的快速發展，資料庫連接和事務處理成為了一個非常重要的主題。 FastAPI是一個高效能的Python Web框架，因其快速且易於使用而受到開發者的喜愛。在本文中，我們將介紹如何在FastAPI中實現資料庫連接和事務處理，以協助您建立可靠且高效的網路應用程式。
第一部分：資料庫連線
在FastAPI中，可以使用各種資料庫連線工具，例如SQLAlchemy和Tortoise-ORM。我們將以SQLAlchemy為例進行示範。
pip install sqlalchemy pip install mysql-connector-python
from fastapi import Depends, FastAPI
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm.session import Session
app = FastAPI()
SQLALCHEMY_DATABASE_URL = "mysql+mysqlconnector://user:password@localhost/db_name"
engine = create_engine(
SQLALCHEMY_DATABASE_URL,
pool_size=10,
max_overflow=20
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()
async def get_db() -> Session:
try:
db = SessionLocal()
yield db
finally:
db.close()在上述程式碼中，我們先建立了資料庫連線的URL，然後使用SQLAlchemy的create_engine函式建立資料庫引擎。接下來，我們定義了一個SessionLocal類，用於維護資料庫會話。最後，我們建立了一個依賴項函數get_db，它將傳回一個資料庫會話實例，在每個請求處理程序中使用。
第二部分：交易處理
在FastAPI中，可以使用交易處理來確保資料庫操作的原子性和一致性。以下是一些範例程式碼，示範如何在FastAPI中使用事務處理。
from fastapi import Depends, FastAPI
from sqlalchemy.orm import Session
from .models import User
from .crud import create_user
from .database import get_db
app = FastAPI()
@app.post("/users/")
async def create_user_handler(user: User, db: Session = Depends(get_db)):
return create_user(db=db, user=user)在上述程式碼中，我們定義了一個路由處理程序create_user_handler，它接受一個使用者物件和資料庫會話作為參數。在該處理程序中，我們呼叫了一個名為create_user的函數，該函數將在資料庫中建立一個新使用者。
from sqlalchemy.orm import Session
from .models import User
def create_user(db: Session, user: User):
try:
db.begin()
db.add(user)
db.commit()
db.refresh(user)
return user
except:
db.rollback()
raise在上述程式碼中，我們先使用db.begin()語句開始一個事務。然後，我們使用db.add()語句將使用者物件加入到資料庫會話中。接下來，我們使用db.commit()語句將交易提交到資料庫。如果在提交過程中發生錯誤，我們使用db.rollback()語句回滾交易。最後，我們使用db.refresh()語句來刷新使用者對象，以確保它包含了資料庫中產生的主鍵值。
結束語：
在本文中，我們介紹如何在FastAPI中實作資料庫連線和交易處理。我們首先使用SQLAlchemy建立了資料庫連接，然後使用依賴項函數將資料庫會話注入到路由處理程序中。接下來，我們示範如何在資料庫會話中使用事務處理來確保資料庫操作的原子性和一致性。透過學習和實踐這些技術，您將能夠建立可靠且高效的Web應用程式。
以上是如何在FastAPI中實現資料庫連線與事務處理的詳細內容。更多資訊請關注PHP中文網其他相關文章！
