vendor/symfony/http-client/DataCollector/HttpClientDataCollector.php line 39

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the Symfony package.
  5.  *
  6.  * (c) Fabien Potencier <fabien@symfony.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace Symfony\Component\HttpClient\DataCollector;
  14. use Symfony\Component\HttpClient\TraceableHttpClient;
  15. use Symfony\Component\HttpFoundation\Request;
  16. use Symfony\Component\HttpFoundation\Response;
  17. use Symfony\Component\HttpKernel\DataCollector\DataCollector;
  18. use Symfony\Component\HttpKernel\DataCollector\LateDataCollectorInterface;
  19. use Symfony\Component\VarDumper\Caster\ImgStub;
  21. /**
  22.  * @author Jérémy Romey <jeremy@free-agent.fr>
  23.  */
  24. final class HttpClientDataCollector extends DataCollector implements LateDataCollectorInterface
  25. {
  26.     /**
  27.      * @var TraceableHttpClient[]
  28.      */
  29.     private $clients = [];
  31.     public function registerClient(string $name, TraceableHttpClient $client)
  32.     {
  33.         $this->clients[$name] = $client;
  34.     }
  36.     /**
  37.      * {@inheritdoc}
  38.      */
  39.     public function collect(Request $request, Response $response, \Throwable $exception = null)
  40.     {
  41.         $this->reset();
  43.         foreach ($this->clients as $name => $client) {
  44.             [$errorCount, $traces] = $this->collectOnClient($client);
  46.             $this->data['clients'][$name] = [
  47.                 'traces' => $traces,
  48.                 'error_count' => $errorCount,
  49.             ];
  51.             $this->data['request_count'] += \count($traces);
  52.             $this->data['error_count'] += $errorCount;
  53.         }
  54.     }
  56.     public function lateCollect()
  57.     {
  58.         foreach ($this->clients as $client) {
  59.             $client->reset();
  60.         }
  61.     }
  63.     public function getClients(): array
  64.     {
  65.         return $this->data['clients'] ?? [];
  66.     }
  68.     public function getRequestCount(): int
  69.     {
  70.         return $this->data['request_count'] ?? 0;
  71.     }
  73.     public function getErrorCount(): int
  74.     {
  75.         return $this->data['error_count'] ?? 0;
  76.     }
  78.     /**
  79.      * {@inheritdoc}
  80.      */
  81.     public function getName(): string
  82.     {
  83.         return 'http_client';
  84.     }
  86.     public function reset()
  87.     {
  88.         $this->data = [
  89.             'clients' => [],
  90.             'request_count' => 0,
  91.             'error_count' => 0,
  92.         ];
  93.     }
  95.     private function collectOnClient(TraceableHttpClient $client): array
  96.     {
  97.         $traces = $client->getTracedRequests();
  98.         $errorCount = 0;
  99.         $baseInfo = [
  100.             'response_headers' => 1,
  101.             'retry_count' => 1,
  102.             'redirect_count' => 1,
  103.             'redirect_url' => 1,
  104.             'user_data' => 1,
  105.             'error' => 1,
  106.             'url' => 1,
  107.         ];
  109.         foreach ($traces as $i => $trace) {
  110.             if (400 <= ($trace['info']['http_code'] ?? 0)) {
  111.                 ++$errorCount;
  112.             }
  114.             $info = $trace['info'];
  115.             $traces[$i]['http_code'] = $info['http_code'] ?? 0;
  117.             unset($info['filetime'], $info['http_code'], $info['ssl_verify_result'], $info['content_type']);
  119.             if (($info['http_method'] ?? null) === $trace['method']) {
  120.                 unset($info['http_method']);
  121.             }
  123.             if (($info['url'] ?? null) === $trace['url']) {
  124.                 unset($info['url']);
  125.             }
  127.             foreach ($info as $k => $v) {
  128.                 if (!$v || (is_numeric($v) && 0 > $v)) {
  129.                     unset($info[$k]);
  130.                 }
  131.             }
  133.             if (\is_string($content = $trace['content'])) {
  134.                 $contentType = 'application/octet-stream';
  136.                 foreach ($info['response_headers'] ?? [] as $h) {
  137.                     if (0 === stripos($h, 'content-type: ')) {
  138.                         $contentType = substr($h, \strlen('content-type: '));
  139.                         break;
  140.                     }
  141.                 }
  143.                 if (0 === strpos($contentType, 'image/') && class_exists(ImgStub::class)) {
  144.                     $content = new ImgStub($content, $contentType, '');
  145.                 } else {
  146.                     $content = [$content];
  147.                 }
  149.                 $content = ['response_content' => $content];
  150.             } elseif (\is_array($content)) {
  151.                 $content = ['response_json' => $content];
  152.             } else {
  153.                 $content = [];
  154.             }
  156.             if (isset($info['retry_count'])) {
  157.                 $content['retries'] = $info['previous_info'];
  158.                 unset($info['previous_info']);
  159.             }
  161.             $debugInfo = array_diff_key($info, $baseInfo);
  162.             $info = ['info' => $debugInfo] + array_diff_key($info, $debugInfo) + $content;
  163.             unset($traces[$i]['info']); // break PHP reference used by TraceableHttpClient
  164.             $traces[$i]['info'] = $this->cloneVar($info);
  165.             $traces[$i]['options'] = $this->cloneVar($trace['options']);
  166.         }
  168.         return [$errorCount, $traces];
  169.     }
  170. }