guke
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""自报计数快照落库。一次事务:插 1 条快照头 + N 条 event 行,返回快照 id。"""
|
|
from __future__ import annotations
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.models.analytics_selfstat import AnalyticsSelfStat, AnalyticsSelfStatEvent
|
|
from app.schemas.analytics_selfstat import SelfStatBatchIn
|
|
|
|
|
|
def record_selfstat(db: Session, batch: SelfStatBatchIn) -> int:
|
|
snap = AnalyticsSelfStat(
|
|
device_id=batch.device_id,
|
|
epoch_id=batch.epoch_id,
|
|
app_ver=batch.app_ver,
|
|
oem=batch.oem,
|
|
os=batch.os,
|
|
batches_attempted=batch.batches_attempted,
|
|
batches_ok=batch.batches_ok,
|
|
batches_fail=batch.batches_fail,
|
|
retries=batch.retries,
|
|
queue_depth=batch.queue_depth,
|
|
sent_at=batch.sent_at,
|
|
)
|
|
db.add(snap)
|
|
db.flush() # 拿到 snap.id
|
|
db.add_all([
|
|
AnalyticsSelfStatEvent(
|
|
snapshot_id=snap.id,
|
|
event=e.event,
|
|
attempted=e.attempted,
|
|
drop_capture=e.drop_capture,
|
|
delivered=e.delivered,
|
|
drop_undelivered=e.drop_undelivered,
|
|
)
|
|
for e in batch.events
|
|
])
|
|
db.commit()
|
|
return snap.id
|