FastAPI를 사용하여 음악 스트리밍 서비스 구현하기

25 Sep 2023

FastAPI

FastAPI는 Python 기반의 빠르고 간편한 웹 프레임워크로서, 음악 스트리밍 서비스와 같은 대규모 애플리케이션을 개발하는 데 적합한 선택입니다. 이 글에서는 FastAPI를 사용하여 음악 스트리밍 서비스를 구현하는 방법에 대해 알아보겠습니다.

소개

음악 스트리밍 서비스를 구현하기 위해 FastAPI와 함께 다음과 같은 주요 기술을 사용할 수 있습니다:

Django ORM: 데이터베이스 모델 및 쿼리를 관리하기 위해 Django의 ORM을 사용할 수 있습니다.
JWT 인증: 사용자 인증 및 권한 관리를 위해 JSON Web Token(JWT)을 사용할 수 있습니다.
AWS S3: 음악 파일을 저장하고 제공하기 위해 Amazon S3와 함께 사용할 수 있습니다.

설치 및 설정

FastAPI를 설치하려면 다음 명령어를 실행하세요:

pip install fastapi

Django ORM을 사용하려면 Django 또한 설치해야 합니다:

pip install django

그리고 프로젝트를 생성한 후 Django 설정 파일(settings.py)에 FastAPI를 연결해야 합니다. FastAPI 애플리케이션을 연결하는 방법에 대한 자세한 내용은 FastAPI 공식 문서를 참조하세요.

모델 및 엔드포인트 구현

먼저, Django ORM을 사용하여 음악 트랙을 나타내는 모델을 정의합니다. 예를 들어, Track 모델을 생성할 수 있습니다:

from django.db import models

class Track(models.Model):
    title = models.CharField(max_length=100)
    artist = models.CharField(max_length=100)
    duration = models.IntegerField()
    file = models.FileField(upload_to='tracks/')

다음으로, FastAPI에서 HTTP 엔드포인트를 정의합니다. 이 예제에서는 /tracks 엔드포인트를 사용하여 음악 트랙 목록을 반환하는 API를 구현합니다:

from fastapi import FastAPI
from fastapi.responses import JSONResponse
from .models import Track

app = FastAPI()

@app.get("/tracks")
async def get_tracks():
    tracks = Track.objects.all()
    return JSONResponse(content={"tracks": tracks})

인증 및 권한 관리

음악 스트리밍 서비스에서는 사용자 인증 및 권한 관리가 중요합니다. FastAPI와 JWT를 사용하여 이를 구현할 수 있습니다.

먼저, 사용자를 나타내는 모델을 정의합니다:

from django.contrib.auth.models import AbstractUser

class User(AbstractUser):
    pass

그런 다음, JWT 토큰을 발급하고 검증하는 FastAPI 함수를 작성합니다:

import jwt

def create_token(user_id: int) -> str:
    payload = {
        'user_id': user_id
    }
    token = jwt.encode(payload, 'secret_key', algorithm='HS256')
    return token

def verify_token(token: str) -> dict:
    try:
        payload = jwt.decode(token, 'secret_key', algorithms=['HS256'])
        return payload
    except jwt.DecodeError:
        return None

인증이 필요한 엔드포인트를 보호하기 위해 FastAPI의 데코레이터를 사용할 수 있습니다:

from fastapi import Depends, HTTPException
from fastapi.security import OAuth2PasswordBearer

oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/token")

@app.post("/tracks")
async def create_track(track: Track, token: str = Depends(oauth2_scheme)):
    user_id = verify_token(token)
    if not user_id:
        raise HTTPException(status_code=401, detail="Unauthorized")
    # 트랙 생성 로직 작성

결론

이제 FastAPI를 사용하여 강력하고 확장 가능한 음악 스트리밍 서비스를 구현하는 방법을 알아보았습니다. FastAPI의 강력한 기능과 Django ORM, JWT 인증 및 AWS S3와의 통합을 통해 높은 성능과 안정성을 가진 서비스를 구축할 수 있습니다. FastAPI 공식 문서와 Django 공식 문서를 참조하여 자세한 내용을 학습해 보세요.

#음악스트리밍 #FastAPI