Refactor pause/resume to doctrine

[MAILPOET-3876]
This commit is contained in:
Pavel Dohnal
2021-11-04 10:43:55 +01:00
committed by Veljko V
parent f5614ed2bd
commit ee00f03ed3
2 changed files with 54 additions and 17 deletions

View File

@@ -7,6 +7,7 @@ use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\WP\Functions as WPFunctions;
use MailPoetVendor\Doctrine\ORM\EntityManager;
/**
@@ -16,12 +17,17 @@ class SendingQueuesRepository extends Repository {
/** @var ScheduledTaskSubscribersRepository */
private $scheduledTaskSubscribersRepository;
/** @var WPFunctions */
private $wp;
public function __construct(
EntityManager $entityManager,
WPFunctions $wp,
ScheduledTaskSubscribersRepository $scheduledTaskSubscribersRepository
) {
parent::__construct($entityManager);
$this->scheduledTaskSubscribersRepository = $scheduledTaskSubscribersRepository;
$this->wp = $wp;
}
protected function getEntityClassName() {
@@ -77,4 +83,31 @@ class SendingQueuesRepository extends Repository {
}
return $qb->getQuery()->getResult();
}
public function pause(SendingQueueEntity $queue): void {
if ($queue->getCountProcessed() !== $queue->getCountTotal()) {
$task = $queue->getTask();
if ($task instanceof ScheduledTaskEntity) {
$task->setStatus(ScheduledTaskEntity::STATUS_PAUSED);
$this->flush();
}
}
}
public function resume(SendingQueueEntity $queue): void {
$task = $queue->getTask();
if (!$task instanceof ScheduledTaskEntity) return;
if ($queue->getCountProcessed() === $queue->getCountTotal()) {
$task->setProcessedAt($this->wp->currentTime('mysql'));
$task->setStatus(ScheduledTaskEntity::STATUS_COMPLETED);
$this->flush();
} else {
$newsletter = $queue->getNewsletter();
if (!$newsletter instanceof NewsletterEntity) return;
$newsletter->setStatus(NewsletterEntity::STATUS_SENDING);
$task->setStatus(null);
$this->flush();
}
}
}