如何在FastAPI中實現資料庫連線與事務處理

如何在FastAPI中實現資料庫連接和事務處理

引言：
隨著網路應用程式的快速發展，資料庫連接和事務處理成為了一個非常重要的主題。 FastAPI是一個高效能的Python Web框架，因其快速且易於使用而受到開發者的喜愛。在本文中，我們將介紹如何在FastAPI中實現資料庫連接和事務處理，以協助您建立可靠且高效的網路應用程式。

第一部分：資料庫連線
在FastAPI中，可以使用各種資料庫連線工具，例如SQLAlchemy和Tortoise-ORM。我們將以SQLAlchemy為例進行示範。

1. 安裝SQLAlchemy和MySQL驅動程式
   首先，我們需要安裝SQLAlchemy函式庫和對應的MySQL驅動程式。您可以使用以下命令進行安裝：

pip install sqlalchemy
pip install mysql-connector-python

2. 建立資料庫連線
   在FastAPI應用程式中，我們可以在主應用程式中建立資料庫連線並將其註入到路由處理程序中。以下是一段範例程式碼：

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中使用事務處理。

1. 使用依賴項注入資料庫會話
   我們將在這個範例中繼續使用先前建立的get_db依賴項函數，並在路由處理程序中註入資料庫會話。

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的函數，該函數將在資料庫中建立一個新使用者。

2. 實作交易處理
   下面是一個範例的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中文網其他相關文章！