JiuContinent
bf309ecdf7
refactor(core,ai): 调整模块导入路径并移除废弃文件 - 修复 scheduler.py 中的导入路径错误，将 reports.service 改为 reports.services - 移除废弃的 app/core/background/task_queue.py 文件 - 移除废弃的 app/modules/ai_agent/adapters.py 文件 - 修复 ai_memory/service.py 中的导入路径错误，将 events.service 改为 events.services ```
90 lines
3.1 KiB
Python
90 lines
3.1 KiB
Python
from typing import Any
|
|
|
|
from fastapi import HTTPException, status
|
|
from sqlalchemy import func, select
|
|
|
|
from app.core.config import get_settings
|
|
from app.core.constants import ActorValue
|
|
from app.core.http.pagination import bounded_limit
|
|
from app.core.utils.time import utc_now
|
|
from app.modules.events.constants import (
|
|
EVENT_CODE_PREFIX,
|
|
EventErrorDetail,
|
|
EventStatus,
|
|
)
|
|
from app.modules.events.models import DomainEvent
|
|
from app.modules.events.services.serialization import _serialize_event
|
|
|
|
|
|
class EventQueryMixin:
|
|
def emit(
|
|
self,
|
|
event_type: str,
|
|
source: str,
|
|
aggregate_type: str,
|
|
aggregate_id: str | int | None,
|
|
actor: str = ActorValue.SYSTEM,
|
|
payload: dict[str, Any] | None = None,
|
|
idempotency_key: str | None = None,
|
|
dispatch: bool = False,
|
|
) -> DomainEvent:
|
|
if idempotency_key:
|
|
existing = self.db.execute(
|
|
select(DomainEvent).where(DomainEvent.idempotency_key == idempotency_key)
|
|
).scalar_one_or_none()
|
|
if existing is not None:
|
|
if dispatch and existing.status == EventStatus.PENDING:
|
|
return self.dispatch_event(existing.event_id)
|
|
return existing
|
|
|
|
settings = get_settings()
|
|
now = utc_now()
|
|
record = DomainEvent(
|
|
event_id=f"{EVENT_CODE_PREFIX}-{now:%Y%m%d%H%M%S%f}",
|
|
event_type=event_type,
|
|
source=source,
|
|
aggregate_type=aggregate_type,
|
|
aggregate_id=str(aggregate_id) if aggregate_id is not None else None,
|
|
actor=actor,
|
|
payload=payload or {},
|
|
idempotency_key=idempotency_key,
|
|
next_attempt_at=now,
|
|
max_attempts=settings.event_dispatch_max_attempts,
|
|
)
|
|
self.db.add(record)
|
|
self.db.commit()
|
|
self.db.refresh(record)
|
|
if dispatch:
|
|
return self.dispatch_event(record.event_id)
|
|
return record
|
|
|
|
def list_events(
|
|
self,
|
|
status_filter: str | None = None,
|
|
event_type: str | None = None,
|
|
limit: int = 100,
|
|
) -> list[dict[str, Any]]:
|
|
stmt = select(DomainEvent).order_by(DomainEvent.id.desc()).limit(bounded_limit(limit))
|
|
if status_filter:
|
|
stmt = stmt.where(DomainEvent.status == status_filter)
|
|
if event_type:
|
|
stmt = stmt.where(DomainEvent.event_type == event_type)
|
|
return [_serialize_event(item) for item in self.db.execute(stmt).scalars()]
|
|
|
|
def count_by_status(self) -> dict[str, int]:
|
|
rows = self.db.execute(
|
|
select(DomainEvent.status, func.count()).group_by(DomainEvent.status)
|
|
).all()
|
|
return {str(status_value): int(count) for status_value, count in rows}
|
|
|
|
def get_event(self, event_id: str) -> DomainEvent:
|
|
record = self.db.execute(
|
|
select(DomainEvent).where(DomainEvent.event_id == event_id)
|
|
).scalar_one_or_none()
|
|
if record is None:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail=EventErrorDetail.EVENT_NOT_FOUND,
|
|
)
|
|
return record
|