Make RetryTillSaveStore implements the SharedLockStoreInterface

jderusse · jderusse · commit a0321e66c918 · 2020-09-16T13:06:13.000+02:00
diff --git a/src/Symfony/Component/Lock/Store/RetryTillSaveStore.php b/src/Symfony/Component/Lock/Store/RetryTillSaveStore.php
@@ -16,16 +16,18 @@
 use Psr\Log\NullLogger;
 use Symfony\Component\Lock\BlockingStoreInterface;
 use Symfony\Component\Lock\Exception\LockConflictedException;
+use Symfony\Component\Lock\Exception\NotSupportedException;
 use Symfony\Component\Lock\Key;
 use Symfony\Component\Lock\PersistingStoreInterface;
+use Symfony\Component\Lock\SharedLockStoreInterface;
 
 /**
  * RetryTillSaveStore is a PersistingStoreInterface implementation which decorate a non blocking PersistingStoreInterface to provide a
  * blocking storage.
  *
  * @author Jérémy Derussé <jeremy@derusse.com>
  */
-class RetryTillSaveStore implements BlockingStoreInterface, LoggerAwareInterface
+class RetryTillSaveStore implements BlockingStoreInterface, SharedLockStoreInterface, LoggerAwareInterface
 {
     use LoggerAwareTrait;
 
@@ -76,6 +78,24 @@ public function waitAndSave(Key $key)
         throw new LockConflictedException();
     }
 
+    public function saveRead(Key $key)
+    {
+        if (!$this->decorated instanceof SharedLockStoreInterface) {
+            throw new NotSupportedException(sprintf('The "%s" store must decorate a "%s" store.', get_debug_type($this), ShareLockStoreInterface::class));
+        }
+
+        $this->decorated->saveRead($key);
+    }
+
+    public function waitAndSaveRead(Key $key)
+    {
+        if (!$this->decorated instanceof SharedLockStoreInterface) {
+            throw new NotSupportedException(sprintf('The "%s" store must decorate a "%s" store.', get_debug_type($this), ShareLockStoreInterface::class));
+        }
+
+        $this->decorated->waitAndSaveRead($key);
+    }
+
     /**
      * {@inheritdoc}
      */
