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

Add a possibility to schedule next step by index

Browse Source 
[MAILPOET-5586]
This commit is contained in:
Jan Jakes
2023-09-14 14:55:00 +02:00
committed by Aschepikov
parent 8720d8881f
commit 72ce5fb569
3 changed files with 38 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
mailpoet/lib/Automation/Engine/Control/StepScheduler.php
View File

@ -4,6 +4,7 @@ namespace MailPoet\Automation\Engine\Control;
use MailPoet\Automation\Engine\Data\AutomationRun;
use MailPoet\Automation\Engine\Data\StepRunArgs;
use MailPoet\Automation\Engine\Exceptions;
use MailPoet\Automation\Engine\Hooks;
use MailPoet\Automation\Engine\Storage\AutomationRunStorage;
@ -29,16 +30,30 @@ class StepScheduler {
}
public function scheduleNextStep(StepRunArgs $args, int $timestamp = null): int {
$runId = $args->getAutomationRun()->getId();
$nextSteps = $args->getStep()->getNextSteps();
// complete the automation run if there are no more steps
if (count($args->getStep()->getNextSteps()) === 0) {
$this->automationRunStorage->updateNextStep($runId, null);
$this->automationRunStorage->updateStatus($runId, AutomationRun::STATUS_COMPLETE);
if (count($nextSteps) === 0) {
$this->completeAutomationRun($args);
return 0;
}
return $this->scheduleNextStepByIndex($args, 0, $timestamp);
}
public function scheduleNextStepByIndex(StepRunArgs $args, int $nextStepIndex, int $timestamp = null): int {
$step = $args->getStep();
$nextStep = $step->getNextSteps()[$nextStepIndex] ?? null;
if (!$nextStep) {
throw Exceptions::nextStepNotFound($step->getId(), $nextStepIndex);
}
$runId = $args->getAutomationRun()->getId();
$nextStepId = $nextStep->getId();
if (!$nextStepId) {
$this->completeAutomationRun($args);
return 0;
}
$nextStepId = $args->getStep()->getNextSteps()[0]->getId();
$data = $this->getActionData($runId, $nextStepId);
$id = $this->scheduleStepAction($data, $timestamp);
$this->automationRunStorage->updateNextStep($runId, $nextStepId);
@ -80,6 +95,12 @@ class StepScheduler {
: $this->actionScheduler->schedule($timestamp, Hooks::AUTOMATION_STEP, $data);
}
private function completeAutomationRun(StepRunArgs $args): void {
$runId = $args->getAutomationRun()->getId();
$this->automationRunStorage->updateNextStep($runId, null);
$this->automationRunStorage->updateStatus($runId, AutomationRun::STATUS_COMPLETE);
}
private function getActionData(int $runId, string $stepId, int $runNumber = 1): array {
return [
[