Migrate methods findDueByType() and findRunningByType() to Doctrine

[MAILPOET-2996]

lib/Newsletter/Sending/ScheduledTasksRepository.php

@@ -7,6 +7,8 @@ use MailPoet\Entities\NewsletterEntity;
 use MailPoet\Entities\ScheduledTaskEntity;
 use MailPoet\Entities\ScheduledTaskSubscriberEntity;
 use MailPoet\Entities\SendingQueueEntity;
+use MailPoet\WP\Functions as WPFunctions;
+use MailPoetVendor\Carbon\Carbon;
 use MailPoetVendor\Doctrine\ORM\Query\Expr\Join;
 
 /**
@@ -87,6 +89,45 @@ class ScheduledTasksRepository extends Repository {
       ->getOneOrNullResult();
   }
 
+  public function findDueByType($type, $limit = null) {
+    return $this->findByTypeAndStatus($type, ScheduledTaskEntity::STATUS_SCHEDULED, $limit);
+  }
+
+  public function findRunningByType($type, $limit = null) {
+    return $this->findByTypeAndStatus($type, null, $limit);
+  }
+
+  protected function findByTypeAndStatus($type, $status, $limit = null, $future = false) {
+    $queryBuilder = $this->doctrineRepository->createQueryBuilder('st')
+      ->select('st')
+      ->where('st.type = :type')
+      ->setParameter('type', $type)
+      ->andWhere('st.deletedAt IS NULL');
+
+    if (is_null($status)) {
+      $queryBuilder->andWhere('st.status IS NULL');
+    } else {
+      $queryBuilder
+        ->andWhere('st.status = :status')
+        ->setParameter('status', $status);
+    }
+
+    if ($future) {
+      $queryBuilder->andWhere('st.scheduledAt >= :now');
+    } else {
+      $queryBuilder->andWhere('st.scheduledAt <= :now');
+    }
+
+    $now = Carbon::createFromTimestamp(WPFunctions::get()->currentTime('timestamp'));
+    $queryBuilder->setParameter('now', $now);
+
+    if ($limit) {
+      $queryBuilder->setMaxResults($limit);
+    }
+
+    return $queryBuilder->getQuery()->getResult();
+  }
+
   protected function getEntityClassName() {
     return ScheduledTaskEntity::class;
   }