490 lines
9.9 KiB
PHP
490 lines
9.9 KiB
PHP
|
<?php
|
||
|
/**
|
||
|
* @package Joomla.Platform
|
||
|
* @subpackage Cache
|
||
|
*
|
||
|
* @copyright Copyright (C) 2005 - 2013 Open Source Matters, Inc. All rights reserved.
|
||
|
* @license GNU General Public License version 2 or later; see LICENSE
|
||
|
*/
|
||
|
|
||
|
defined('JPATH_PLATFORM') or die;
|
||
|
|
||
|
/**
|
||
|
* Memcached cache storage handler
|
||
|
*
|
||
|
* @package Joomla.Platform
|
||
|
* @subpackage Cache
|
||
|
* @see http://php.net/manual/en/book.memcached.php
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
class JCacheStorageMemcached extends JCacheStorage
|
||
|
{
|
||
|
/**
|
||
|
* @var Memcached
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
protected static $_db = null;
|
||
|
|
||
|
/**
|
||
|
* @var boolean
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
protected $_persistent = false;
|
||
|
|
||
|
/**
|
||
|
* @var
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
protected $_compress = 0;
|
||
|
|
||
|
/**
|
||
|
* Constructor
|
||
|
*
|
||
|
* @param array $options Optional parameters.
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function __construct($options = array())
|
||
|
{
|
||
|
parent::__construct($options);
|
||
|
if (self::$_db === null)
|
||
|
{
|
||
|
$this->getConnection();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Return memcached connection object
|
||
|
*
|
||
|
* @return object memcached connection object
|
||
|
*
|
||
|
* @since 12.1
|
||
|
* @throws RuntimeException
|
||
|
*/
|
||
|
protected function getConnection()
|
||
|
{
|
||
|
if ((extension_loaded('memcached') && class_exists('Memcached')) != true)
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
$config = JFactory::getConfig();
|
||
|
$this->_persistent = $config->get('memcache_persist', true);
|
||
|
$this->_compress = $config->get('memcache_compress', false) == false ? 0 : Memcached::OPT_COMPRESSION;
|
||
|
|
||
|
/*
|
||
|
* This will be an array of loveliness
|
||
|
* @todo: multiple servers
|
||
|
* $servers = (isset($params['servers'])) ? $params['servers'] : array();
|
||
|
*/
|
||
|
$server = array();
|
||
|
$server['host'] = $config->get('memcache_server_host', 'localhost');
|
||
|
$server['port'] = $config->get('memcache_server_port', 11211);
|
||
|
|
||
|
// Create the memcache connection
|
||
|
if ($this->_persistent)
|
||
|
{
|
||
|
$session = JFactory::getSession();
|
||
|
self::$_db = new Memcached($session->getId());
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
self::$_db = new Memcached;
|
||
|
}
|
||
|
$memcachedtest = self::$_db->addServer($server['host'], $server['port']);
|
||
|
|
||
|
if ($memcachedtest == false)
|
||
|
{
|
||
|
throw new RuntimeException('Could not connect to memcached server', 404);
|
||
|
}
|
||
|
|
||
|
self::$_db->setOption(Memcached::OPT_COMPRESSION, $this->_compress);
|
||
|
|
||
|
// Memcached has no list keys, we do our own accounting, initialise key index
|
||
|
if (self::$_db->get($this->_hash . '-index') === false)
|
||
|
{
|
||
|
$empty = array();
|
||
|
self::$_db->set($this->_hash . '-index', $empty, 0);
|
||
|
}
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get cached data from memcached by id and group
|
||
|
*
|
||
|
* @param string $id The cache data id
|
||
|
* @param string $group The cache data group
|
||
|
* @param boolean $checkTime True to verify cache time expiration threshold
|
||
|
*
|
||
|
* @return mixed Boolean false on failure or a cached data string
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function get($id, $group, $checkTime = true)
|
||
|
{
|
||
|
$cache_id = $this->_getCacheId($id, $group);
|
||
|
$back = self::$_db->get($cache_id);
|
||
|
return $back;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get all cached data
|
||
|
*
|
||
|
* @return array data
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function getAll()
|
||
|
{
|
||
|
parent::getAll();
|
||
|
|
||
|
$keys = self::$_db->get($this->_hash . '-index');
|
||
|
$secret = $this->_hash;
|
||
|
|
||
|
$data = array();
|
||
|
|
||
|
if (!empty($keys) && is_array($keys))
|
||
|
{
|
||
|
foreach ($keys as $key)
|
||
|
{
|
||
|
if (empty($key))
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
$namearr = explode('-', $key->name);
|
||
|
|
||
|
if ($namearr !== false && $namearr[0] == $secret && $namearr[1] == 'cache')
|
||
|
{
|
||
|
|
||
|
$group = $namearr[2];
|
||
|
|
||
|
if (!isset($data[$group]))
|
||
|
{
|
||
|
$item = new JCacheStorageHelper($group);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
$item = $data[$group];
|
||
|
}
|
||
|
|
||
|
$item->updateSize($key->size / 1024);
|
||
|
|
||
|
$data[$group] = $item;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return $data;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Store the data to memcached by id and group
|
||
|
*
|
||
|
* @param string $id The cache data id
|
||
|
* @param string $group The cache data group
|
||
|
* @param string $data The data to store in cache
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function store($id, $group, $data)
|
||
|
{
|
||
|
$cache_id = $this->_getCacheId($id, $group);
|
||
|
|
||
|
if (!$this->lockindex())
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
$index = self::$_db->get($this->_hash . '-index');
|
||
|
if ($index === false)
|
||
|
{
|
||
|
$index = array();
|
||
|
}
|
||
|
|
||
|
$tmparr = new stdClass;
|
||
|
$tmparr->name = $cache_id;
|
||
|
$tmparr->size = strlen($data);
|
||
|
$index[] = $tmparr;
|
||
|
self::$_db->replace($this->_hash . '-index', $index, 0);
|
||
|
$this->unlockindex();
|
||
|
|
||
|
// Prevent double writes, write only if it doesn't exist else replace
|
||
|
if (!self::$_db->replace($cache_id, $data, $this->_lifetime))
|
||
|
{
|
||
|
self::$_db->set($cache_id, $data, $this->_lifetime);
|
||
|
}
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Remove a cached data entry by id and group
|
||
|
*
|
||
|
* @param string $id The cache data id
|
||
|
* @param string $group The cache data group
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function remove($id, $group)
|
||
|
{
|
||
|
$cache_id = $this->_getCacheId($id, $group);
|
||
|
|
||
|
if (!$this->lockindex())
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
$index = self::$_db->get($this->_hash . '-index');
|
||
|
if ($index === false)
|
||
|
{
|
||
|
$index = array();
|
||
|
}
|
||
|
|
||
|
foreach ($index as $key => $value)
|
||
|
{
|
||
|
if ($value->name == $cache_id)
|
||
|
{
|
||
|
unset($index[$key]);
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
self::$_db->replace($this->_hash . '-index', $index, 0);
|
||
|
$this->unlockindex();
|
||
|
|
||
|
return self::$_db->delete($cache_id);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Clean cache for a group given a mode.
|
||
|
*
|
||
|
* @param string $group The cache data group
|
||
|
* @param string $mode The mode for cleaning cache [group|notgroup]
|
||
|
* group mode : cleans all cache in the group
|
||
|
* notgroup mode : cleans all cache not in the group
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function clean($group, $mode = null)
|
||
|
{
|
||
|
if (!$this->lockindex())
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
$index = self::$_db->get($this->_hash . '-index');
|
||
|
if ($index === false)
|
||
|
{
|
||
|
$index = array();
|
||
|
}
|
||
|
|
||
|
$secret = $this->_hash;
|
||
|
foreach ($index as $key => $value)
|
||
|
{
|
||
|
|
||
|
if (strpos($value->name, $secret . '-cache-' . $group . '-') === 0 xor $mode != 'group')
|
||
|
{
|
||
|
self::$_db->delete($value->name, 0);
|
||
|
unset($index[$key]);
|
||
|
}
|
||
|
}
|
||
|
self::$_db->replace($this->_hash . '-index', $index, 0);
|
||
|
$this->unlockindex();
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Test to see if the cache storage is available.
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise.
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public static function isSupported()
|
||
|
{
|
||
|
if ((extension_loaded('memcached') && class_exists('Memcached')) != true)
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
$config = JFactory::getConfig();
|
||
|
$host = $config->get('memcache_server_host', 'localhost');
|
||
|
$port = $config->get('memcache_server_port', 11211);
|
||
|
|
||
|
$memcached = new Memcached;
|
||
|
$memcachedtest = @$memcached->addServer($host, $port);
|
||
|
|
||
|
if (!$memcachedtest)
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Lock cached item - override parent as this is more efficient
|
||
|
*
|
||
|
* @param string $id The cache data id
|
||
|
* @param string $group The cache data group
|
||
|
* @param integer $locktime Cached item max lock time
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise.
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function lock($id, $group, $locktime)
|
||
|
{
|
||
|
$returning = new stdClass;
|
||
|
$returning->locklooped = false;
|
||
|
|
||
|
$looptime = $locktime * 10;
|
||
|
|
||
|
$cache_id = $this->_getCacheId($id, $group);
|
||
|
|
||
|
if (!$this->lockindex())
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
$index = self::$_db->get($this->_hash . '-index');
|
||
|
if ($index === false)
|
||
|
{
|
||
|
$index = array();
|
||
|
}
|
||
|
|
||
|
$tmparr = new stdClass;
|
||
|
$tmparr->name = $cache_id;
|
||
|
$tmparr->size = 1;
|
||
|
|
||
|
$index[] = $tmparr;
|
||
|
self::$_db->replace($this->_hash . '-index', $index, 0);
|
||
|
|
||
|
$this->unlockindex();
|
||
|
|
||
|
$data_lock = self::$_db->add($cache_id . '_lock', 1, $locktime);
|
||
|
|
||
|
if ($data_lock === false)
|
||
|
{
|
||
|
|
||
|
$lock_counter = 0;
|
||
|
|
||
|
// Loop until you find that the lock has been released.
|
||
|
// That implies that data get from other thread has finished
|
||
|
while ($data_lock === false)
|
||
|
{
|
||
|
|
||
|
if ($lock_counter > $looptime)
|
||
|
{
|
||
|
$returning->locked = false;
|
||
|
$returning->locklooped = true;
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
usleep(100);
|
||
|
$data_lock = self::$_db->add($cache_id . '_lock', 1, $locktime);
|
||
|
$lock_counter++;
|
||
|
}
|
||
|
|
||
|
}
|
||
|
$returning->locked = $data_lock;
|
||
|
|
||
|
return $returning;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Unlock cached item - override parent for cacheid compatibility with lock
|
||
|
*
|
||
|
* @param string $id The cache data id
|
||
|
* @param string $group The cache data group
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise.
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
public function unlock($id, $group = null)
|
||
|
{
|
||
|
$cache_id = $this->_getCacheId($id, $group) . '_lock';
|
||
|
|
||
|
if (!$this->lockindex())
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
$index = self::$_db->get($this->_hash . '-index');
|
||
|
if ($index === false)
|
||
|
{
|
||
|
$index = array();
|
||
|
}
|
||
|
|
||
|
foreach ($index as $key => $value)
|
||
|
{
|
||
|
if ($value->name == $cache_id)
|
||
|
{
|
||
|
unset($index[$key]);
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
self::$_db->replace($this->_hash . '-index', $index, 0);
|
||
|
$this->unlockindex();
|
||
|
|
||
|
return self::$_db->delete($cache_id);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Lock cache index
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise.
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
protected function lockindex()
|
||
|
{
|
||
|
$looptime = 300;
|
||
|
$data_lock = self::$_db->add($this->_hash . '-index_lock', 1, 30);
|
||
|
|
||
|
if ($data_lock === false)
|
||
|
{
|
||
|
|
||
|
$lock_counter = 0;
|
||
|
|
||
|
// Loop until you find that the lock has been released. that implies that data get from other thread has finished
|
||
|
while ($data_lock === false)
|
||
|
{
|
||
|
if ($lock_counter > $looptime)
|
||
|
{
|
||
|
return false;
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
usleep(100);
|
||
|
$data_lock = self::$_db->add($this->_hash . '-index_lock', 1, 30);
|
||
|
$lock_counter++;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Unlock cache index
|
||
|
*
|
||
|
* @return boolean True on success, false otherwise.
|
||
|
*
|
||
|
* @since 12.1
|
||
|
*/
|
||
|
protected function unlockindex()
|
||
|
{
|
||
|
return self::$_db->delete($this->_hash . '-index_lock');
|
||
|
}
|
||
|
}
|