mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-22 08:40:03 +01:00
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from datetime import timedelta
|
|
|
|
import structlog
|
|
from celery import shared_task
|
|
from django.utils import timezone
|
|
from prometheus_client import Counter
|
|
|
|
from ee.session_recordings.session_recording_extensions import persist_recording
|
|
from posthog.session_recordings.models.session_recording import SessionRecording
|
|
from posthog.tasks.utils import CeleryQueue
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
REPLAY_NEEDS_PERSISTENCE_COUNTER = Counter(
|
|
"snapshot_persist_persistence_task_queued",
|
|
"Count of session recordings that need to be persisted",
|
|
# we normally avoid team label but not all teams pin recordings so there shouldn't be _too_ many labels here
|
|
labelnames=["team_id"],
|
|
)
|
|
|
|
|
|
@shared_task(
|
|
ignore_result=True,
|
|
queue=CeleryQueue.SESSION_REPLAY_PERSISTENCE.value,
|
|
)
|
|
def persist_single_recording(id: str, team_id: int) -> None:
|
|
persist_recording(id, team_id)
|
|
|
|
|
|
@shared_task(
|
|
ignore_result=True,
|
|
queue=CeleryQueue.SESSION_REPLAY_PERSISTENCE.value,
|
|
)
|
|
def persist_finished_recordings() -> None:
|
|
one_day_old = timezone.now() - timedelta(hours=24)
|
|
finished_recordings = SessionRecording.objects.filter(created_at__lte=one_day_old, object_storage_path=None)
|
|
|
|
logger.info("Persisting finished recordings", count=finished_recordings.count())
|
|
|
|
for recording in finished_recordings:
|
|
REPLAY_NEEDS_PERSISTENCE_COUNTER.labels(team_id=recording.team_id).inc()
|
|
persist_single_recording.delay(recording.session_id, recording.team_id)
|