Cavemanon/piratepoet
7
Fork 0
Code Pull Requests Actions Releases 16 Activity

Enable setStatus to handle multiple WorkflowRuns at the same time

Browse Source 
[MAILPOET-4731]
This commit is contained in:
David Remer
2022-10-24 13:02:31 +03:00
parent addf541905
commit e8cfb25651
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mailpoet/lib/Automation/Engine/Control/StepHandler.php
View File

@ -100,7 +100,7 @@ class StepHandler {
$this->handleStep($args);
} catch (Throwable $e) {
$status = $e instanceof InvalidStateException && $e->getErrorCode() === 'mailpoet_automation_workflow_not_active' ? WorkflowRun::STATUS_CANCELLED : WorkflowRun::STATUS_FAILED;
$this->workflowRunStorage->updateStatus((int)$args['workflow_run_id'], $status);
$this->workflowRunStorage->updateStatus($status, (int)$args['workflow_run_id']);
if (!$e instanceof Exception) {
throw new Exception($e->getMessage(), intval($e->getCode()), $e);
}
@ -131,7 +131,7 @@ class StepHandler {
// complete workflow run
if (!$stepId) {
$this->workflowRunStorage->updateStatus($workflowRunId, WorkflowRun::STATUS_COMPLETE);
$this->workflowRunStorage->updateStatus(WorkflowRun::STATUS_COMPLETE, $workflowRunId);
return;
}
@ -183,7 +183,7 @@ class StepHandler {
// no need to schedule a new step if the next step is null, complete the run
if (!$nextStep) {
$this->workflowRunStorage->updateStatus($workflowRunId, WorkflowRun::STATUS_COMPLETE);
$this->workflowRunStorage->updateStatus(WorkflowRun::STATUS_COMPLETE, $workflowRunId);
return;
}

8
mailpoet/lib/Automation/Engine/Storage/WorkflowRunStorage.php
View File

@ -34,9 +34,13 @@ class WorkflowRunStorage {
return $result ? WorkflowRun::fromArray((array)$result) : null;
}
public function updateStatus(int $id, string $status): void {
public function updateStatus(string $status, int ...$ids): void {
if (!$ids) {
return;
}
$table = esc_sql($this->table);
$query = (string)$this->wpdb->prepare("UPDATE $table SET status = %s WHERE id = %d", $status, $id);
$ids = esc_sql(implode(',', $ids));
$query = (string)$this->wpdb->prepare("UPDATE $table SET status = %s WHERE id IN ($ids)", $status);
$result = $this->wpdb->query($query);
if ($result === false) {
throw Exceptions::databaseError($this->wpdb->last_error);