Display whether a query ran on a primary or replica in the profiler

Author: MatTheCat

config/middlewares.php

@@ -33,6 +33,7 @@
             ->args([
                 service('doctrine.debug_data_holder'),
                 service('debug.stopwatch')->nullOnInvalid(),
+                service('doctrine'),
             ])
 
         ->set('doctrine.dbal.idle_connection_middleware', IdleConnectionMiddleware::class)

src/Middleware/DebugMiddleware.php

@@ -6,6 +6,7 @@
 
 use Doctrine\DBAL\Driver as DriverInterface;
 use Doctrine\DBAL\Driver\Middleware;
+use Doctrine\Persistence\ConnectionRegistry;
 use Symfony\Bridge\Doctrine\Middleware\Debug\DebugDataHolder;
 use Symfony\Bridge\Doctrine\Middleware\Debug\Driver;
 use Symfony\Component\Stopwatch\Stopwatch;
@@ -17,6 +18,7 @@ class DebugMiddleware implements Middleware, ConnectionNameAwareInterface
     public function __construct(
         private readonly DebugDataHolder $debugDataHolder,
         private readonly Stopwatch|null $stopwatch,
+        private readonly ConnectionRegistry|null $connectionRegistry = null,
     ) {
     }
 
@@ -27,6 +29,6 @@ public function setConnectionName(string $name): void
 
     public function wrap(DriverInterface $driver): DriverInterface
     {
-        return new Driver($driver, $this->debugDataHolder, $this->stopwatch, $this->connectionName);
+        return new Driver($driver, $this->debugDataHolder, $this->stopwatch, $this->connectionName, $this->connectionRegistry);
     }
 }

templates/Collector/db.html.twig

@@ -227,7 +227,13 @@
                                             <td class="nowrap">{{ query.count }}</td>
                                         {% else %}
                                             <td class="nowrap">{{ loop.index }}</td>
-                                            <td class="nowrap">{{ '%0.2f'|format(query.executionMS * 1000) }}&nbsp;ms</td>
+                                            <td>
+                                                <span class="nowrap">{{ '%0.2f'|format(query.executionMS * 1000) }}&nbsp;ms</span>
+                                                {% if query.ranOnPrimary is defined and query.ranOnPrimary is not null %}
+                                                    <br />
+                                                    <span class="badge">{{ query.ranOnPrimary ? 'Primary' : 'Replica' }}</span>
+                                                {% endif %}
+                                            </td>
                                         {% endif %}
                                         <td>
                                             {{ query.sql|doctrine_prettify_sql }}