Function refresh_all_segment_counts

api/segment_membership/tasks.py:172–193

Refresh counts for every project with a live segment

()

    timeout=timedelta(minutes=10),
)
def refresh_all_segment_counts() -> None:
    """Refresh counts for every project with a live segment"""
    if not settings.CLICKHOUSE_ENABLED:
        return

    projects = (
        Project.objects.filter(
            Exists(Segment.live_objects.filter(project=OuterRef("pk")))
        )
        .select_related("organisation")
        .order_by("organisation_id", "id")
    )
    total = projects.count()
    if not total:
        return

    spacing = timedelta(
        hours=settings.SEGMENT_MEMBERSHIP_REFRESH_PROJECT_STAGGER_WINDOW_HOURS
    ) / (total + 1)
    now = timezone.now()
    for index, project in enumerate(projects.iterator()):
        enqueue_membership_refresh(project, delay_until=now + spacing * index)


@register_task_handler(