Refactor pause/resume to doctrine
[MAILPOET-3876]
This commit is contained in:
@@ -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();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user