파이썬 데이터베이스 프로그래밍 실전 가이드 백오프 재시도 데드락 감지 멱등 키 설계

from tenacity import retry, stop_after_attempt, wait_exponential, wait_random

@retry(stop=stop_after_attempt(5), wait=wait_exponential(multiplier=1, min=1, max=16) + wait_random(0, 2))
def query_database():
    # 데이터베이스 쿼리 실행
    print("DB 쿼리 실행 중...")
    raise Exception("일시적 오류 발생")

query_database()

import psycopg2
from psycopg2 import OperationalError, errors
import time

def run_transaction(conn):
    for attempt in range(5):
        try:
            with conn:
                with conn.cursor() as cur:
                    cur.execute("UPDATE accounts SET balance = balance - 100 WHERE id=1")
                    cur.execute("UPDATE accounts SET balance = balance + 100 WHERE id=2")
            break
        except errors.DeadlockDetected as e:
            print(f"데드락 감지됨, {attempt+1}번째 재시도")
            time.sleep(2 ** attempt)  # 지수 백오프와 함께 재시도

import uuid
import sqlite3

def process_payment(user_id, amount, idempotency_key):
    conn = sqlite3.connect("payments.db")
    cur = conn.cursor()
    
    # 멱등 키 확인
    cur.execute("SELECT status FROM requests WHERE key=?", (idempotency_key,))
    row = cur.fetchone()
    if row:
        print("이미 처리된 요청입니다:", row[0])
        return
    
    # 새로운 요청 처리
    cur.execute("INSERT INTO requests(key, status) VALUES (?, ?)", (idempotency_key, "processing"))
    conn.commit()
    
    # 결제 로직 실행
    print(f"{user_id}님 결제 {amount}원 진행 중...")
    
    # 완료 후 상태 업데이트
    cur.execute("UPDATE requests SET status=? WHERE key=?", ("completed", idempotency_key))
    conn.commit()
    print("결제 완료")

# 실행 예시
key = str(uuid.uuid4())
process_payment("user123", 10000, key)
process_payment("user123", 10000, key)  # 두 번째 호출은 무시됨

import uuid, time
import psycopg2
from psycopg2 import errors
from tenacity import retry, stop_after_attempt, wait_exponential, wait_random

# 백오프 + 재시도 적용
@retry(stop=stop_after_attempt(5), wait=wait_exponential(multiplier=1, min=1, max=16) + wait_random(0, 2))
def run_payment(user_id, amount, idempotency_key):
    conn = psycopg2.connect("dbname=test user=postgres password=secret")
    try:
        with conn:
            with conn.cursor() as cur:
                # 멱등 키 확인
                cur.execute("SELECT status FROM requests WHERE key=%s", (idempotency_key,))
                row = cur.fetchone()
                if row:
                    print("이미 처리된 요청:", row[0])
                    return

                # 멱등 키 등록
                cur.execute("INSERT INTO requests(key, status) VALUES (%s, %s)", (idempotency_key, "processing"))

                # 결제 로직 (데드락 발생 가능성 존재)
                cur.execute("UPDATE accounts SET balance = balance - %s WHERE id=1", (amount,))
                cur.execute("UPDATE accounts SET balance = balance + %s WHERE id=2", (amount,))

                # 상태 완료 업데이트
                cur.execute("UPDATE requests SET status=%s WHERE key=%s", ("completed", idempotency_key))
                print("결제 성공")

    except errors.DeadlockDetected:
        print("데드락 발생 → 재시도")
        raise
    finally:
        conn.close()

# 실행 예시
key = str(uuid.uuid4())
run_payment("user123", 5000, key)