Compatibility with the new logging middleware

composer.json

@@ -30,6 +30,7 @@
         "phpunit/phpunit": "<8.0"
     },
     "require-dev": {
+        "doctrine/annotations": "^2.0",
         "doctrine/doctrine-bundle": "^2.0",
         "doctrine/orm": "^2.9",
         "phpunit/phpunit": "^9.3",

config/services.php

@@ -17,6 +17,9 @@
         ->set('monitoring.metrics', MetricRegistry::class)
 
         ->set('monitoring.metric.doctrine', DoctrineMetric::class)
+            ->args([
+                service('doctrine.debug_data_holder')->nullOnInvalid()
+            ])
             ->tag('monitoring.metric')
         ->set('monitoring.metric.memory', MemoryMetric::class)
             ->tag('monitoring.metric')

src/DependencyInjection/Compiler/AddDoctrineLoggerPass.php

@@ -16,6 +16,11 @@ public function process(ContainerBuilder $container): void
             return;
         }
 
+        // With DebugDataHolder, no need to use logger
+        if ($container->has('doctrine.debug_data_holder')) {
+            return;
+        }
+
         $definition = $container->getDefinition('monitoring.metric.doctrine');
 
         $connections = $container->getParameter('doctrine.connections');

src/DependencyInjection/Compiler/MetricRegistryPass.php

@@ -9,7 +9,7 @@
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Reference;
 
-class MetricRegistryPass implements CompilerPassInterface
+final class MetricRegistryPass implements CompilerPassInterface
 {
     public function process(ContainerBuilder $container): void
     {

src/DependencyInjection/MonitoringMetricsExtension.php

@@ -9,7 +9,7 @@
 use Symfony\Component\DependencyInjection\Extension\Extension;
 use Symfony\Component\DependencyInjection\Loader\PhpFileLoader;
 
-class MonitoringMetricsExtension extends Extension
+final class MonitoringMetricsExtension extends Extension
 {
     public function load(array $configs, ContainerBuilder $container): void
     {

src/EventListener/MonitoringListener.php

@@ -21,7 +21,7 @@ public function __construct(MetricRegistry $metricRegistry)
     public function onKernelResponse(ResponseEvent $event): void
     {
         $route = $event->getRequest()->attributes->get('_route');
-        if (null === $route || strpos($route, '_') === 0) {
+        if (null === $route || str_starts_with($route, '_')) {
             return;
         }
 

src/Metric/DoctrineMetric.php

@@ -5,17 +5,25 @@
 namespace Odandb\MonitoringMetricsBundle\Metric;
 
 use Doctrine\DBAL\Logging\DebugStack;
+use Symfony\Bridge\Doctrine\Middleware\Debug\DebugDataHolder;
 use Symfony\Component\HttpFoundation\Response;
 
 class DoctrineMetric extends AbstractMetric
 {
     public const HEADER_NAME = 'X-Monitor-Doctrine-Queries';
 
+    protected ?DebugDataHolder $debugDataHolder;
+
     /**
      * @var DebugStack[]
      */
     private array $loggers = [];
 
+    public function __construct(?DebugDataHolder $debugDataHolder = null)
+    {
+        $this->debugDataHolder = $debugDataHolder;
+    }
+
     /**
      * Adds the stack logger for a connection.
      */
@@ -27,6 +35,21 @@ public function addLogger(string $name, DebugStack $logger): void
     public function metric(Response $response): void
     {
         $queries = [];
+
+        if (null !== $this->debugDataHolder) {
+            foreach ($this->debugDataHolder->getData() as $name => $data) {
+                if (!isset($queries[$name])) {
+                    $queries[$name] = 0;
+                }
+
+                $queries[$name] += count($data);
+            }
+
+            $response->headers->set(self::HEADER_NAME, (string) array_sum($queries));
+
+            return;
+        }
+
         foreach ($this->loggers as $name => $logger) {
             if (!isset($queries[$name])) {
                 $queries[$name] = 0;
@@ -40,6 +63,12 @@ public function metric(Response $response): void
 
     public function reset(): void
     {
+        if (null !== $this->debugDataHolder) {
+            $this->debugDataHolder->reset();
+
+            return;
+        }
+
         foreach ($this->loggers as $logger) {
             $logger->queries = [];
             $logger->currentQuery = 0;

tests/Fixtures/config.yaml

@@ -5,6 +5,7 @@ framework:
     annotations:
         enabled: true
     secret: '123456789'
+    http_method_override: false
     router:
         utf8: true
         resource: '%kernel.project_dir%/routing.yaml'