1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279 |
- <?php
- /*
- ==New BSD License==
- Copyright (c) 2013, Colin Mollenhour
- All rights reserved.
- Redistribution and use in source and binary forms, with or without
- modification, are permitted provided that the following conditions are met:
- * Redistributions of source code must retain the above copyright
- notice, this list of conditions and the following disclaimer.
- * Redistributions in binary form must reproduce the above copyright
- notice, this list of conditions and the following disclaimer in the
- documentation and/or other materials provided with the distribution.
- * The name of Colin Mollenhour may not be used to endorse or promote products
- derived from this software without specific prior written permission.
- * The class name must remain as Cm_Cache_Backend_Redis.
- THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
- ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER BE LIABLE FOR ANY
- DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- /**
- * Redis adapter for Zend_Cache
- *
- * @copyright Copyright (c) 2013 Colin Mollenhour (http://colin.mollenhour.com)
- * @license http://framework.zend.com/license/new-bsd New BSD License
- * @author Colin Mollenhour (http://colin.mollenhour.com)
- */
- class Cm_Cache_Backend_Redis extends Zend_Cache_Backend implements Zend_Cache_Backend_ExtendedInterface
- {
- const SET_IDS = 'zc:ids';
- const SET_TAGS = 'zc:tags';
- const PREFIX_KEY = 'zc:k:';
- const PREFIX_TAG_IDS = 'zc:ti:';
- const FIELD_DATA = 'd';
- const FIELD_MTIME = 'm';
- const FIELD_TAGS = 't';
- const FIELD_INF = 'i';
- const MAX_LIFETIME = 2592000; /* Redis backend limit */
- const COMPRESS_PREFIX = ":\x1f\x8b";
- const DEFAULT_CONNECT_TIMEOUT = 2.5;
- const DEFAULT_CONNECT_RETRIES = 1;
- const LUA_SAVE_SH1 = '1617c9fb2bda7d790bb1aaa320c1099d81825e64';
- const LUA_CLEAN_SH1 = '42ab2fe548aee5ff540123687a2c39a38b54e4a2';
- const LUA_GC_SH1 = 'c00416b970f1aa6363b44965d4cf60ee99a6f065';
- /** @var Credis_Client */
- protected $_redis;
- /** @var bool */
- protected $_notMatchingTags = FALSE;
- /** @var int */
- protected $_lifetimelimit = self::MAX_LIFETIME; /* Redis backend limit */
- /** @var int|bool */
- protected $_compressTags = 1;
- /** @var int|bool */
- protected $_compressData = 1;
- /** @var int */
- protected $_compressThreshold = 20480;
- /** @var string */
- protected $_compressionLib;
- /**
- * On large data sets SUNION slows down considerably when used with too many arguments
- * so this is used to chunk the SUNION into a few commands where the number of set ids
- * exceeds this setting.
- *
- * @var int
- */
- protected $_sunionChunkSize = 500;
- /** @var bool */
- protected $_useLua = false;
- /** @var integer */
- protected $_autoExpireLifetime = 0;
- /** @var string */
- protected $_autoExpirePattern = '/REQEST/';
- /** @var boolean */
- protected $_autoExpireRefreshOnLoad = false;
- /**
- * Lua's unpack() has a limit on the size of the table imposed by
- * the number of Lua stack slots that a C function can use.
- * This value is defined by LUAI_MAXCSTACK in luaconf.h and for Redis it is set to 8000.
- *
- * @see https://github.com/antirez/redis/blob/b903145/deps/lua/src/luaconf.h#L439
- * @var int
- */
- protected $_luaMaxCStack = 5000;
- /**
- * If 'retry_reads_on_master' is truthy then reads will be retried against master when slave returns "(nil)" value
- *
- * @var boolean
- */
- protected $_retryReadsOnMaster = false;
- /**
- * @var stdClass
- */
- protected $_clientOptions;
- /**
- * If 'load_from_slaves' is truthy then reads are performed on a randomly selected slave server
- *
- * @var Credis_Client
- */
- protected $_slave;
- protected function getClientOptions($options = array())
- {
- $clientOptions = new stdClass();
- $clientOptions->forceStandalone = isset($options['force_standalone']) && $options['force_standalone'];
- $clientOptions->connectRetries = isset($options['connect_retries']) ? (int) $options['connect_retries'] : self::DEFAULT_CONNECT_RETRIES;
- $clientOptions->readTimeout = isset($options['read_timeout']) ? (float) $options['read_timeout'] : NULL;
- $clientOptions->password = isset($options['password']) ? $options['password'] : NULL;
- $clientOptions->database = isset($options['database']) ? (int) $options['database'] : 0;
- $clientOptions->persistent = isset($options['persistent']) ? $options['persistent'] : '';
- $clientOptions->timeout = isset($options['timeout']) ? $options['timeout'] : self::DEFAULT_CONNECT_TIMEOUT;
- return $clientOptions;
- }
- /**
- * Construct Zend_Cache Redis backend
- * @param array $options
- * @return \Cm_Cache_Backend_Redis
- */
- public function __construct($options = array())
- {
- if ( empty($options['server']) ) {
- Zend_Cache::throwException('Redis \'server\' not specified.');
- }
- $port = isset($options['port']) ? $options['port'] : 6379;
- $slaveSelect = isset($options['slave_select_callable']) && is_callable($options['slave_select_callable']) ? $options['slave_select_callable'] : null;
- $sentinelMaster = empty($options['sentinel_master']) ? NULL : $options['sentinel_master'];
- $this->_clientOptions = $this->getClientOptions($options);
- // If 'sentinel_master' is specified then server is actually sentinel and master address should be fetched from server.
- if ($sentinelMaster) {
- $sentinelClientOptions = isset($options['sentinel']) && is_array($options['sentinel'])
- ? $this->getClientOptions($options['sentinel'] + $options)
- : $this->_clientOptions;
- $servers = preg_split('/\s*,\s*/', trim($options['server']), NULL, PREG_SPLIT_NO_EMPTY);
- $sentinel = NULL;
- $exception = NULL;
- for ($i = 0; $i <= $sentinelClientOptions->connectRetries; $i++) // Try each sentinel in round-robin fashion
- foreach ($servers as $server) {
- try {
- $sentinelClient = new Credis_Client($server, NULL, $sentinelClientOptions->timeout, $sentinelClientOptions->persistent);
- $sentinelClient->forceStandalone();
- $sentinelClient->setMaxConnectRetries(0);
- if ($sentinelClientOptions->readTimeout) {
- $sentinelClient->setReadTimeout($sentinelClientOptions->readTimeout);
- }
- // Sentinel currently doesn't support AUTH
- //if ($password) {
- // $sentinelClient->auth($password) or Zend_Cache::throwException('Unable to authenticate with the redis sentinel.');
- //}
- $sentinel = new Credis_Sentinel($sentinelClient);
- $sentinel
- ->setClientTimeout($this->_clientOptions->timeout)
- ->setClientPersistent($this->_clientOptions->persistent);
- $redisMaster = $sentinel->getMasterClient($sentinelMaster);
- $this->_applyClientOptions($redisMaster);
- // Verify connected server is actually master as per Sentinel client spec
- if ( ! empty($options['sentinel_master_verify'])) {
- $roleData = $redisMaster->role();
- if ( ! $roleData || $roleData[0] != 'master') {
- usleep(100000); // Sleep 100ms and try again
- $redisMaster = $sentinel->getMasterClient($sentinelMaster);
- $this->_applyClientOptions($redisMaster);
- $roleData = $redisMaster->role();
- if ( ! $roleData || $roleData[0] != 'master') {
- Zend_Cache::throwException('Unable to determine master redis server.');
- }
- }
- }
- $this->_redis = $redisMaster;
- break 2;
- } catch (Exception $e) {
- unset($sentinelClient);
- $exception = $e;
- }
- }
- if ( ! $this->_redis) {
- Zend_Cache::throwException('Unable to connect to a redis sentinel: '.$exception->getMessage(), $exception);
- }
- // Optionally use read slaves - will only be used for 'load' operation
- if ( ! empty($options['load_from_slaves'])) {
- $slaves = $sentinel->getSlaveClients($sentinelMaster);
- if ($slaves) {
- if ($options['load_from_slaves'] == 2) {
- array_push($slaves, $this->_redis); // Also send reads to the master
- }
- if ($slaveSelect) {
- $slave = $slaveSelect($slaves, $this->_redis);
- } else {
- $slaveKey = array_rand($slaves, 1);
- $slave = $slaves[$slaveKey]; /* @var $slave Credis_Client */
- }
- if ($slave instanceof Credis_Client && $slave != $this->_redis) {
- try {
- $this->_applyClientOptions($slave, TRUE);
- $this->_slave = $slave;
- } catch (Exception $e) {
- // If there is a problem with first slave then skip 'load_from_slaves' option
- }
- }
- }
- }
- unset($sentinel);
- }
- // Instantiate Credis_Cluster
- else if ( ! empty($options['cluster'])) {
- $this->_setupReadWriteCluster($options);
- }
- // Direct connection to single Redis server
- else {
- $this->_redis = new Credis_Client($options['server'], $port, $this->_clientOptions->timeout, $this->_clientOptions->persistent);
- $this->_applyClientOptions($this->_redis);
- // Support loading from a replication slave
- if (isset($options['load_from_slave'])) {
- if (is_array($options['load_from_slave'])) {
- $server = $options['load_from_slave']['server'];
- $port = $options['load_from_slave']['port'];
- $clientOptions = $this->getClientOptions($options['load_from_slave'] + $options);
- } else {
- $server = $options['load_from_slave'];
- $port = 6379;
- $clientOptions = $this->_clientOptions;
- }
- if (is_string($server)) {
- try {
- $slave = new Credis_Client($server, $port, $clientOptions->timeout, $clientOptions->persistent);
- $this->_applyClientOptions($slave, TRUE, $clientOptions);
- $this->_slave = $slave;
- } catch (Exception $e) {
- // Slave will not be used
- }
- }
- }
- }
- if ( isset($options['notMatchingTags']) ) {
- $this->_notMatchingTags = (bool) $options['notMatchingTags'];
- }
- if ( isset($options['compress_tags'])) {
- $this->_compressTags = (int) $options['compress_tags'];
- }
- if ( isset($options['compress_data'])) {
- $this->_compressData = (int) $options['compress_data'];
- }
- if ( isset($options['lifetimelimit'])) {
- $this->_lifetimelimit = (int) min($options['lifetimelimit'], self::MAX_LIFETIME);
- }
- if ( isset($options['compress_threshold'])) {
- $this->_compressThreshold = (int) $options['compress_threshold'];
- if ($this->_compressThreshold < 1) {
- $this->_compressThreshold = 1;
- }
- }
- if ( isset($options['automatic_cleaning_factor']) ) {
- $this->_options['automatic_cleaning_factor'] = (int) $options['automatic_cleaning_factor'];
- } else {
- $this->_options['automatic_cleaning_factor'] = 0;
- }
- if ( isset($options['compression_lib']) ) {
- $this->_compressionLib = (string) $options['compression_lib'];
- }
- else if ( function_exists('snappy_compress') ) {
- $this->_compressionLib = 'snappy';
- }
- else if ( function_exists('lz4_compress')) {
- $version = phpversion("lz4");
- if (version_compare($version, "0.3.0") < 0)
- {
- $this->_compressTags = $this->_compressTags > 1 ? true : false;
- $this->_compressData = $this->_compressData > 1 ? true : false;
- }
- $this->_compressionLib = 'l4z';
- }
- else if ( function_exists('zstd_compress')) {
- $version = phpversion("zstd");
- if (version_compare($version, "0.4.13") < 0)
- {
- $this->_compressTags = $this->_compressTags > 1 ? true : false;
- $this->_compressData = $this->_compressData > 1 ? true : false;
- }
- $this->_compressionLib = 'zstd';
- }
- else if ( function_exists('lzf_compress') ) {
- $this->_compressionLib = 'lzf';
- }
- else {
- $this->_compressionLib = 'gzip';
- }
- $this->_compressPrefix = substr($this->_compressionLib,0,2).self::COMPRESS_PREFIX;
- if ( isset($options['sunion_chunk_size']) && $options['sunion_chunk_size'] > 0) {
- $this->_sunionChunkSize = (int) $options['sunion_chunk_size'];
- }
- if (isset($options['use_lua'])) {
- $this->_useLua = (bool) $options['use_lua'];
- }
- if (isset($options['lua_max_c_stack'])) {
- $this->_luaMaxCStack = (int) $options['lua_max_c_stack'];
- }
- if (isset($options['retry_reads_on_master'])) {
- $this->_retryReadsOnMaster = (bool) $options['retry_reads_on_master'];
- }
- if (isset($options['auto_expire_lifetime'])) {
- $this->_autoExpireLifetime = (int) $options['auto_expire_lifetime'];
- }
- if (isset($options['auto_expire_pattern'])) {
- $this->_autoExpirePattern = (string) $options['auto_expire_pattern'];
- }
- if (isset($options['auto_expire_refresh_on_load'])) {
- $this->_autoExpireRefreshOnLoad = (bool) $options['auto_expire_refresh_on_load'];
- }
- }
- /**
- * Apply common configuration to client instances.
- *
- * @param Credis_Client $client
- */
- protected function _applyClientOptions(Credis_Client $client, $forceSelect = FALSE, $clientOptions = null)
- {
- if ($clientOptions === null) {
- $clientOptions = $this->_clientOptions;
- }
- if ($clientOptions->forceStandalone) {
- $client->forceStandalone();
- }
- $client->setMaxConnectRetries($clientOptions->connectRetries);
- if ($clientOptions->readTimeout) {
- $client->setReadTimeout($clientOptions->readTimeout);
- }
- if ($clientOptions->password) {
- $client->auth($clientOptions->password) or Zend_Cache::throwException('Unable to authenticate with the redis server.');
- }
- // Always select database when persistent is used in case connection is re-used by other clients
- if ($forceSelect || $clientOptions->database || $client->getPersistence()) {
- $client->select($clientOptions->database) or Zend_Cache::throwException('The redis database could not be selected.');
- }
- }
- protected function _setupReadWriteCluster($options) {
- $clusterNodes = array();
- if (array_key_exists('master', $options['cluster']) && !empty($options['cluster']['master'])) {
- foreach ($options['cluster']['master'] as $masterNode) {
- if (empty($masterNode['server']) || empty($masterNode['port'])) {
- continue;
- }
- $clusterNodes[] = array(
- 'host' => $masterNode['server'],
- 'port' => $masterNode['port'],
- 'alias' => 'master',
- 'master' => true,
- 'write_only' => true,
- 'timeout' => $this->_clientOptions->timeout,
- 'persistent' => $this->_clientOptions->persistent,
- 'db' => (int) $options['database'],
- );
- break; // limit to 1
- }
- }
- if (!empty($clusterNodes) && array_key_exists('slave', $options['cluster']) && !empty($options['cluster']['slave'])) {
- foreach ($options['cluster']['slave'] as $slaveNodes) {
- if (empty($masterNode['server']) || empty($masterNode['port'])) {
- continue;
- }
- $clusterNodes[] = array(
- 'host' => $slaveNodes['server'],
- 'port' => $slaveNodes['port'],
- 'alias' => 'slave' . count($clusterNodes),
- 'timeout' => $this->_clientOptions->timeout,
- 'persistent' => $this->_clientOptions->persistent,
- 'db' => (int) $options['database'],
- 'password' => $options['password'],
- );
- }
- }
- if (!empty($clusterNodes)) {
- $this->_redis = new Credis_Cluster($clusterNodes);
- }
- }
- /**
- * Load value with given id from cache
- *
- * @param string $id Cache id
- * @param boolean $doNotTestCacheValidity If set to true, the cache validity won't be tested
- * @return bool|string
- */
- public function load($id, $doNotTestCacheValidity = false)
- {
- if ($this->_slave) {
- $data = $this->_slave->hGet(self::PREFIX_KEY.$id, self::FIELD_DATA);
- // Prevent compounded effect of cache flood on asynchronously replicating master/slave setup
- if ($this->_retryReadsOnMaster && $data === false) {
- $data = $this->_redis->hGet(self::PREFIX_KEY.$id, self::FIELD_DATA);
- }
- } else {
- $data = $this->_redis->hGet(self::PREFIX_KEY.$id, self::FIELD_DATA);
- }
- if ($data === NULL) {
- return FALSE;
- }
- $decoded = $this->_decodeData($data);
- if ($this->_autoExpireLifetime === 0 || !$this->_autoExpireRefreshOnLoad) {
- return $decoded;
- }
- $matches = $this->_matchesAutoExpiringPattern($id);
- if (!$matches) {
- return $decoded;
- }
- $this->_redis->expire(self::PREFIX_KEY.$id, min($this->_autoExpireLifetime, self::MAX_LIFETIME));
- return $decoded;
- }
- /**
- * Test if a cache is available or not (for the given id)
- *
- * @param string $id Cache id
- * @return bool|int False if record is not available or "last modified" timestamp of the available cache record
- */
- public function test($id)
- {
- // Don't use slave for this since `test` is usually used for locking
- $mtime = $this->_redis->hGet(self::PREFIX_KEY.$id, self::FIELD_MTIME);
- return ($mtime ? $mtime : FALSE);
- }
- /**
- * Save some string datas into a cache record
- *
- * Note : $data is always "string" (serialization is done by the
- * core not by the backend)
- *
- * @param string $data Datas to cache
- * @param string $id Cache id
- * @param array $tags Array of strings, the cache record will be tagged by each string entry
- * @param bool|int $specificLifetime If != false, set a specific lifetime for this cache record (null => infinite lifetime)
- * @throws CredisException
- * @return boolean True if no problem
- */
- public function save($data, $id, $tags = array(), $specificLifetime = false)
- {
- if(!is_array($tags))
- $tags = $tags ? array($tags) : array();
- else
- $tags = array_flip(array_flip($tags));
- $lifetime = $this->_getAutoExpiringLifetime($this->getLifetime($specificLifetime), $id);
- if ($this->_useLua) {
- $sArgs = array(
- self::PREFIX_KEY,
- self::FIELD_DATA,
- self::FIELD_TAGS,
- self::FIELD_MTIME,
- self::FIELD_INF,
- self::SET_TAGS,
- self::PREFIX_TAG_IDS,
- self::SET_IDS,
- $id,
- $this->_encodeData($data, $this->_compressData),
- $this->_encodeData(implode(',',$tags), $this->_compressTags),
- time(),
- $lifetime ? 0 : 1,
- min($lifetime, self::MAX_LIFETIME),
- $this->_notMatchingTags ? 1 : 0
- );
- $res = $this->_redis->evalSha(self::LUA_SAVE_SH1, $tags, $sArgs);
- if (is_null($res)) {
- $script =
- "local oldTags = redis.call('HGET', ARGV[1]..ARGV[9], ARGV[3]) ".
- "redis.call('HMSET', ARGV[1]..ARGV[9], ARGV[2], ARGV[10], ARGV[3], ARGV[11], ARGV[4], ARGV[12], ARGV[5], ARGV[13]) ".
- "if (ARGV[13] == '0') then ".
- "redis.call('EXPIRE', ARGV[1]..ARGV[9], ARGV[14]) ".
- "end ".
- "if next(KEYS) ~= nil then ".
- "redis.call('SADD', ARGV[6], unpack(KEYS)) ".
- "for _, tagname in ipairs(KEYS) do ".
- "redis.call('SADD', ARGV[7]..tagname, ARGV[9]) ".
- "end ".
- "end ".
- "if (ARGV[15] == '1') then ".
- "redis.call('SADD', ARGV[8], ARGV[9]) ".
- "end ".
- "if (oldTags ~= false) then ".
- "return oldTags ".
- "else ".
- "return '' ".
- "end";
- $res = $this->_redis->eval($script, $tags, $sArgs);
- }
- // Process removed tags if cache entry already existed
- if ($res) {
- $oldTags = explode(',', $this->_decodeData($res));
- if ($remTags = ($oldTags ? array_diff($oldTags, $tags) : FALSE))
- {
- // Update the id list for each tag
- foreach($remTags as $tag)
- {
- $this->_redis->sRem(self::PREFIX_TAG_IDS . $tag, $id);
- }
- }
- }
- return TRUE;
- }
- // Get list of tags previously assigned
- $oldTags = $this->_decodeData($this->_redis->hGet(self::PREFIX_KEY.$id, self::FIELD_TAGS));
- $oldTags = $oldTags ? explode(',', $oldTags) : array();
- $this->_redis->pipeline()->multi();
- // Set the data
- $result = $this->_redis->hMSet(self::PREFIX_KEY.$id, array(
- self::FIELD_DATA => $this->_encodeData($data, $this->_compressData),
- self::FIELD_TAGS => $this->_encodeData(implode(',',$tags), $this->_compressTags),
- self::FIELD_MTIME => time(),
- self::FIELD_INF => $lifetime ? 0 : 1,
- ));
- if( ! $result) {
- throw new CredisException("Could not set cache key $id");
- }
- // Set expiration if specified
- if ($lifetime) {
- $this->_redis->expire(self::PREFIX_KEY.$id, min($lifetime, self::MAX_LIFETIME));
- }
- // Process added tags
- if ($tags)
- {
- // Update the list with all the tags
- $this->_redis->sAdd( self::SET_TAGS, $tags);
- // Update the id list for each tag
- foreach($tags as $tag)
- {
- $this->_redis->sAdd(self::PREFIX_TAG_IDS . $tag, $id);
- }
- }
- // Process removed tags
- if ($remTags = ($oldTags ? array_diff($oldTags, $tags) : FALSE))
- {
- // Update the id list for each tag
- foreach($remTags as $tag)
- {
- $this->_redis->sRem(self::PREFIX_TAG_IDS . $tag, $id);
- }
- }
- // Update the list with all the ids
- if($this->_notMatchingTags) {
- $this->_redis->sAdd(self::SET_IDS, $id);
- }
- $this->_redis->exec();
- return TRUE;
- }
- /**
- * Remove a cache record
- *
- * @param string $id Cache id
- * @return boolean True if no problem
- */
- public function remove($id)
- {
- // Get list of tags for this id
- $tags = explode(',', $this->_decodeData($this->_redis->hGet(self::PREFIX_KEY.$id, self::FIELD_TAGS)));
- $this->_redis->pipeline()->multi();
- // Remove data
- $this->_redis->del(self::PREFIX_KEY.$id);
- // Remove id from list of all ids
- if($this->_notMatchingTags) {
- $this->_redis->sRem( self::SET_IDS, $id );
- }
- // Update the id list for each tag
- foreach($tags as $tag) {
- $this->_redis->sRem(self::PREFIX_TAG_IDS . $tag, $id);
- }
- $result = $this->_redis->exec();
- return (bool) $result[0];
- }
- /**
- * @param array $tags
- */
- protected function _removeByNotMatchingTags($tags)
- {
- $ids = $this->getIdsNotMatchingTags($tags);
- if($ids)
- {
- $this->_redis->pipeline()->multi();
- // Remove data
- $this->_redis->del( $this->_preprocessIds($ids));
- // Remove ids from list of all ids
- if($this->_notMatchingTags) {
- $this->_redis->sRem( self::SET_IDS, $ids);
- }
- $this->_redis->exec();
- }
- }
- /**
- * @param array $tags
- */
- protected function _removeByMatchingTags($tags)
- {
- $ids = $this->getIdsMatchingTags($tags);
- if($ids)
- {
- $this->_redis->pipeline()->multi();
- // Remove data
- $this->_redis->del( $this->_preprocessIds($ids));
- // Remove ids from list of all ids
- if($this->_notMatchingTags) {
- $this->_redis->sRem( self::SET_IDS, $ids);
- }
- $this->_redis->exec();
- }
- }
- /**
- * @param array $tags
- */
- protected function _removeByMatchingAnyTags($tags)
- {
- if ($this->_useLua) {
- $tags = array_chunk($tags, $this->_sunionChunkSize);
- foreach ($tags as $chunk) {
- $chunk = $this->_preprocessTagIds($chunk);
- $args = array(self::PREFIX_KEY, self::SET_TAGS, self::SET_IDS, ($this->_notMatchingTags ? 1 : 0), (int) $this->_luaMaxCStack);
- if ( ! $this->_redis->evalSha(self::LUA_CLEAN_SH1, $chunk, $args)) {
- $script =
- "for i = 1, #KEYS, ARGV[5] do ".
- "local keysToDel = redis.call('SUNION', unpack(KEYS, i, math.min(#KEYS, i + ARGV[5] - 1))) ".
- "for _, keyname in ipairs(keysToDel) do ".
- "redis.call('DEL', ARGV[1]..keyname) ".
- "if (ARGV[4] == '1') then ".
- "redis.call('SREM', ARGV[3], keyname) ".
- "end ".
- "end ".
- "redis.call('DEL', unpack(KEYS, i, math.min(#KEYS, i + ARGV[5] - 1))) ".
- "redis.call('SREM', ARGV[2], unpack(KEYS, i, math.min(#KEYS, i + ARGV[5] - 1))) ".
- "end ".
- "return true";
- $this->_redis->eval($script, $chunk, $args);
- }
- }
- return;
- }
- $ids = $this->getIdsMatchingAnyTags($tags);
- $this->_redis->pipeline()->multi();
- if($ids)
- {
- // Remove data
- $this->_redis->del( $this->_preprocessIds($ids));
- // Remove ids from list of all ids
- if($this->_notMatchingTags) {
- $this->_redis->sRem( self::SET_IDS, $ids);
- }
- }
- // Remove tag id lists
- $this->_redis->del( $this->_preprocessTagIds($tags));
- // Remove tags from list of tags
- $this->_redis->sRem( self::SET_TAGS, $tags);
- $this->_redis->exec();
- }
- /**
- * Clean up tag id lists since as keys expire the ids remain in the tag id lists
- */
- protected function _collectGarbage()
- {
- // Clean up expired keys from tag id set and global id set
- if ($this->_useLua) {
- $sArgs = array(self::PREFIX_KEY, self::SET_TAGS, self::SET_IDS, self::PREFIX_TAG_IDS, ($this->_notMatchingTags ? 1 : 0));
- $allTags = (array) $this->_redis->sMembers(self::SET_TAGS);
- $tagsCount = count($allTags);
- $counter = 0;
- $tagsBatch = array();
- foreach ($allTags as $tag) {
- $tagsBatch[] = $tag;
- $counter++;
- if (count($tagsBatch) == 10 || $counter == $tagsCount ) {
- if ( ! $this->_redis->evalSha(self::LUA_GC_SH1, $tagsBatch, $sArgs)) {
- $script =
- "local tagKeys = {} ".
- "local expired = {} ".
- "local expiredCount = 0 ".
- "local notExpiredCount = 0 ".
- "for _, tagName in ipairs(KEYS) do ".
- "tagKeys = redis.call('SMEMBERS', ARGV[4]..tagName) ".
- "for __, keyName in ipairs(tagKeys) do ".
- "if (redis.call('EXISTS', ARGV[1]..keyName) == 0) then ".
- "expiredCount = expiredCount + 1 ".
- "expired[expiredCount] = keyName ".
- /* Redis Lua scripts have a hard limit of 8000 parameters per command */
- "if (expiredCount == 7990) then ".
- "redis.call('SREM', ARGV[4]..tagName, unpack(expired)) ".
- "if (ARGV[5] == '1') then ".
- "redis.call('SREM', ARGV[3], unpack(expired)) ".
- "end ".
- "expiredCount = 0 ".
- "expired = {} ".
- "end ".
- "else ".
- "notExpiredCount = notExpiredCount + 1 ".
- "end ".
- "end ".
- "if (expiredCount > 0) then ".
- "redis.call('SREM', ARGV[4]..tagName, unpack(expired)) ".
- "if (ARGV[5] == '1') then ".
- "redis.call('SREM', ARGV[3], unpack(expired)) ".
- "end ".
- "end ".
- "if (notExpiredCount == 0) then ".
- "redis.call ('DEL', ARGV[4]..tagName) ".
- "redis.call ('SREM', ARGV[2], tagName) ".
- "end ".
- "expired = {} ".
- "expiredCount = 0 ".
- "notExpiredCount = 0 ".
- "end ".
- "return true";
- $this->_redis->eval($script, $tagsBatch, $sArgs);
- }
- $tagsBatch = array();
- /* Give Redis some time to handle other requests */
- usleep(20000);
- }
- }
- return;
- }
- $exists = array();
- $tags = (array) $this->_redis->sMembers(self::SET_TAGS);
- foreach($tags as $tag)
- {
- // Get list of expired ids for each tag
- $tagMembers = $this->_redis->sMembers(self::PREFIX_TAG_IDS . $tag);
- $numTagMembers = count($tagMembers);
- $expired = array();
- $numExpired = $numNotExpired = 0;
- if($numTagMembers) {
- while ($id = array_pop($tagMembers)) {
- if( ! isset($exists[$id])) {
- $exists[$id] = $this->_redis->exists(self::PREFIX_KEY.$id);
- }
- if ($exists[$id]) {
- $numNotExpired++;
- }
- else {
- $numExpired++;
- $expired[] = $id;
- // Remove incrementally to reduce memory usage
- if (count($expired) % 100 == 0 && $numNotExpired > 0) {
- $this->_redis->sRem( self::PREFIX_TAG_IDS . $tag, $expired);
- if($this->_notMatchingTags) { // Clean up expired ids from ids set
- $this->_redis->sRem( self::SET_IDS, $expired);
- }
- $expired = array();
- }
- }
- }
- if( ! count($expired)) continue;
- }
- // Remove empty tags or completely expired tags
- if ($numExpired == $numTagMembers) {
- $this->_redis->del(self::PREFIX_TAG_IDS . $tag);
- $this->_redis->sRem(self::SET_TAGS, $tag);
- }
- // Clean up expired ids from tag ids set
- else if (count($expired)) {
- $this->_redis->sRem( self::PREFIX_TAG_IDS . $tag, $expired);
- if($this->_notMatchingTags) { // Clean up expired ids from ids set
- $this->_redis->sRem( self::SET_IDS, $expired);
- }
- }
- unset($expired);
- }
- // Clean up global list of ids for ids with no tag
- if($this->_notMatchingTags) {
- // TODO
- }
- }
- /**
- * Clean some cache records
- *
- * Available modes are :
- * 'all' (default) => remove all cache entries ($tags is not used)
- * 'old' => runs _collectGarbage()
- * 'matchingTag' => supported
- * 'notMatchingTag' => supported
- * 'matchingAnyTag' => supported
- *
- * @param string $mode Clean mode
- * @param array $tags Array of tags
- * @throws Zend_Cache_Exception
- * @return boolean True if no problem
- */
- public function clean($mode = Zend_Cache::CLEANING_MODE_ALL, $tags = array())
- {
- if( $tags && ! is_array($tags)) {
- $tags = array($tags);
- }
- try {
- if ($mode == Zend_Cache::CLEANING_MODE_ALL) {
- return $this->_redis->flushDb();
- }
- if ($mode == Zend_Cache::CLEANING_MODE_OLD) {
- $this->_collectGarbage();
- return TRUE;
- }
- if ( ! count($tags)) {
- return TRUE;
- }
- switch ($mode)
- {
- case Zend_Cache::CLEANING_MODE_MATCHING_TAG:
- $this->_removeByMatchingTags($tags);
- break;
- case Zend_Cache::CLEANING_MODE_NOT_MATCHING_TAG:
- $this->_removeByNotMatchingTags($tags);
- break;
- case Zend_Cache::CLEANING_MODE_MATCHING_ANY_TAG:
- $this->_removeByMatchingAnyTags($tags);
- break;
- default:
- Zend_Cache::throwException('Invalid mode for clean() method: '.$mode);
- }
- } catch (CredisException $e) {
- Zend_Cache::throwException('Error cleaning cache by mode '.$mode.': '.$e->getMessage(), $e);
- }
- return TRUE;
- }
- /**
- * Return true if the automatic cleaning is available for the backend
- *
- * @return boolean
- */
- public function isAutomaticCleaningAvailable()
- {
- return TRUE;
- }
- /**
- * Set the frontend directives
- *
- * @param array $directives Assoc of directives
- * @throws Zend_Cache_Exception
- * @return void
- */
- public function setDirectives($directives)
- {
- parent::setDirectives($directives);
- $lifetime = $this->getLifetime(false);
- if ($lifetime > self::MAX_LIFETIME) {
- Zend_Cache::throwException('Redis backend has a limit of 30 days (2592000 seconds) for the lifetime');
- }
- }
- /**
- * Get the auto expiring lifetime.
- *
- * Mainly a workaround for the issues that arise due to the fact that
- * Magento's Enterprise_PageCache module doesn't set any expiry.
- *
- * @param int $specificLifetime
- * @param string $id
- * @return int Cache life time
- */
- protected function _getAutoExpiringLifetime($lifetime, $id)
- {
- if ($lifetime || !$this->_autoExpireLifetime) {
- // If it's already truthy, or there's no auto expire go with it.
- return $lifetime;
- }
- $matches = $this->_matchesAutoExpiringPattern($id);
- if (!$matches) {
- // Only apply auto expire for keys that match the pattern
- return $lifetime;
- }
- if ($this->_autoExpireLifetime > 0) {
- // Return the auto expire lifetime if set
- return $this->_autoExpireLifetime;
- }
- // Return whatever it was set to.
- return $lifetime;
- }
- protected function _matchesAutoExpiringPattern($id)
- {
- $matches = array();
- preg_match($this->_autoExpirePattern, $id, $matches);
- return !empty($matches);
- }
- /**
- * Return an array of stored cache ids
- *
- * @return array array of stored cache ids (string)
- */
- public function getIds()
- {
- if($this->_notMatchingTags) {
- return (array) $this->_redis->sMembers(self::SET_IDS);
- } else {
- $keys = $this->_redis->keys(self::PREFIX_KEY . '*');
- $prefixLen = strlen(self::PREFIX_KEY);
- foreach($keys as $index => $key) {
- $keys[$index] = substr($key, $prefixLen);
- }
- return $keys;
- }
- }
- /**
- * Return an array of stored tags
- *
- * @return array array of stored tags (string)
- */
- public function getTags()
- {
- return (array) $this->_redis->sMembers(self::SET_TAGS);
- }
- /**
- * Return an array of stored cache ids which match given tags
- *
- * In case of multiple tags, a logical AND is made between tags
- *
- * @param array $tags array of tags
- * @return array array of matching cache ids (string)
- */
- public function getIdsMatchingTags($tags = array())
- {
- if ($tags) {
- return (array) $this->_redis->sInter( $this->_preprocessTagIds($tags) );
- }
- return array();
- }
- /**
- * Return an array of stored cache ids which don't match given tags
- *
- * In case of multiple tags, a negated logical AND is made between tags
- *
- * @param array $tags array of tags
- * @return array array of not matching cache ids (string)
- */
- public function getIdsNotMatchingTags($tags = array())
- {
- if( ! $this->_notMatchingTags) {
- Zend_Cache::throwException("notMatchingTags is currently disabled.");
- }
- if ($tags) {
- return (array) $this->_redis->sDiff( self::SET_IDS, $this->_preprocessTagIds($tags) );
- }
- return (array) $this->_redis->sMembers( self::SET_IDS );
- }
- /**
- * Return an array of stored cache ids which match any given tags
- *
- * In case of multiple tags, a logical OR is made between tags
- *
- * @param array $tags array of tags
- * @return array array of any matching cache ids (string)
- */
- public function getIdsMatchingAnyTags($tags = array())
- {
- $result = array();
- if ($tags) {
- $chunks = array_chunk($tags, $this->_sunionChunkSize);
- foreach ($chunks as $chunk) {
- $result = array_merge($result, (array) $this->_redis->sUnion( $this->_preprocessTagIds($chunk)));
- }
- if (count($chunks) > 1) {
- $result = array_unique($result); // since we are chunking requests, we must de-duplicate member names
- }
- }
- return $result;
- }
- /**
- * Return the filling percentage of the backend storage
- *
- * @throws Zend_Cache_Exception
- * @return int integer between 0 and 100
- */
- public function getFillingPercentage()
- {
- $maxMem = $this->_redis->config('GET','maxmemory');
- if (0 == (int) $maxMem['maxmemory']) {
- return 1;
- }
- $info = $this->_redis->info();
- return (int) round(
- ($info['used_memory']/$maxMem['maxmemory']*100)
- ,0
- ,PHP_ROUND_HALF_UP
- );
- }
- /**
- * Return an array of metadatas for the given cache id
- *
- * The array must include these keys :
- * - expire : the expire timestamp
- * - tags : a string array of tags
- * - mtime : timestamp of last modification time
- *
- * @param string $id cache id
- * @return array array of metadatas (false if the cache id is not found)
- */
- public function getMetadatas($id)
- {
- list($tags, $mtime, $inf) = array_values(
- $this->_redis->hMGet(self::PREFIX_KEY.$id, array(self::FIELD_TAGS, self::FIELD_MTIME, self::FIELD_INF))
- );
- if( ! $mtime) {
- return FALSE;
- }
- $tags = explode(',', $this->_decodeData($tags));
- $expire = $inf === '1' ? FALSE : time() + $this->_redis->ttl(self::PREFIX_KEY.$id);
- return array(
- 'expire' => $expire,
- 'tags' => $tags,
- 'mtime' => $mtime,
- );
- }
- /**
- * Give (if possible) an extra lifetime to the given cache id
- *
- * @param string $id cache id
- * @param int $extraLifetime
- * @return boolean true if ok
- */
- public function touch($id, $extraLifetime)
- {
- list($inf) = $this->_redis->hGet(self::PREFIX_KEY.$id, self::FIELD_INF);
- if ($inf === '0') {
- $expireAt = time() + $this->_redis->ttl(self::PREFIX_KEY.$id) + $extraLifetime;
- return (bool) $this->_redis->expireAt(self::PREFIX_KEY.$id, $expireAt);
- }
- return false;
- }
- /**
- * Return an associative array of capabilities (booleans) of the backend
- *
- * The array must include these keys :
- * - automatic_cleaning (is automating cleaning necessary)
- * - tags (are tags supported)
- * - expired_read (is it possible to read expired cache records
- * (for doNotTestCacheValidity option for example))
- * - priority does the backend deal with priority when saving
- * - infinite_lifetime (is infinite lifetime can work with this backend)
- * - get_list (is it possible to get the list of cache ids and the complete list of tags)
- *
- * @return array associative of with capabilities
- */
- public function getCapabilities()
- {
- return array(
- 'automatic_cleaning' => ($this->_options['automatic_cleaning_factor'] > 0),
- 'tags' => true,
- 'expired_read' => false,
- 'priority' => false,
- 'infinite_lifetime' => true,
- 'get_list' => true,
- );
- }
- /**
- * @param string $data
- * @param int $level
- * @throws CredisException
- * @return string
- */
- protected function _encodeData($data, $level)
- {
- if ($this->_compressionLib && $level !== 0 && strlen($data) >= $this->_compressThreshold) {
- switch($this->_compressionLib) {
- case 'snappy': $data = snappy_compress($data); break;
- case 'lzf': $data = lzf_compress($data); break;
- case 'l4z': $data = lz4_compress($data, $level); break;
- case 'zstd': $data = zstd_compress($data, $level); break;
- case 'gzip': $data = gzcompress($data, $level); break;
- default: throw new CredisException("Unrecognized 'compression_lib'.");
- }
- if( ! $data) {
- throw new CredisException("Could not compress cache data.");
- }
- return $this->_compressPrefix.$data;
- }
- return $data;
- }
- /**
- * @param bool|string $data
- * @return string
- */
- protected function _decodeData($data)
- {
- if (substr($data,2,3) == self::COMPRESS_PREFIX) {
- switch(substr($data,0,2)) {
- case 'sn': return snappy_uncompress(substr($data,5));
- case 'lz': return lzf_decompress(substr($data,5));
- case 'l4': return lz4_uncompress(substr($data,5));
- case 'zs': return zstd_uncompress(substr($data,5));
- case 'gz': case 'zc': return gzuncompress(substr($data,5));
- }
- }
- return $data;
- }
- /**
- * @param $item
- * @param $index
- * @param $prefix
- */
- protected function _preprocess(&$item, $index, $prefix)
- {
- $item = $prefix . $item;
- }
- /**
- * @param $ids
- * @return array
- */
- protected function _preprocessIds($ids)
- {
- array_walk($ids, array($this, '_preprocess'), self::PREFIX_KEY);
- return $ids;
- }
- /**
- * @param $tags
- * @return array
- */
- protected function _preprocessTagIds($tags)
- {
- array_walk($tags, array($this, '_preprocess'), self::PREFIX_TAG_IDS);
- return $tags;
- }
- /**
- * Required to pass unit tests
- *
- * @param string $id
- * @return void
- */
- public function ___expire($id)
- {
- $this->_redis->del(self::PREFIX_KEY.$id);
- }
- /**
- * Only for unit tests
- */
- public function ___scriptFlush()
- {
- $this->_redis->script('flush');
- }
- }
|