src/EventSubscriber/DoctrineSlowQuerySubscriber.php line 34

Open in your IDE?
  1. <?php
  3. namespace App\EventSubscriber;
  5. use App\Doctrine\Logging\ChainSqlLogger;
  6. use App\Doctrine\Logging\SlowQuerySqlLogger;
  7. use Doctrine\DBAL\Connection;
  8. use Symfony\Component\Console\ConsoleEvents;
  9. use Symfony\Component\Console\Event\ConsoleCommandEvent;
  10. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  11. use Symfony\Component\HttpKernel\Event\RequestEvent;
  12. use Symfony\Component\HttpKernel\KernelEvents;
  14. class DoctrineSlowQuerySubscriber implements EventSubscriberInterface
  15. {
  16.     private Connection $connection;
  17.     private SlowQuerySqlLogger $slowQuerySqlLogger;
  18.     private bool $initialized = false;
  20.     public function __construct(Connection $connection, SlowQuerySqlLogger $slowQuerySqlLogger)
  21.     {
  22.         $this->connection = $connection;
  23.         $this->slowQuerySqlLogger = $slowQuerySqlLogger;
  24.     }
  26.     public static function getSubscribedEvents(): array
  27.     {
  28.         return [
  29.             KernelEvents::REQUEST => ['onKernelRequest', 1000],
  30.             ConsoleEvents::COMMAND => ['onConsoleCommand', 1000],
  31.         ];
  32.     }
  34.     public function onKernelRequest(RequestEvent $event): void
  35.     {
  36.         if (!$event->isMainRequest()) {
  37.             return;
  38.         }
  40.         $this->enableSlowQueryLogger();
  41.     }
  43.     public function onConsoleCommand(ConsoleCommandEvent $event): void
  44.     {
  45.         $this->enableSlowQueryLogger();
  46.     }
  48.     private function enableSlowQueryLogger(): void
  49.     {
  50.         if ($this->initialized) {
  51.             return;
  52.         }
  54.         $configuration = $this->connection->getConfiguration();
  55.         $currentLogger = $configuration->getSQLLogger();
  57.         if ($currentLogger instanceof SlowQuerySqlLogger || $currentLogger instanceof ChainSqlLogger) {
  58.             $this->initialized = true;
  59.             return;
  60.         }
  62.         if ($currentLogger !== null) {
  63.             $configuration->setSQLLogger(new ChainSqlLogger([$currentLogger, $this->slowQuerySqlLogger]));
  64.         } else {
  65.             $configuration->setSQLLogger($this->slowQuerySqlLogger);
  66.         }
  68.         $this->initialized = true;
  69.     }
  70. }