mirror of
https://github.com/langgenius/dify.git
synced 2026-02-09 15:10:13 -05:00
22 lines
843 B
Python
22 lines
843 B
Python
"""
|
|
Celery tasks for asynchronous workflow execution storage operations.
|
|
|
|
These tasks provide asynchronous storage capabilities for workflow execution data,
|
|
improving performance by offloading storage operations to background workers.
|
|
"""
|
|
|
|
from celery import shared_task # type: ignore[import-untyped]
|
|
|
|
from core.db.session_factory import session_factory
|
|
from services.workflow_draft_variable_service import DraftVarFileDeletion, WorkflowDraftVariableService
|
|
|
|
|
|
@shared_task(queue="workflow_draft_var", bind=True, max_retries=3, default_retry_delay=60)
|
|
def save_workflow_execution_task(
|
|
self,
|
|
deletions: list[DraftVarFileDeletion],
|
|
):
|
|
with session_factory.create_session() as session, session.begin():
|
|
srv = WorkflowDraftVariableService(session=session)
|
|
srv.delete_workflow_draft_variable_file(deletions=deletions)
|