FastAPI에서 데이터베이스 연결 및 트랜잭션 처리를 구현하는 방법

FastAPI에서 데이터베이스 연결 및 트랜잭션 처리를 구현하는 방법

소개:
웹 애플리케이션의 급속한 발전과 함께 데이터베이스 연결 및 트랜잭션 처리가 매우 중요한 주제가 되었습니다. FastAPI는 속도와 사용 편의성으로 인해 개발자들에게 사랑받는 고성능 Python 웹 프레임워크입니다. 이 기사에서는 안정적이고 효율적인 웹 애플리케이션을 구축하는 데 도움이 되도록 FastAPI에서 데이터베이스 연결 및 트랜잭션을 구현하는 방법을 소개합니다.

1부: 데이터베이스 연결
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를 만듭니다.

2부: 트랜잭션 처리
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를 사용하여 데이터베이스 연결을 만든 다음 종속성 기능을 사용하여 데이터베이스 세션을 경로 처리기에 삽입했습니다. 다음으로 데이터베이스 세션 내에서 트랜잭션을 사용하여 데이터베이스 작업의 원자성과 일관성을 보장하는 방법을 시연했습니다. 이러한 기술을 배우고 연습하면 안정적이고 효율적인 웹 애플리케이션을 구축할 수 있습니다.

위 내용은 FastAPI에서 데이터베이스 연결 및 트랜잭션 처리를 구현하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!