mirrors/posthog
0
0
Fork 0
mirror of https://github.com/PostHog/posthog.git synced 2024-11-21 13:39:22 +01:00
Code Issues Releases Activity

fix(insights): prevent race condition (#26265)

Browse Source
This commit is contained in:
Eric Duong 2024-11-20 11:05:02 -05:00 committed by GitHub
parent d836bc860a
commit 386382b51f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
posthog/clickhouse/client/async_task_chain.py
View File

@ -2,6 +2,7 @@ import threading
import typing
from contextlib import contextmanager
from functools import partial
import uuid
from celery import chain
from celery.canvas import Signature
@ -22,12 +23,15 @@ def kick_off_task(
query_status: QueryStatus,
task_signature: Signature,
) -> None:
task = task_signature.apply_async()
task_id = str(uuid.uuid4())
query_status.task_id = task_id
manager.store_query_status(query_status)
task = task_signature.apply_async(task_id=task_id)
# During end-to-end tests, the task is executed synchronously, so we have to refresh the status.
if isinstance(task, EagerResult):
query_status = manager.get_query_status()
query_status.task_id = task.id
manager.store_query_status(query_status)
manager.store_query_status(query_status)
def get_task_chain() -> list[tuple[Signature, "QueryStatusManager", QueryStatus]]: