Socket.IO 서버 (Python)

import socketio
from fastapi import FastAPI

# REST app
api = FastAPI()

@api.get('/health')
async def health():
    return {'ok': True}

# Socket.IO server
sio = socketio.AsyncServer(async_mode='asgi', cors_allowed_origins='*')

@sio.event
async def connect(sid, environ, auth):
    token = (auth or {}).get('token')
    if not token or not validate_token(token):
        raise socketio.exceptions.ConnectionRefusedError('unauthorized')
    user = decode(token)
    await sio.save_session(sid, {'user': user})
    print('connected', sid, user['name'])

@sio.event
async def disconnect(sid):
    print('disconnected', sid)

@sio.on('chat:message')
async def chat_message(sid, data):
    sess = await sio.get_session(sid)
    user = sess['user']
    await sio.emit('chat:message', {
        'from': user['name'],
        'text': data['text'],
    }, room=data['room'])

# Mount everything together
app = socketio.ASGIApp(sio, other_asgi_app=api)
# Run: uvicorn main:app --host 0.0.0.0 --port 8000