init

wp-content/plugins/mailpoet/lib/Cron/CronWorkerScheduler.php

@@ -0,0 +1,74 @@
+<?php // phpcs:ignore SlevomatCodingStandard.TypeHints.DeclareStrictTypes.DeclareStrictTypesMissing
+
+namespace MailPoet\Cron;
+
+if (!defined('ABSPATH')) exit;
+
+
+use MailPoet\Entities\ScheduledTaskEntity;
+use MailPoet\Newsletter\Sending\ScheduledTasksRepository;
+use MailPoetVendor\Carbon\Carbon;
+
+class CronWorkerScheduler {
+  /** @var ScheduledTasksRepository */
+  private $scheduledTaskRepository;
+
+  public function __construct(
+    ScheduledTasksRepository $scheduledTaskRepository
+  ) {
+    $this->scheduledTaskRepository = $scheduledTaskRepository;
+  }
+
+  public function scheduleImmediatelyIfNotRunning($taskType, $priority = ScheduledTaskEntity::PRIORITY_LOW): ScheduledTaskEntity {
+    $task = $this->scheduledTaskRepository->findScheduledOrRunningTask($taskType);
+    // Do nothing when task is running
+    if (($task instanceof ScheduledTaskEntity) && $task->getStatus() === null) {
+      return $task;
+    }
+    $now = Carbon::now()->millisecond(0);
+    // Reschedule existing scheduled task
+    if ($task instanceof ScheduledTaskEntity) {
+      $task->setScheduledAt($now);
+      $task->setPriority($priority);
+      $this->scheduledTaskRepository->flush();
+    }
+    // Schedule new task
+    return $this->schedule($taskType, $now, $priority);
+  }
+
+  public function schedule($taskType, $nextRunDate, $priority = ScheduledTaskEntity::PRIORITY_LOW): ScheduledTaskEntity {
+    $alreadyScheduled = $this->scheduledTaskRepository->findScheduledTask($taskType);
+    if ($alreadyScheduled) {
+      return $alreadyScheduled;
+    }
+    $task = new ScheduledTaskEntity();
+    $task->setType($taskType);
+    $task->setStatus(ScheduledTaskEntity::STATUS_SCHEDULED);
+    $task->setPriority($priority);
+    $task->setScheduledAt($nextRunDate);
+    $this->scheduledTaskRepository->persist($task);
+    $this->scheduledTaskRepository->flush();
+    return $task;
+  }
+
+  public function reschedule(ScheduledTaskEntity $task, $timeout) {
+    $scheduledAt = Carbon::now()->millisecond(0);
+    $task->setScheduledAt($scheduledAt->addMinutes($timeout));
+    $task->setStatus(ScheduledTaskEntity::STATUS_SCHEDULED);
+    $this->scheduledTaskRepository->persist($task);
+    $this->scheduledTaskRepository->flush();
+  }
+
+  public function rescheduleProgressively(ScheduledTaskEntity $task): int {
+    $scheduledAt = Carbon::now()->millisecond(0);
+    $rescheduleCount = $task->getRescheduleCount();
+    $timeout = (int)min(ScheduledTaskEntity::BASIC_RESCHEDULE_TIMEOUT * pow(2, $rescheduleCount), ScheduledTaskEntity::MAX_RESCHEDULE_TIMEOUT);
+    $task->setScheduledAt($scheduledAt->addMinutes($timeout));
+    $task->setRescheduleCount($rescheduleCount + 1);
+    $task->setStatus(ScheduledTaskEntity::STATUS_SCHEDULED);
+    $this->scheduledTaskRepository->persist($task);
+    $this->scheduledTaskRepository->flush();
+
+    return $timeout;
+  }
+}