添加日志
This commit is contained in:
@@ -1,8 +1,9 @@
|
||||
import json
|
||||
import logging
|
||||
from collections.abc import Mapping
|
||||
from typing import Any
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query, Response, status
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query, Request, Response, status
|
||||
from sqlalchemy import text
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
@@ -29,6 +30,7 @@ except ModuleNotFoundError:
|
||||
|
||||
|
||||
router = APIRouter(prefix="/messages", tags=["messages"])
|
||||
logger = logging.getLogger("app.messages")
|
||||
|
||||
|
||||
def _build_preview(content_type: int, content_text: str | None) -> str:
|
||||
@@ -141,6 +143,7 @@ def _assert_conversation_access(
|
||||
@router.post("", response_model=MessageCreateResponse, status_code=status.HTTP_201_CREATED)
|
||||
def create_message(
|
||||
payload: MessageCreateRequest,
|
||||
request: Request,
|
||||
response: Response,
|
||||
current_user_id: int = Depends(get_current_user_id),
|
||||
db: Session = Depends(get_db),
|
||||
@@ -223,6 +226,19 @@ def create_message(
|
||||
|
||||
existing = _get_existing_message(db, conversation_id, payload.client_msg_id)
|
||||
if existing:
|
||||
logger.info(
|
||||
"message idempotent hit",
|
||||
extra={
|
||||
"event": "message_create",
|
||||
"request_id": getattr(request.state, "request_id", None),
|
||||
"user_id": sender_user_id,
|
||||
"conversation_id": conversation_id,
|
||||
"message_id": int(existing["id"]),
|
||||
"seq": int(existing["seq"]),
|
||||
"client_msg_id": payload.client_msg_id,
|
||||
"idempotent": True,
|
||||
},
|
||||
)
|
||||
response.status_code = status.HTTP_200_OK
|
||||
return MessageCreateResponse(
|
||||
idempotent=True,
|
||||
@@ -328,6 +344,19 @@ def create_message(
|
||||
if not created:
|
||||
raise HTTPException(status_code=500, detail="failed to load created message")
|
||||
|
||||
logger.info(
|
||||
"message created",
|
||||
extra={
|
||||
"event": "message_create",
|
||||
"request_id": getattr(request.state, "request_id", None),
|
||||
"user_id": sender_user_id,
|
||||
"conversation_id": conversation_id,
|
||||
"message_id": int(created["id"]),
|
||||
"seq": int(created["seq"]),
|
||||
"client_msg_id": payload.client_msg_id,
|
||||
"idempotent": False,
|
||||
},
|
||||
)
|
||||
return MessageCreateResponse(
|
||||
idempotent=False,
|
||||
message=_row_to_message_item(created),
|
||||
@@ -336,6 +365,7 @@ def create_message(
|
||||
|
||||
@router.get("", response_model=MessageListResponse)
|
||||
def list_messages(
|
||||
request: Request,
|
||||
conversation_id: int = Query(gt=0),
|
||||
cursor_seq: int | None = Query(default=None, ge=1),
|
||||
limit: int = Query(default=20, ge=1, le=100),
|
||||
@@ -382,6 +412,19 @@ def list_messages(
|
||||
items = [_row_to_message_item(row) for row in rows]
|
||||
|
||||
next_cursor_seq = items[0].seq if has_more and items else None
|
||||
logger.info(
|
||||
"messages listed",
|
||||
extra={
|
||||
"event": "message_list",
|
||||
"request_id": getattr(request.state, "request_id", None),
|
||||
"user_id": current_user_id,
|
||||
"conversation_id": conversation_id,
|
||||
"cursor_seq": cursor_seq,
|
||||
"limit": limit,
|
||||
"count": len(items),
|
||||
"has_more": has_more,
|
||||
},
|
||||
)
|
||||
return MessageListResponse(
|
||||
conversation_id=conversation_id,
|
||||
has_more=has_more,
|
||||
|
||||
Reference in New Issue
Block a user