[PostgreSQL/Python] DB 트랜잭션 비동기 처리하기 | asyncio, asyncpg | 데이터베이스 연결 풀 만들기

asyncio

asyncio는 Python의 비동기 프로그래밍을 지원하는 표준 라이브러리로 이벤트 루프와 코루틴을 기반으로 동작하며, 다음과 같은 주요 역할을 한다.

• 비동기 코드 관리: asyncio는 비동기 작업을 쉽게 관리하고 스케줄링이 가능. 
• 이벤트 루프: asyncio의 핵심 요소는 이벤트 루프로, 비동기 작업의 스케줄링과 실행을 관리하며 이벤트 루프가 반복되면서 비동기 작업을 처리.
• 코루틴 지원: asyncio는 코루틴을 사용하여 비동기 작업을 정의하고 실행 가능. async def로 정의된 코루틴 함수는 await 키워드를 사용하여 다른 비동기 함수를 호출하고 작업을 일시 중단시킬 수 있음.
• 이벤트 처리: asyncio를 사용하면 이벤트 처리를 쉽게 구현할 수 있음. 예를 들어, 네트워크 연결, 소켓 통신, 키보드 및 마우스 입력과 같은 이벤트를 비동기적으로 처리할 수 있음.
• 병렬성과 동시성: asyncio를 통해 여러 비동기 작업을 동시에 처리하거나 병렬로 실행할 수 있음. 이는 I/O 중심의 작업을 효율적으로 처리하고 응답성을 향상시키는 데 유용.
• 비동기 입출력(IO): asyncio는 비동기 입출력을 지원하며 파일 입출력, 네트워크 통신 및 다른 I/O 작업을 비동기적으로 처리할 수 있습니다.

 

asyncpg

asyncpg는 PostgreSQL 데이터베이스와 상호작용하기 위한 비동기 Python 드라이버로, 비동기 프로그래밍을 지원하여 데이터베이스 작업을 효율적으로 처리할 수 있도록 해준다.

• 비동기 지원: asyncpg는 비동기 프로그래밍을 위해 설계되어, 비동기 코드를 사용하여 데이터베이스 쿼리, 업데이트 및 트랜잭션을 비동기적으로 처리할 수 있음.
• PostgreSQL과의 통합: asyncpg는 PostgreSQL 데이터베이스와 원활하게 통합되어 PostgreSQL의 고급 기능을 활용하거나 복잡한 쿼리를 실행할 수 있음.
• 트랜잭션 관리: asyncpg는 트랜잭션을 지원하여 여러 데이터베이스 작업을 원자적으로 실행하고 롤백할 수 있음.
• 커넥션 풀링: asyncpg는 커넥션 풀을 사용하여 데이터베이스 연결 관리를 단순화하며 높은 동시성을 처리하는 데 도움을 줌. (데이터베이스 연결을 효율적으로 관리하고 재사용하는 메커니즘)

 

코드 예제

import asyncio
import asyncpg

async def main():
    # PostgreSQL 연결 설정 및 데이터베이스 연결 풀 생성
    pool = await asyncpg.create_pool(
    		host="...",
            port=...,  
            user="...",
            password="...",
            database="..."
            )
    
    try:
        # 풀에서 연결을 가져옴
        async with pool.acquire() as conn:
            # 트랜잭션 시작
            async with conn.transaction():
                # 여기에 비동기로 실행하려는 데이터베이스 작업을 추가합니다.
                # 예를 들어, 데이터베이스에서 데이터를 쿼리하거나 업데이트할 수 있습니다.
                
                # 쿼리 실행 예시:
                result = await conn.fetch("SELECT * FROM my_table")
                for row in result:
                    print(row)
                
                # 업데이트 예시:
                await conn.execute("UPDATE my_table SET column1 = $1 WHERE column2 = $2", new_value, some_condition)
    
    except asyncpg.exceptions.PostgresError as e:
        # 에러 처리
        print(f"PostgreSQL 에러: {e}")
    
    finally:
        # 풀 닫기
        await pool.close()

if __name__ == "__main__":
    asyncio.run(main())

 

• asyncpg 로 postgresql DB 연결 pool을 만들어 트랜잭션을 비동기로 처리
• API 설계시 API 호출할 때 DB에 트랜잭션을 하는 경우엔 비동기 처리가 필수