Leverage DBAL's getNativeConnection() method

symfony · Nov 27, 2021 · cde5ec7 · cde5ec7
1 parent c236dfe
commit cde5ec7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/Symfony/Component/Messenger/Bridge/Doctrine/Transport/PostgreSqlConnection.php b/src/Symfony/Component/Messenger/Bridge/Doctrine/Transport/PostgreSqlConnection.php
@@ -73,9 +73,13 @@ public function get(): ?array
             $this->listening = true;
         }
 
-        $wrappedConnection = $this->driverConnection->getWrappedConnection();
-        if (!$wrappedConnection instanceof \PDO && $wrappedConnection instanceof DoctrinePdoConnection) {
-            $wrappedConnection = $wrappedConnection->getWrappedConnection();
+        if (method_exists($this->driverConnection, 'getNativeConnection')) {
+            $wrappedConnection = $this->driverConnection->getNativeConnection();
+        } else {
+            $wrappedConnection = $this->driverConnection->getWrappedConnection();
+            if (!$wrappedConnection instanceof \PDO && $wrappedConnection instanceof DoctrinePdoConnection) {
+                $wrappedConnection = $wrappedConnection->getWrappedConnection();
+            }
         }
 
         $notification = $wrappedConnection->pgsqlGetNotify(\PDO::FETCH_ASSOC, $this->configuration['get_notify_timeout']);