feat: add method to check if there're pre fetched tasks

oat-sa · Aug 30, 2023 · ddd01f2 · ddd01f2
1 parent d86a5f8
commit ddd01f2
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 1 deletion.
diff --git a/models/classes/taskQueue/Queue.php b/models/classes/taskQueue/Queue.php
@@ -227,6 +227,14 @@ public function getNumberOfTasksToReceive()
         return $this->getBroker()->getNumberOfTasksToReceive();
     }
 
+    /**
+     * @inheritdoc
+     */
+    public function hasPreFetchedMessages(): bool
+    {
+        return $this->getBroker()->hasPreFetchedMessages();
+    }
+
     /**
      * @param TaskInterface $task
      *

diff --git a/models/classes/taskQueue/Queue/Broker/AbstractQueueBroker.php b/models/classes/taskQueue/Queue/Broker/AbstractQueueBroker.php
@@ -92,14 +92,22 @@ public function pop()
         return $this->popPreFetchedMessage();
     }
 
+    /**
+     * @inheritdoc
+     */
+    public function hasPreFetchedMessages(): bool
+    {
+        return $this->preFetchedQueue->count();
+    }
+
     /**
      * Pop a task from the internal queue.
      *
      * @return TaskInterface|null
      */
     private function popPreFetchedMessage()
     {
-        if ($this->preFetchedQueue->count()) {
+        if ($this->hasPreFetchedMessages()) {
             return $this->preFetchedQueue->dequeue();
         }
 

diff --git a/models/classes/taskQueue/Queue/Broker/QueueBrokerInterface.php b/models/classes/taskQueue/Queue/Broker/QueueBrokerInterface.php
@@ -74,4 +74,9 @@ public function delete(TaskInterface $task);
      * @return int
      */
     public function getNumberOfTasksToReceive();
+
+    /**
+     * Check if the internal queue contains any tasks.
+     */
+    public function hasPreFetchedMessages(): bool;
 }
diff --git a/models/classes/taskQueue/QueueInterface.php b/models/classes/taskQueue/QueueInterface.php
@@ -87,6 +87,11 @@ public function isSync();
      */
     public function getNumberOfTasksToReceive();
 
+    /**
+     * Check if the internal queue contains any tasks.
+     */
+    public function hasPreFetchedMessages(): bool;
+
     /**
      * Set new broker.
      *