Cavemanon/piratepoet
7
Fork 0
Code Pull Requests Actions Releases 16 Activity

Create StatsNotifications entity

Browse Source 
[MAILPOET-2439]
This commit is contained in:
Pavel Dohnal
2019-10-23 15:53:08 +02:00
committed by Jack Kitterhing
parent 1095cba1f1
commit 92bbb3f4e8
2 changed files with 69 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
lib/Cron/Workers/StatsNotifications/Scheduler.php
View File

@ -3,6 +3,9 @@
namespace MailPoet\Cron\Workers\StatsNotifications;
use Carbon\Carbon;
use Doctrine\ORM\EntityManager;
use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\StatsNotificationEntity;
use MailPoet\Models\Newsletter;
use MailPoet\Models\ScheduledTask;
use MailPoet\Models\StatsNotification;
@ -24,8 +27,12 @@ class Scheduler {
Newsletter::TYPE_STANDARD,
];
function __construct(SettingsController $settings) {
/** @var EntityManager */
private $entity_manager;
function __construct(SettingsController $settings, EntityManager $entity_manager) {
$this->settings = $settings;
$this->entity_manager = $entity_manager;
}
function schedule(Newsletter $newsletter) {
@ -33,16 +40,16 @@ class Scheduler {
return false;
}
$task = ScheduledTask::create();
$task->type = Worker::TASK_TYPE;
$task->status = ScheduledTask::STATUS_SCHEDULED;
$task->scheduled_at = $this->getNextRunDate();
$task->save();
$task = new ScheduledTaskEntity();
$task->setType(Worker::TASK_TYPE);
$task->setStatus(ScheduledTaskEntity::STATUS_SCHEDULED);
$task->setScheduledAt($this->getNextRunDate());
$this->entity_manager->persist($task);
$this->entity_manager->flush();
$stats_notifications = StatsNotification::create();
$stats_notifications->newsletter_id = $newsletter->id;
$stats_notifications->task_id = $task->id;
$stats_notifications->save();
$stats_notifications = new StatsNotificationEntity($newsletter->id, $task->getId());
$this->entity_manager->persist($stats_notifications);
$this->entity_manager->flush();
}
private function shouldSchedule(Newsletter $newsletter) {

52
lib/Entities/StatsNotificationEntity.php Normal file
View File

@ -0,0 +1,52 @@
<?php
namespace MailPoet\Entities;
use MailPoet\Doctrine\EntityTraits\AutoincrementedIdTrait;
use MailPoet\Doctrine\EntityTraits\CreatedAtTrait;
use MailPoet\Doctrine\EntityTraits\UpdatedAtTrait;
use MailPoetVendor\Doctrine\ORM\Mapping as ORM;
/**
* @ORM\Entity()
* @ORM\Table(name="stats_notifications")
*/
class StatsNotificationEntity {
use AutoincrementedIdTrait;
use CreatedAtTrait;
use UpdatedAtTrait;
/**
* @ORM\OneToOne(targetEntity="MailPoet\Entities\NewsletterEntity")
* @ORM\JoinColumn(name="newsletter_id", referencedColumnName="id")
* @var NewsletterEntity
*/
private $newsletter;
/**
* @ORM\OneToOne(targetEntity="MailPoet\Entities\ScheduledTaskEntity")
* @ORM\JoinColumn(name="task_id", referencedColumnName="id")
* @var ScheduledTaskEntity
*/
private $task;
function __construct(NewsletterEntity $newsletter, ScheduledTaskEntity $task) {
$this->newsletter = $newsletter;
$this->task = $task;
}
/**
* @return NewsletterEntity
*/
public function getNewsletter() {
return $this->newsletter;
}
/**
* @return ScheduledTaskEntity
*/
public function getTask() {
return $this->task;
}
}