Merge pull request #7255 from annando/issue-7223

[friendica.git] / src / Core / Cache / MemcacheCacheDriver.php

diff --git a/src/Core/Cache/MemcacheCacheDriver.php b/src/Core/Cache/MemcacheCacheDriver.php
index cd12be8ecd9a58d7f40490936b4ccfcdf17cce45..fd928c6fcc15ed2a5245f2bca27c664a6f696531 100644 (file)
--- a/src/Core/Cache/MemcacheCacheDriver.php
+++ b/src/Core/Cache/MemcacheCacheDriver.php
@@ -4,10 +4,13 @@ namespace Friendica\Core\Cache;
 
 use Friendica\Core\Cache;
 
+use Exception;
+use Memcache;
+
 /**
  * Memcache Cache Driver
  *
- * @author Hypolite Petovan <mrpetovan@gmail.com>
+ * @author Hypolite Petovan <hypolite@mrpetovan.com>
  */
 class MemcacheCacheDriver extends AbstractCacheDriver implements IMemoryCacheDriver
 {
@@ -15,21 +18,50 @@ class MemcacheCacheDriver extends AbstractCacheDriver implements IMemoryCacheDri
        use TraitCompareDelete;
 
        /**
-        * @var \Memcache
+        * @var Memcache
         */
        private $memcache;
 
+       /**
+        * @param string $memcache_host
+        * @param int    $memcache_port
+        * @throws Exception
+        */
        public function __construct($memcache_host, $memcache_port)
        {
                if (!class_exists('Memcache', false)) {
-                       throw new \Exception('Memcache class isn\'t available');
+                       throw new Exception('Memcache class isn\'t available');
                }
 
-               $this->memcache = new \Memcache();
+               $this->memcache = new Memcache();
 
                if (!$this->memcache->connect($memcache_host, $memcache_port)) {
-                       throw new \Exception('Expected Memcache server at ' . $memcache_host . ':' . $memcache_port . ' isn\'t available');
+                       throw new Exception('Expected Memcache server at ' . $memcache_host . ':' . $memcache_port . ' isn\'t available');
+               }
+       }
+
+       /**
+        * (@inheritdoc)
+        */
+       public function getAllKeys($prefix = null)
+       {
+               $keys = [];
+               $allSlabs = $this->memcache->getExtendedStats('slabs');
+               foreach ($allSlabs as $slabs) {
+                       foreach (array_keys($slabs) as $slabId) {
+                               $cachedump = $this->memcache->getExtendedStats('cachedump', (int)$slabId);
+                               foreach ($cachedump as $key => $arrVal) {
+                                       if (!is_array($arrVal)) {
+                                               continue;
+                                       }
+                                       $keys = array_merge($keys, array_keys($arrVal));
+                               }
+                       }
                }
+
+               $keys = $this->getOriginalKeys($keys);
+
+               return $this->filterArrayKeysByPrefix($keys, $prefix);
        }
 
        /**
@@ -111,6 +143,6 @@ class MemcacheCacheDriver extends AbstractCacheDriver implements IMemoryCacheDri
        public function add($key, $value, $ttl = Cache::FIVE_MINUTES)
        {
                $cachekey = $this->getCacheKey($key);
-               return $this->memcache->add($cachekey, $value, $ttl);
+               return $this->memcache->add($cachekey, serialize($value), MEMCACHE_COMPRESSED, $ttl);
        }
 }