change lock collection signature

Author: mmasiukevich

src/InMemoryLockCollection.php

@@ -47,10 +47,12 @@ public function place(string $key, Lock $lock): Promise
     /**
      * @inheritDoc
      */
-    public function remove(string $key): Promise
+    public function extract(string $key): Promise
     {
+        $lock = InMemoryCollectionStorage::instance()->get($key);
+
         InMemoryCollectionStorage::instance()->remove($key);
 
-        return new Success();
+        return new Success($lock);
     }
 }

src/LockCollection.php

@@ -23,5 +23,8 @@ public function has(string $key): Promise;
 
     public function place(string $key, Lock $lock): Promise;
 
-    public function remove(string $key): Promise;
+    /**
+     * Returns \ServiceBus\Mutex\Lock|null
+     */
+    public function extract(string $key): Promise;
 }

src/Storage/InMemoryCollectionStorage.php

@@ -55,6 +55,11 @@ public function add(string $key, Lock $lock): void
         $this->localStorage[$key] = $lock;
     }
 
+    public function get(string $key): ?Lock
+    {
+        return $this->localStorage[$key] ?: null;
+    }
+
     /**
      * Reset instance.
      */