/vendor/symfony/symfony/src/Symfony/Component/HttpKernel/Profiler/RedisProfilerStorage.php
PHP | 373 lines | 208 code | 71 blank | 94 comment | 30 complexity | c2bde7341721b4ebb335efdb6875a535 MD5 | raw file
- <?php
- /*
- * This file is part of the Symfony package.
- *
- * (c) Fabien Potencier <fabien@symfony.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace Symfony\Component\HttpKernel\Profiler;
- use Redis;
- /**
- * RedisProfilerStorage stores profiling information in Redis.
- *
- * @author Andrej Hudec <pulzarraider@gmail.com>
- */
- class RedisProfilerStorage implements ProfilerStorageInterface
- {
- const TOKEN_PREFIX = 'sf_profiler_';
- const REDIS_OPT_SERIALIZER = 1;
- const REDIS_OPT_PREFIX = 2;
- const REDIS_SERIALIZER_NONE = 0;
- const REDIS_SERIALIZER_PHP = 1;
- protected $dsn;
- protected $lifetime;
- /**
- * @var Redis
- */
- private $redis;
- /**
- * Constructor.
- *
- * @param string $dsn A data source name
- * @param string $username Not used
- * @param string $password Not used
- * @param int $lifetime The lifetime to use for the purge
- */
- public function __construct($dsn, $username = '', $password = '', $lifetime = 86400)
- {
- $this->dsn = $dsn;
- $this->lifetime = (int) $lifetime;
- }
- /**
- * {@inheritdoc}
- */
- public function find($ip, $url, $limit, $method)
- {
- $indexName = $this->getIndexName();
- if (!$indexContent = $this->getValue($indexName, self::REDIS_SERIALIZER_NONE)) {
- return array();
- }
- $profileList = array_reverse(explode("\n", $indexContent));
- $result = array();
- foreach ($profileList as $item) {
- if ($limit === 0) {
- break;
- }
- if ($item == '') {
- continue;
- }
- list($itemToken, $itemIp, $itemMethod, $itemUrl, $itemTime, $itemParent) = explode("\t", $item, 6);
- if ($ip && false === strpos($itemIp, $ip) || $url && false === strpos($itemUrl, $url) || $method && false === strpos($itemMethod, $method)) {
- continue;
- }
- $result[] = array(
- 'token' => $itemToken,
- 'ip' => $itemIp,
- 'method' => $itemMethod,
- 'url' => $itemUrl,
- 'time' => $itemTime,
- 'parent' => $itemParent,
- );
- --$limit;
- }
- return $result;
- }
- /**
- * {@inheritdoc}
- */
- public function purge()
- {
- // delete only items from index
- $indexName = $this->getIndexName();
- $indexContent = $this->getValue($indexName, self::REDIS_SERIALIZER_NONE);
- if (!$indexContent) {
- return false;
- }
- $profileList = explode("\n", $indexContent);
- $result = array();
- foreach ($profileList as $item) {
- if ($item == '') {
- continue;
- }
- if (false !== $pos = strpos($item, "\t")) {
- $result[] = $this->getItemName(substr($item, 0, $pos));
- }
- }
- $result[] = $indexName;
- return $this->delete($result);
- }
- /**
- * {@inheritdoc}
- */
- public function read($token)
- {
- if (empty($token)) {
- return false;
- }
- $profile = $this->getValue($this->getItemName($token), self::REDIS_SERIALIZER_PHP);
- if (false !== $profile) {
- $profile = $this->createProfileFromData($token, $profile);
- }
- return $profile;
- }
- /**
- * {@inheritdoc}
- */
- public function write(Profile $profile)
- {
- $data = array(
- 'token' => $profile->getToken(),
- 'parent' => $profile->getParentToken(),
- 'children' => array_map(function ($p) { return $p->getToken(); }, $profile->getChildren()),
- 'data' => $profile->getCollectors(),
- 'ip' => $profile->getIp(),
- 'method' => $profile->getMethod(),
- 'url' => $profile->getUrl(),
- 'time' => $profile->getTime(),
- );
- $profileIndexed = false !== $this->getValue($this->getItemName($profile->getToken()));
- if ($this->setValue($this->getItemName($profile->getToken()), $data, $this->lifetime, self::REDIS_SERIALIZER_PHP)) {
- if (!$profileIndexed) {
- // Add to index
- $indexName = $this->getIndexName();
- $indexRow = implode("\t", array(
- $profile->getToken(),
- $profile->getIp(),
- $profile->getMethod(),
- $profile->getUrl(),
- $profile->getTime(),
- $profile->getParentToken(),
- ))."\n";
- return $this->appendValue($indexName, $indexRow, $this->lifetime);
- }
- return true;
- }
- return false;
- }
- /**
- * Internal convenience method that returns the instance of Redis.
- *
- * @return Redis
- */
- protected function getRedis()
- {
- if (null === $this->redis) {
- if (!preg_match('#^redis://(?(?=\[.*\])\[(.*)\]|(.*)):(.*)$#', $this->dsn, $matches)) {
- throw new \RuntimeException(sprintf('Please check your configuration. You are trying to use Redis with an invalid dsn "%s". The expected format is "redis://[host]:port".', $this->dsn));
- }
- $host = $matches[1] ?: $matches[2];
- $port = $matches[3];
- if (!extension_loaded('redis')) {
- throw new \RuntimeException('RedisProfilerStorage requires that the redis extension is loaded.');
- }
- $redis = new Redis;
- $redis->connect($host, $port);
- $redis->setOption(self::REDIS_OPT_PREFIX, self::TOKEN_PREFIX);
- $this->redis = $redis;
- }
- return $this->redis;
- }
- /**
- * Set instance of the Redis
- *
- * @param Redis $redis
- */
- public function setRedis($redis)
- {
- $this->redis = $redis;
- }
- private function createProfileFromData($token, $data, $parent = null)
- {
- $profile = new Profile($token);
- $profile->setIp($data['ip']);
- $profile->setMethod($data['method']);
- $profile->setUrl($data['url']);
- $profile->setTime($data['time']);
- $profile->setCollectors($data['data']);
- if (!$parent && $data['parent']) {
- $parent = $this->read($data['parent']);
- }
- if ($parent) {
- $profile->setParent($parent);
- }
- foreach ($data['children'] as $token) {
- if (!$token) {
- continue;
- }
- if (!$childProfileData = $this->getValue($this->getItemName($token), self::REDIS_SERIALIZER_PHP)) {
- continue;
- }
- $profile->addChild($this->createProfileFromData($token, $childProfileData, $profile));
- }
- return $profile;
- }
- /**
- * Gets the item name.
- *
- * @param string $token
- *
- * @return string
- */
- private function getItemName($token)
- {
- $name = $token;
- if ($this->isItemNameValid($name)) {
- return $name;
- }
- return false;
- }
- /**
- * Gets the name of the index.
- *
- * @return string
- */
- private function getIndexName()
- {
- $name = 'index';
- if ($this->isItemNameValid($name)) {
- return $name;
- }
- return false;
- }
- private function isItemNameValid($name)
- {
- $length = strlen($name);
- if ($length > 2147483648) {
- throw new \RuntimeException(sprintf('The Redis item key "%s" is too long (%s bytes). Allowed maximum size is 2^31 bytes.', $name, $length));
- }
- return true;
- }
- /**
- * Retrieves an item from the Redis server.
- *
- * @param string $key
- * @param int $serializer
- *
- * @return mixed
- */
- private function getValue($key, $serializer = self::REDIS_SERIALIZER_NONE)
- {
- $redis = $this->getRedis();
- $redis->setOption(self::REDIS_OPT_SERIALIZER, $serializer);
- return $redis->get($key);
- }
- /**
- * Stores an item on the Redis server under the specified key.
- *
- * @param string $key
- * @param mixed $value
- * @param int $expiration
- * @param int $serializer
- *
- * @return Boolean
- */
- private function setValue($key, $value, $expiration = 0, $serializer = self::REDIS_SERIALIZER_NONE)
- {
- $redis = $this->getRedis();
- $redis->setOption(self::REDIS_OPT_SERIALIZER, $serializer);
- return $redis->setex($key, $expiration, $value);
- }
- /**
- * Appends data to an existing item on the Redis server.
- *
- * @param string $key
- * @param string $value
- * @param int $expiration
- *
- * @return Boolean
- */
- private function appendValue($key, $value, $expiration = 0)
- {
- $redis = $this->getRedis();
- $redis->setOption(self::REDIS_OPT_SERIALIZER, self::REDIS_SERIALIZER_NONE);
- if ($redis->exists($key)) {
- $redis->append($key, $value);
- return $redis->setTimeout($key, $expiration);
- }
- return $redis->setex($key, $expiration, $value);
- }
- /**
- * Removes the specified keys.
- *
- * @param array $keys
- *
- * @return Boolean
- */
- private function delete(array $keys)
- {
- return (bool) $this->getRedis()->delete($keys);
- }
- }