123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314 |
- <?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;
- /**
- * Storage for profiler using files.
- *
- * @author Alexandre Salomé <alexandre.salome@gmail.com>
- */
- class FileProfilerStorage implements ProfilerStorageInterface
- {
- /**
- * Folder where profiler data are stored.
- *
- * @var string
- */
- private $folder;
- /**
- * Constructs the file storage using a "dsn-like" path.
- *
- * Example : "file:/path/to/the/storage/folder"
- *
- * @throws \RuntimeException
- */
- public function __construct(string $dsn)
- {
- if (!str_starts_with($dsn, 'file:')) {
- throw new \RuntimeException(sprintf('Please check your configuration. You are trying to use FileStorage with an invalid dsn "%s". The expected format is "file:/path/to/the/storage/folder".', $dsn));
- }
- $this->folder = substr($dsn, 5);
- if (!is_dir($this->folder) && false === @mkdir($this->folder, 0777, true) && !is_dir($this->folder)) {
- throw new \RuntimeException(sprintf('Unable to create the storage directory (%s).', $this->folder));
- }
- }
- /**
- * {@inheritdoc}
- */
- public function find($ip, $url, $limit, $method, $start = null, $end = null, $statusCode = null): array
- {
- $file = $this->getIndexFilename();
- if (!file_exists($file)) {
- return [];
- }
- $file = fopen($file, 'r');
- fseek($file, 0, \SEEK_END);
- $result = [];
- while (\count($result) < $limit && $line = $this->readLineFromFile($file)) {
- $values = str_getcsv($line);
- [$csvToken, $csvIp, $csvMethod, $csvUrl, $csvTime, $csvParent, $csvStatusCode] = $values;
- $csvTime = (int) $csvTime;
- if ($ip && !str_contains($csvIp, $ip) || $url && !str_contains($csvUrl, $url) || $method && !str_contains($csvMethod, $method) || $statusCode && !str_contains($csvStatusCode, $statusCode)) {
- continue;
- }
- if (!empty($start) && $csvTime < $start) {
- continue;
- }
- if (!empty($end) && $csvTime > $end) {
- continue;
- }
- $result[$csvToken] = [
- 'token' => $csvToken,
- 'ip' => $csvIp,
- 'method' => $csvMethod,
- 'url' => $csvUrl,
- 'time' => $csvTime,
- 'parent' => $csvParent,
- 'status_code' => $csvStatusCode,
- ];
- }
- fclose($file);
- return array_values($result);
- }
- /**
- * {@inheritdoc}
- */
- public function purge()
- {
- $flags = \FilesystemIterator::SKIP_DOTS;
- $iterator = new \RecursiveDirectoryIterator($this->folder, $flags);
- $iterator = new \RecursiveIteratorIterator($iterator, \RecursiveIteratorIterator::CHILD_FIRST);
- foreach ($iterator as $file) {
- if (is_file($file)) {
- unlink($file);
- } else {
- rmdir($file);
- }
- }
- }
- /**
- * {@inheritdoc}
- */
- public function read($token): ?Profile
- {
- if (!$token || !file_exists($file = $this->getFilename($token))) {
- return null;
- }
- if (\function_exists('gzcompress')) {
- $file = 'compress.zlib://'.$file;
- }
- if (!$data = unserialize(file_get_contents($file))) {
- return null;
- }
- return $this->createProfileFromData($token, $data);
- }
- /**
- * {@inheritdoc}
- *
- * @throws \RuntimeException
- */
- public function write(Profile $profile): bool
- {
- $file = $this->getFilename($profile->getToken());
- $profileIndexed = is_file($file);
- if (!$profileIndexed) {
- // Create directory
- $dir = \dirname($file);
- if (!is_dir($dir) && false === @mkdir($dir, 0777, true) && !is_dir($dir)) {
- throw new \RuntimeException(sprintf('Unable to create the storage directory (%s).', $dir));
- }
- }
- $profileToken = $profile->getToken();
- // when there are errors in sub-requests, the parent and/or children tokens
- // may equal the profile token, resulting in infinite loops
- $parentToken = $profile->getParentToken() !== $profileToken ? $profile->getParentToken() : null;
- $childrenToken = array_filter(array_map(function (Profile $p) use ($profileToken) {
- return $profileToken !== $p->getToken() ? $p->getToken() : null;
- }, $profile->getChildren()));
- // Store profile
- $data = [
- 'token' => $profileToken,
- 'parent' => $parentToken,
- 'children' => $childrenToken,
- 'data' => $profile->getCollectors(),
- 'ip' => $profile->getIp(),
- 'method' => $profile->getMethod(),
- 'url' => $profile->getUrl(),
- 'time' => $profile->getTime(),
- 'status_code' => $profile->getStatusCode(),
- ];
- $context = stream_context_create();
- if (\function_exists('gzcompress')) {
- $file = 'compress.zlib://'.$file;
- stream_context_set_option($context, 'zlib', 'level', 3);
- }
- if (false === file_put_contents($file, serialize($data), 0, $context)) {
- return false;
- }
- if (!$profileIndexed) {
- // Add to index
- if (false === $file = fopen($this->getIndexFilename(), 'a')) {
- return false;
- }
- fputcsv($file, [
- $profile->getToken(),
- $profile->getIp(),
- $profile->getMethod(),
- $profile->getUrl(),
- $profile->getTime(),
- $profile->getParentToken(),
- $profile->getStatusCode(),
- ]);
- fclose($file);
- }
- return true;
- }
- /**
- * Gets filename to store data, associated to the token.
- *
- * @param string $token
- *
- * @return string The profile filename
- */
- protected function getFilename($token)
- {
- // Uses 4 last characters, because first are mostly the same.
- $folderA = substr($token, -2, 2);
- $folderB = substr($token, -4, 2);
- return $this->folder.'/'.$folderA.'/'.$folderB.'/'.$token;
- }
- /**
- * Gets the index filename.
- *
- * @return string The index filename
- */
- protected function getIndexFilename()
- {
- return $this->folder.'/index.csv';
- }
- /**
- * Reads a line in the file, backward.
- *
- * This function automatically skips the empty lines and do not include the line return in result value.
- *
- * @param resource $file The file resource, with the pointer placed at the end of the line to read
- *
- * @return mixed A string representing the line or null if beginning of file is reached
- */
- protected function readLineFromFile($file)
- {
- $line = '';
- $position = ftell($file);
- if (0 === $position) {
- return null;
- }
- while (true) {
- $chunkSize = min($position, 1024);
- $position -= $chunkSize;
- fseek($file, $position);
- if (0 === $chunkSize) {
- // bof reached
- break;
- }
- $buffer = fread($file, $chunkSize);
- if (false === ($upTo = strrpos($buffer, "\n"))) {
- $line = $buffer.$line;
- continue;
- }
- $position += $upTo;
- $line = substr($buffer, $upTo + 1).$line;
- fseek($file, max(0, $position), \SEEK_SET);
- if ('' !== $line) {
- break;
- }
- }
- return '' === $line ? null : $line;
- }
- protected 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->setStatusCode($data['status_code']);
- $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 || !file_exists($file = $this->getFilename($token))) {
- continue;
- }
- if (\function_exists('gzcompress')) {
- $file = 'compress.zlib://'.$file;
- }
- if (!$childData = unserialize(file_get_contents($file))) {
- continue;
- }
- $profile->addChild($this->createProfileFromData($token, $childData, $profile));
- }
- return $profile;
- }
- }
|