vendor/ekino/newrelic-bundle/Listener/ResponseListener.php line 56

  1. <?php
  3. declare(strict_types=1);
  5. /*
  6.  * This file is part of Ekino New Relic bundle.
  7.  *
  8.  * (c) Ekino - Thomas Rabaix <thomas.rabaix@ekino.com>
  9.  *
  10.  * For the full copyright and license information, please view the LICENSE
  11.  * file that was distributed with this source code.
  12.  */
  14. namespace Ekino\NewRelicBundle\Listener;
  16. use Ekino\NewRelicBundle\NewRelic\Config;
  17. use Ekino\NewRelicBundle\NewRelic\NewRelicInteractorInterface;
  18. use Ekino\NewRelicBundle\Twig\NewRelicExtension;
  19. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  20. use Symfony\Component\HttpFoundation\StreamedResponse;
  21. use Symfony\Component\HttpKernel\Event\FilterResponseEvent;
  22. use Symfony\Component\HttpKernel\Event\ResponseEvent;
  23. use Symfony\Component\HttpKernel\KernelEvents;
  25. class ResponseListener implements EventSubscriberInterface
  26. {
  27.     private $newRelic;
  28.     private $interactor;
  29.     private $instrument;
  30.     private $symfonyCache;
  31.     private $newRelicTwigExtension;
  33.     public function __construct(
  34.         Config $newRelic,
  35.         NewRelicInteractorInterface $interactor,
  36.         bool $instrument = false,
  37.         bool $symfonyCache = false,
  38.         NewRelicExtension $newRelicTwigExtension = null
  39.     ) {
  40.         $this->newRelic = $newRelic;
  41.         $this->interactor = $interactor;
  42.         $this->instrument = $instrument;
  43.         $this->symfonyCache = $symfonyCache;
  44.         $this->newRelicTwigExtension = $newRelicTwigExtension;
  45.     }
  47.     public static function getSubscribedEvents(): array
  48.     {
  49.         return [
  50.             KernelEvents::RESPONSE => [
  51.                 ['onKernelResponse', -255],
  52.             ],
  53.         ];
  54.     }
  56.     public function onKernelResponse(KernelResponseEvent $event): void
  57.     {
  58.         $isMainRequest = method_exists($event, 'isMainRequest') ? $event->isMainRequest() : $event->isMasterRequest();
  60.         if (!$isMainRequest) {
  61.             return;
  62.         }
  64.         if (null === $this->newRelicTwigExtension || false === $this->newRelicTwigExtension->isUsed()) {
  65.             foreach ($this->newRelic->getCustomMetrics() as $name => $value) {
  66.                 $this->interactor->addCustomMetric((string) $name, (float) $value);
  67.             }
  69.             foreach ($this->newRelic->getCustomParameters() as $name => $value) {
  70.                 $this->interactor->addCustomParameter((string) $name, $value);
  71.             }
  72.         }
  74.         foreach ($this->newRelic->getCustomEvents() as $name => $events) {
  75.             foreach ($events as $attributes) {
  76.                 $this->interactor->addCustomEvent((string) $name, $attributes);
  77.             }
  78.         }
  80.         if ($this->instrument) {
  81.             if (null === $this->newRelicTwigExtension || false === $this->newRelicTwigExtension->isUsed()) {
  82.                 $this->interactor->disableAutoRUM();
  83.             }
  85.             // Some requests might not want to get instrumented
  86.             if ($event->getRequest()->attributes->get('_instrument', true)) {
  87.                 $response = $event->getResponse();
  89.                 // We can only instrument HTML responses
  90.                 if (!$response instanceof StreamedResponse
  91.                     && 'text/html' === substr($response->headers->get('Content-Type', ''), 0, 9)
  92.                 ) {
  93.                     $responseContent = $response->getContent();
  94.                     $response->setContent(''); // free the memory
  96.                     if (null === $this->newRelicTwigExtension || false === $this->newRelicTwigExtension->isHeaderCalled()) {
  97.                         $responseContent = preg_replace('|<head>|i', '$0'.$this->interactor->getBrowserTimingHeader(), $responseContent);
  98.                     }
  100.                     if (null === $this->newRelicTwigExtension || false === $this->newRelicTwigExtension->isFooterCalled()) {
  101.                         $responseContent = preg_replace('|</body>|i', $this->interactor->getBrowserTimingFooter().'$0', $responseContent);
  102.                     }
  104.                     $response->setContent($responseContent);
  105.                 }
  106.             }
  107.         }
  109.         if ($this->symfonyCache) {
  110.             $this->interactor->endTransaction();
  111.         }
  112.     }
  113. }
  115. if (!class_exists(KernelResponseEvent::class)) {
  116.     if (class_exists(ResponseEvent::class)) {
  117.         class_alias(ResponseEvent::class, KernelResponseEvent::class);
  118.     } else {
  119.         class_alias(FilterResponseEvent::class, KernelResponseEvent::class);
  120.     }
  121. }