skip locked

Author: asika32764

packages/queue/src/Driver/DatabaseQueueDriver.php

@@ -24,6 +24,8 @@ class DatabaseQueueDriver implements QueueDriverInterface
 {
     use UuidDriverTrait;
 
+    protected ?bool $canSkipLocked = null;
+
     /**
      * DatabaseQueueDriver constructor.
      *
@@ -89,6 +91,8 @@ public function pop(?string $channel = null): ?QueueMessage
 
         $query = $this->db->getQuery(true);
 
+        $do = $this->checkCanSkipLocked() ? 'SKIP LOCKED' : '';
+
         $query->select('*')
             ->from($this->table)
             ->where('channel', $channel)
@@ -100,7 +104,7 @@ function (Query $query) use ($now) {
                         ->where('reserved', '<', $now->modify('-' . $this->timeout . 'seconds'));
                 }
             )
-            ->forUpdate();
+            ->forUpdate($do);
 
         $data = $this->db->transaction(
             function () use ($now, $query) {
@@ -285,4 +289,20 @@ public function disconnect(): static
 
         return $this;
     }
+
+    protected function checkCanSkipLocked(): bool
+    {
+        if ($this->canSkipLocked !== null) {
+            return $this->canSkipLocked;
+        }
+
+        try {
+            $this->db->createQuery()->selectRaw('1')->from($this->table)->forUpdate('SKIP LOCKED')->get();
+            $this->canSkipLocked = true;
+        } catch (Throwable) {
+            $this->canSkipLocked = false;
+        }
+
+        return $this->canSkipLocked;
+    }
 }