123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310 |
- <?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 Symfony\Component\HttpFoundation\Exception\ConflictingHeadersException;
- use Symfony\Component\HttpFoundation\Request;
- use Symfony\Component\HttpFoundation\Response;
- use Symfony\Component\HttpKernel\DataCollector\DataCollectorInterface;
- use Symfony\Component\HttpKernel\DataCollector\LateDataCollectorInterface;
- use Psr\Log\LoggerInterface;
- /**
- * Profiler.
- *
- * @author Fabien Potencier <fabien@symfony.com>
- */
- class Profiler
- {
- /**
- * @var ProfilerStorageInterface
- */
- private $storage;
- /**
- * @var DataCollectorInterface[]
- */
- private $collectors = array();
- /**
- * @var LoggerInterface
- */
- private $logger;
- /**
- * @var bool
- */
- private $enabled = true;
- /**
- * Constructor.
- *
- * @param ProfilerStorageInterface $storage A ProfilerStorageInterface instance
- * @param LoggerInterface $logger A LoggerInterface instance
- */
- public function __construct(ProfilerStorageInterface $storage, LoggerInterface $logger = null)
- {
- $this->storage = $storage;
- $this->logger = $logger;
- }
- /**
- * Disables the profiler.
- */
- public function disable()
- {
- $this->enabled = false;
- }
- /**
- * Enables the profiler.
- */
- public function enable()
- {
- $this->enabled = true;
- }
- /**
- * Loads the Profile for the given Response.
- *
- * @param Response $response A Response instance
- *
- * @return Profile|false A Profile instance
- */
- public function loadProfileFromResponse(Response $response)
- {
- if (!$token = $response->headers->get('X-Debug-Token')) {
- return false;
- }
- return $this->loadProfile($token);
- }
- /**
- * Loads the Profile for the given token.
- *
- * @param string $token A token
- *
- * @return Profile A Profile instance
- */
- public function loadProfile($token)
- {
- return $this->storage->read($token);
- }
- /**
- * Saves a Profile.
- *
- * @param Profile $profile A Profile instance
- *
- * @return bool
- */
- public function saveProfile(Profile $profile)
- {
- // late collect
- foreach ($profile->getCollectors() as $collector) {
- if ($collector instanceof LateDataCollectorInterface) {
- $collector->lateCollect();
- }
- }
- if (!($ret = $this->storage->write($profile)) && null !== $this->logger) {
- $this->logger->warning('Unable to store the profiler information.', array('configured_storage' => get_class($this->storage)));
- }
- return $ret;
- }
- /**
- * Purges all data from the storage.
- */
- public function purge()
- {
- $this->storage->purge();
- }
- /**
- * Exports the current profiler data.
- *
- * @param Profile $profile A Profile instance
- *
- * @return string The exported data
- *
- * @deprecated since Symfony 2.8, to be removed in 3.0.
- */
- public function export(Profile $profile)
- {
- @trigger_error('The '.__METHOD__.' method is deprecated since version 2.8 and will be removed in 3.0.', E_USER_DEPRECATED);
- return base64_encode(serialize($profile));
- }
- /**
- * Imports data into the profiler storage.
- *
- * @param string $data A data string as exported by the export() method
- *
- * @return Profile|false A Profile instance
- *
- * @deprecated since Symfony 2.8, to be removed in 3.0.
- */
- public function import($data)
- {
- @trigger_error('The '.__METHOD__.' method is deprecated since version 2.8 and will be removed in 3.0.', E_USER_DEPRECATED);
- $profile = unserialize(base64_decode($data));
- if ($this->storage->read($profile->getToken())) {
- return false;
- }
- $this->saveProfile($profile);
- return $profile;
- }
- /**
- * Finds profiler tokens for the given criteria.
- *
- * @param string $ip The IP
- * @param string $url The URL
- * @param string $limit The maximum number of tokens to return
- * @param string $method The request method
- * @param string $start The start date to search from
- * @param string $end The end date to search to
- *
- * @return array An array of tokens
- *
- * @see http://php.net/manual/en/datetime.formats.php for the supported date/time formats
- */
- public function find($ip, $url, $limit, $method, $start, $end)
- {
- return $this->storage->find($ip, $url, $limit, $method, $this->getTimestamp($start), $this->getTimestamp($end));
- }
- /**
- * Collects data for the given Response.
- *
- * @param Request $request A Request instance
- * @param Response $response A Response instance
- * @param \Exception $exception An exception instance if the request threw one
- *
- * @return Profile|null A Profile instance or null if the profiler is disabled
- */
- public function collect(Request $request, Response $response, \Exception $exception = null)
- {
- if (false === $this->enabled) {
- return;
- }
- $profile = new Profile(substr(hash('sha256', uniqid(mt_rand(), true)), 0, 6));
- $profile->setTime(time());
- $profile->setUrl($request->getUri());
- $profile->setMethod($request->getMethod());
- $profile->setStatusCode($response->getStatusCode());
- try {
- $profile->setIp($request->getClientIp());
- } catch (ConflictingHeadersException $e) {
- $profile->setIp('Unknown');
- }
- $response->headers->set('X-Debug-Token', $profile->getToken());
- foreach ($this->collectors as $collector) {
- $collector->collect($request, $response, $exception);
- // we need to clone for sub-requests
- $profile->addCollector(clone $collector);
- }
- return $profile;
- }
- /**
- * Gets the Collectors associated with this profiler.
- *
- * @return array An array of collectors
- */
- public function all()
- {
- return $this->collectors;
- }
- /**
- * Sets the Collectors associated with this profiler.
- *
- * @param DataCollectorInterface[] $collectors An array of collectors
- */
- public function set(array $collectors = array())
- {
- $this->collectors = array();
- foreach ($collectors as $collector) {
- $this->add($collector);
- }
- }
- /**
- * Adds a Collector.
- *
- * @param DataCollectorInterface $collector A DataCollectorInterface instance
- */
- public function add(DataCollectorInterface $collector)
- {
- $this->collectors[$collector->getName()] = $collector;
- }
- /**
- * Returns true if a Collector for the given name exists.
- *
- * @param string $name A collector name
- *
- * @return bool
- */
- public function has($name)
- {
- return isset($this->collectors[$name]);
- }
- /**
- * Gets a Collector by name.
- *
- * @param string $name A collector name
- *
- * @return DataCollectorInterface A DataCollectorInterface instance
- *
- * @throws \InvalidArgumentException if the collector does not exist
- */
- public function get($name)
- {
- if (!isset($this->collectors[$name])) {
- throw new \InvalidArgumentException(sprintf('Collector "%s" does not exist.', $name));
- }
- return $this->collectors[$name];
- }
- private function getTimestamp($value)
- {
- if (null === $value || '' == $value) {
- return;
- }
- try {
- $value = new \DateTime(is_numeric($value) ? '@'.$value : $value);
- } catch (\Exception $e) {
- return;
- }
- return $value->getTimestamp();
- }
- }
|