ClaimedTaskRunner.php
2 months ago
Cli.php
2 months ago
CronCommand.php
2 months ago
DaemonRunner.php
2 months ago
ExecutionLimitOverride.php
2 months ago
ScheduledTaskResolver.php
2 months ago
ScheduledTasksLister.php
2 months ago
TaskAdder.php
2 months ago
TaskCanceller.php
2 months ago
TaskRunner.php
2 months ago
TaskTrigger.php
2 months ago
WorkerTypesCatalog.php
2 months ago
index.php
2 months ago
TaskCanceller.php
53 lines
| 1 | <?php declare(strict_types = 1); |
| 2 | |
| 3 | namespace MailPoet\Cron\CliCommands; |
| 4 | |
| 5 | if (!defined('ABSPATH')) exit; |
| 6 | |
| 7 | |
| 8 | use InvalidArgumentException; |
| 9 | use MailPoet\Entities\ScheduledTaskEntity; |
| 10 | use MailPoet\Newsletter\Sending\ScheduledTasksRepository; |
| 11 | use MailPoetVendor\Carbon\Carbon; |
| 12 | |
| 13 | class TaskCanceller { |
| 14 | private ScheduledTasksRepository $scheduledTasksRepository; |
| 15 | |
| 16 | private ScheduledTaskResolver $taskResolver; |
| 17 | |
| 18 | public function __construct( |
| 19 | ScheduledTasksRepository $scheduledTasksRepository, |
| 20 | ScheduledTaskResolver $taskResolver |
| 21 | ) { |
| 22 | $this->scheduledTasksRepository = $scheduledTasksRepository; |
| 23 | $this->taskResolver = $taskResolver; |
| 24 | } |
| 25 | |
| 26 | /** |
| 27 | * Cancels a scheduled, paused, or 'cli' task. Running tasks are owned by their executor and completed |
| 28 | * ones are history, so both are rejected. A 'cli' task is cancellable as the recovery path for a |
| 29 | * zombie left behind by a hard-killed CLI run (cancel, then re-add). |
| 30 | * |
| 31 | * @return array{id: int, type: string} |
| 32 | */ |
| 33 | public function cancel(int $taskId): array { |
| 34 | $task = $this->taskResolver->resolveById($taskId); |
| 35 | |
| 36 | $status = $task->getStatus(); |
| 37 | if (!in_array($status, [ScheduledTaskEntity::STATUS_SCHEDULED, ScheduledTaskEntity::STATUS_PAUSED, ScheduledTaskEntity::STATUS_CLI], true)) { |
| 38 | $current = $this->taskResolver->nameStatus($task); |
| 39 | throw new InvalidArgumentException("Task {$taskId} is '{$current}' and cannot be cancelled. Only scheduled, paused, or cli tasks can be cancelled."); |
| 40 | } |
| 41 | |
| 42 | $task->setStatus(ScheduledTaskEntity::STATUS_CANCELLED); |
| 43 | $task->setCancelledAt(Carbon::now()->millisecond(0)); |
| 44 | $this->scheduledTasksRepository->persist($task); |
| 45 | $this->scheduledTasksRepository->flush(); |
| 46 | |
| 47 | return [ |
| 48 | 'id' => (int)$task->getId(), |
| 49 | 'type' => (string)$task->getType(), |
| 50 | ]; |
| 51 | } |
| 52 | } |
| 53 |