Fix for ConnectionMiddleware to support DBAL 3 and DBAL 4 connection …

composer.json

@@ -31,6 +31,9 @@
         "symfony/phpunit-bridge": "^6.4"
     },
     "autoload": {
+        "files": [
+            "src/aliases.php"
+        ],
         "psr-4": {
             "DataDog\\AuditBundle\\": "src"
         }

src/DBAL/Middleware/ConnectionMiddlewareForV3.php

@@ -0,0 +1,35 @@
+<?php
+
+namespace DataDog\AuditBundle\DBAL\Middleware;
+
+use Doctrine\DBAL\Driver\Connection;
+use Doctrine\DBAL\Driver\Exception;
+use Doctrine\DBAL\Driver\Middleware\AbstractConnectionMiddleware;
+
+class ConnectionMiddlewareForV3 extends AbstractConnectionMiddleware
+{
+    public function __construct(
+        Connection $wrappedConnection,
+        private readonly AuditFlushMiddleware $auditFlushMiddleware
+    )
+    {
+        parent::__construct($wrappedConnection);
+    }
+
+    /**
+     * Override of the commit method
+     *
+     * @throws Exception
+     */
+    public function commit()
+    {
+        // Call the flusher callback if it's available.
+        if ($this->auditFlushMiddleware->flushHandler !== null) {
+            ($this->auditFlushMiddleware->flushHandler)();
+            $this->auditFlushMiddleware->flushHandler = null;
+        }
+
+        // Call the parent's commit method
+        return parent::commit();
+    }
+}

src/DBAL/Middleware/ConnectionMiddleware.php → src/DBAL/Middleware/ConnectionMiddlewareForV4.php

@@ -6,7 +6,7 @@
 use Doctrine\DBAL\Driver\Exception;
 use Doctrine\DBAL\Driver\Middleware\AbstractConnectionMiddleware;
 
-class ConnectionMiddleware extends AbstractConnectionMiddleware
+class ConnectionMiddlewareForV4 extends AbstractConnectionMiddleware
 {
     public function __construct(
         Connection $wrappedConnection,

src/aliases.php

@@ -0,0 +1,18 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DataDog\AuditBundle;
+
+use DataDog\AuditBundle\DBAL\Middleware\ConnectionMiddleware;
+use DataDog\AuditBundle\DBAL\Middleware\ConnectionMiddlewareForV3;
+use DataDog\AuditBundle\DBAL\Middleware\ConnectionMiddlewareForV4;
+use Doctrine\DBAL\VersionAwarePlatformDriver;
+
+if (!class_exists(ConnectionMiddleware::class)) {
+    if (!interface_exists(VersionAwarePlatformDriver::class)) {
+        class_alias(ConnectionMiddlewareForV4::class, ConnectionMiddleware::class);
+    } else {
+        class_alias(ConnectionMiddlewareForV3::class, ConnectionMiddleware::class);
+    }
+}