# Entity: Message

```python
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from typing import List
from core.database import get_db
from models.message import Message
from schemas.message import MessageSchema

router = APIRouter()

@router.get("/messages", status_code=200, response_model=List[MessageSchema])
async def get_user_messages(
    user_id: int,
    db: Session = Depends(get_db)
):
    messages = db.query(Message).filter(
        (Message.sender_id == user_id) | (Message.receiver_id == user_id)
    ).all()
    return messages
```