getChannels()

Author: asika32764

packages/queue/src/Driver/ChannelAwareDriverInterface.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Windwalker\Queue\Driver;
+
+interface ChannelAwareDriverInterface
+{
+    /**
+     * @return  iterable<string>
+     */
+    public function getChannels(): iterable;
+}

packages/queue/src/Driver/DatabaseQueueDriver.php

@@ -20,7 +20,7 @@
  *
  * @since  3.2
  */
-class DatabaseQueueDriver implements QueueDriverInterface
+class DatabaseQueueDriver implements QueueDriverInterface, ChannelAwareDriverInterface
 {
     use UuidDriverTrait;
 
@@ -288,6 +288,16 @@ public function disconnect(): static
         return $this;
     }
 
+    public function getChannels(): iterable
+    {
+        $query = $this->db->createQuery();
+
+        $query->selectRaw('DISTINCT channel')
+            ->from($this->table);
+
+        return $this->db->prepare($query)->loadColumn();
+    }
+
     protected function checkCanSkipLocked(): bool
     {
         if ($this->canSkipLocked !== null) {

packages/queue/src/Driver/PdoQueueDriver.php

@@ -17,7 +17,7 @@
  *
  * @since  3.3
  */
-class PdoQueueDriver implements QueueDriverInterface
+class PdoQueueDriver implements QueueDriverInterface, ChannelAwareDriverInterface
 {
     use UuidDriverTrait;
 
@@ -217,6 +217,16 @@ public function defer(QueueMessage $message): static
         return $this;
     }
 
+    public function getChannels(): iterable
+    {
+        $sql = 'SELECT DISTINCT channel FROM ' . $this->table;
+
+        $stat = $this->pdo->prepare($sql);
+        $stat->execute();
+
+        return $stat->fetchAll(PDO::FETCH_COLUMN);
+    }
+
     /**
      * Method to get property Table
      *

packages/queue/test/QueueDatabaseTest.php

@@ -131,6 +131,28 @@ public function testGetDriver(): void
         self::markTestIncomplete(); // TODO: Complete this test
     }
 
+    /**
+     * @see  ChannelAwareDriverInterface::getChannels
+     */
+    public function testGetChannels(): void
+    {
+        $this->instance->push(new TestJob(['Foo']), 0, 'foo-channel');
+        $this->instance->push(new TestJob(['Bar']), 0, 'bar-channel');
+
+        $channels = [...$this->instance->getDriver()->getChannels()];
+
+        sort($channels);
+
+        self::assertEquals(
+            ['bar-channel', 'default', 'foo-channel'],
+            $channels
+        );
+
+        self::$db->delete('queue_jobs')
+            ->whereIn('channel', ['foo-channel', 'bar-channel'])
+            ->execute();
+    }
+
     /**
      * @see  Queue::__construct
      */