Browse Source

bug: fix sequence number may be duplicated when multi-threads running the same workflow #21047 (#21153)

tags/1.5.0
NeatGuyCoding 4 months ago
parent
commit
918bb9a2f7
No account linked to committer's email address
1 changed files with 10 additions and 5 deletions
  1. 10
    5
      api/core/repositories/sqlalchemy_workflow_execution_repository.py

+ 10
- 5
api/core/repositories/sqlalchemy_workflow_execution_repository.py View File

@@ -151,12 +151,17 @@ class SQLAlchemyWorkflowExecutionRepository(WorkflowExecutionRepository):
existing = session.scalar(select(WorkflowRun).where(WorkflowRun.id == domain_model.id_))
if not existing:
# For new records, get the next sequence number
stmt = select(func.max(WorkflowRun.sequence_number)).where(
WorkflowRun.app_id == self._app_id,
WorkflowRun.tenant_id == self._tenant_id,
# in case multiple executions are created concurrently, use for update
stmt = (
select(func.coalesce(func.max(WorkflowRun.sequence_number), 0) + 1)
.where(
WorkflowRun.app_id == self._app_id,
WorkflowRun.tenant_id == self._tenant_id,
)
.with_for_update()
)
max_sequence = session.scalar(stmt)
db_model.sequence_number = (max_sequence or 0) + 1
next_seq = session.scalar(stmt)
db_model.sequence_number = int(next_seq) if next_seq is not None else 1
else:
# For updates, keep the existing sequence number
db_model.sequence_number = existing.sequence_number

Loading…
Cancel
Save