forked from GNUsocial/gnu-social
182 lines
4.3 KiB
PHP
182 lines
4.3 KiB
PHP
<?php
|
|
|
|
namespace Doctrine\Common\Cache;
|
|
|
|
use Redis;
|
|
use function array_combine;
|
|
use function array_diff_key;
|
|
use function array_fill_keys;
|
|
use function array_filter;
|
|
use function array_keys;
|
|
use function count;
|
|
use function defined;
|
|
use function extension_loaded;
|
|
use function is_bool;
|
|
|
|
/**
|
|
* Redis cache provider.
|
|
*
|
|
* @link www.doctrine-project.org
|
|
*/
|
|
class RedisCache extends CacheProvider
|
|
{
|
|
/** @var Redis|null */
|
|
private $redis;
|
|
|
|
/**
|
|
* Sets the redis instance to use.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function setRedis(Redis $redis)
|
|
{
|
|
$redis->setOption(Redis::OPT_SERIALIZER, $this->getSerializerValue());
|
|
$this->redis = $redis;
|
|
}
|
|
|
|
/**
|
|
* Gets the redis instance used by the cache.
|
|
*
|
|
* @return Redis|null
|
|
*/
|
|
public function getRedis()
|
|
{
|
|
return $this->redis;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doFetch($id)
|
|
{
|
|
return $this->redis->get($id);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doFetchMultiple(array $keys)
|
|
{
|
|
$fetchedItems = array_combine($keys, $this->redis->mget($keys));
|
|
|
|
// Redis mget returns false for keys that do not exist. So we need to filter those out unless it's the real data.
|
|
$keysToFilter = array_keys(array_filter($fetchedItems, static function ($item) : bool {
|
|
return $item === false;
|
|
}));
|
|
|
|
if ($keysToFilter) {
|
|
$multi = $this->redis->multi(Redis::PIPELINE);
|
|
foreach ($keysToFilter as $key) {
|
|
$multi->exists($key);
|
|
}
|
|
$existItems = array_filter($multi->exec());
|
|
$missedItemKeys = array_diff_key($keysToFilter, $existItems);
|
|
$fetchedItems = array_diff_key($fetchedItems, array_fill_keys($missedItemKeys, true));
|
|
}
|
|
|
|
return $fetchedItems;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doSaveMultiple(array $keysAndValues, $lifetime = 0)
|
|
{
|
|
if ($lifetime) {
|
|
// Keys have lifetime, use SETEX for each of them
|
|
$multi = $this->redis->multi(Redis::PIPELINE);
|
|
foreach ($keysAndValues as $key => $value) {
|
|
$multi->setex($key, $lifetime, $value);
|
|
}
|
|
$succeeded = array_filter($multi->exec());
|
|
|
|
return count($succeeded) == count($keysAndValues);
|
|
}
|
|
|
|
// No lifetime, use MSET
|
|
return (bool) $this->redis->mset($keysAndValues);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doContains($id)
|
|
{
|
|
$exists = $this->redis->exists($id);
|
|
|
|
if (is_bool($exists)) {
|
|
return $exists;
|
|
}
|
|
|
|
return $exists > 0;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doSave($id, $data, $lifeTime = 0)
|
|
{
|
|
if ($lifeTime > 0) {
|
|
return $this->redis->setex($id, $lifeTime, $data);
|
|
}
|
|
|
|
return $this->redis->set($id, $data);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doDelete($id)
|
|
{
|
|
return $this->redis->del($id) >= 0;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doDeleteMultiple(array $keys)
|
|
{
|
|
return $this->redis->del($keys) >= 0;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doFlush()
|
|
{
|
|
return $this->redis->flushDB();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function doGetStats()
|
|
{
|
|
$info = $this->redis->info();
|
|
|
|
return [
|
|
Cache::STATS_HITS => $info['keyspace_hits'],
|
|
Cache::STATS_MISSES => $info['keyspace_misses'],
|
|
Cache::STATS_UPTIME => $info['uptime_in_seconds'],
|
|
Cache::STATS_MEMORY_USAGE => $info['used_memory'],
|
|
Cache::STATS_MEMORY_AVAILABLE => false,
|
|
];
|
|
}
|
|
|
|
/**
|
|
* Returns the serializer constant to use. If Redis is compiled with
|
|
* igbinary support, that is used. Otherwise the default PHP serializer is
|
|
* used.
|
|
*
|
|
* @return int One of the Redis::SERIALIZER_* constants
|
|
*/
|
|
protected function getSerializerValue()
|
|
{
|
|
if (defined('Redis::SERIALIZER_IGBINARY') && extension_loaded('igbinary')) {
|
|
return Redis::SERIALIZER_IGBINARY;
|
|
}
|
|
|
|
return Redis::SERIALIZER_PHP;
|
|
}
|
|
}
|