[파이썬] ORM을 사용한 데이터베이스 비동기 처리

01 Sep 2023

python

소개

ORM(Object-Relational Mapping)은 프로그래밍 언어의 객체와 데이터베이스의 관계형 데이터를 자동으로 매핑해주는 도구입니다. 이를 통해 데이터베이스 작업을 쉽게 수행할 수 있으며, 비동기 처리를 지원하면서 성능을 높일 수 있습니다.

이 튜토리얼에서는 Python에서 사용되는 몇 가지 인기있는 ORM 라이브러리를 소개하고, 이를 통해 데이터베이스 작업을 비동기적으로 처리하는 방법을 알아보겠습니다.

1. SQLAlchemy

SQLAlchemy는 Python에서 가장 많이 사용되는 ORM 라이브러리 중 하나입니다. 비동기 처리를 지원하기 위해 SQLAlchemy.Asyncio 라이브러리를 사용할 수 있습니다.

설치

pip install sqlalchemy
pip install sqlalchemy-asyncio

예제 코드

import asyncio
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy.sql import select
from sqlalchemy_aio import ASYNCIO_STRATEGY

# 데이터베이스 연결 설정
DATABASE_URL = "postgresql://username:password@localhost/mydatabase"
engine = create_engine(DATABASE_URL, strategy=ASYNCIO_STRATEGY, echo=True)
Base = declarative_base(bind=engine)
Session = sessionmaker(bind=engine, class_=AsyncioSession)

# 모델 정의
class User(Base):
    __tablename__ = "users"
    id = Column(Integer, primary_key=True)
    name = Column(String)

# 비동기 세션 생성
async def create_user(name):
    async with Session() as session:
        user = User(name=name)
        session.add(user)
        await session.commit()

# 비동기로 데이터베이스 작업 실행
async def main():
    await create_user("John Doe")

# 이벤트 루프 실행
loop = asyncio.get_event_loop()
loop.run_until_complete(main())

2. Tortoise ORM

Tortoise ORM은 비동기 스타일의 asyncio를 기반으로 한 ORM 라이브러리입니다. Tortoise ORM은 SQLite, PostgreSQL, MySQL 등 다양한 백엔드 데이터베이스를 지원합니다.

설치

pip install tortoise-orm

예제 코드

from tortoise import Tortoise, fields, run_async
from tortoise.models import Model

# 데이터베이스 연결 설정
DATABASE_URL = "sqlite://db.sqlite3"
Tortoise.init(db_url=DATABASE_URL, modules={"models": ["__main__"]})

# 모델 정의
class User(Model):
    id = fields.IntField(pk=True)
    name = fields.CharField(max_length=255)

    class Meta:
        table = "users"

# 비동기로 데이터베이스 작업 실행
async def create_user(name):
    user = User(name=name)
    await user.save()

async def main():
    await create_user("John Doe")

# 이벤트 루프 실행
run_async(main())

결론

ORM을 사용하여 데이터베이스 작업을 비동기적으로 처리하는 것은 데이터베이스 성능을 향상시키고 개발자가 간편하게 데이터베이스 작업을 수행할 수 있는 방법입니다. SQLAlchemy와 Tortoise ORM은 두 가지 인기있는 비동기 ORM 라이브러리로, Python 프로젝트에서 ORM을 사용하여 데이터베이스 작업을 처리할 때 고려해볼만한 선택지입니다.