49 lines
1.5 KiB
PHP
49 lines
1.5 KiB
PHP
<?php
|
|
|
|
namespace MailPoet\Cron;
|
|
|
|
use MailPoet\Entities\ScheduledTaskEntity;
|
|
use MailPoet\Models\ScheduledTask;
|
|
use MailPoet\Newsletter\Sending\ScheduledTasksRepository;
|
|
use MailPoet\WP\Functions as WPFunctions;
|
|
use MailPoetVendor\Carbon\Carbon;
|
|
|
|
class CronWorkerScheduler {
|
|
/** @var WPFunctions */
|
|
private $wp;
|
|
|
|
/** @var ScheduledTasksRepository */
|
|
private $scheduledTaskRepository;
|
|
|
|
public function __construct(
|
|
WPFunctions $wp,
|
|
ScheduledTasksRepository $scheduledTaskRepository
|
|
) {
|
|
$this->wp = $wp;
|
|
$this->scheduledTaskRepository = $scheduledTaskRepository;
|
|
}
|
|
|
|
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(ScheduledTask $task, $timeout) {
|
|
$scheduledAt = Carbon::createFromTimestamp($this->wp->currentTime('timestamp'));
|
|
$task->scheduledAt = $scheduledAt->addMinutes($timeout);
|
|
$task->setExpr('updated_at', 'NOW()');
|
|
$task->status = ScheduledTask::STATUS_SCHEDULED;
|
|
$task->save();
|
|
}
|
|
}
|