DBAL Middleware for Transaction Isolation Level

transaction-isolation-level-dbal-middleware.php

<?php

declare(strict_types=1);

namespace MyProject\Database;

use Doctrine\Bundle\DoctrineBundle;
use Doctrine\DBAL\Connection;
use Doctrine\DBAL\Driver;
use Doctrine\DBAL\Driver\API\ExceptionConverter;
use Doctrine\DBAL\Driver\Connection as DriverConnection;
use Doctrine\DBAL\Driver\Middleware;
use Doctrine\DBAL\Platforms\AbstractPlatform;
use Doctrine\DBAL\Schema\AbstractSchemaManager;
use Doctrine\DBAL\TransactionIsolationLevel;
use SensitiveParameter;

#[DoctrineBundle\Attribute\AsMiddleware]
final class TransactionIsolationLevelDoctrineMiddleware implements Middleware
{
    public function wrap(Driver $driver): Driver
    {
        return new class ($driver) implements Driver {
            public function __construct(
                private Driver $wrappedDriver,
            ) {}

            public function connect(#[SensitiveParameter] array $params): DriverConnection
            {
                $connection = $this->wrappedDriver->connect($params);
                $connection->exec(
                    $this->wrappedDriver->getDatabasePlatform()
                        ->getSetTransactionIsolationSQL(TransactionIsolationLevel::READ_COMMITTED)
                );
                return $connection;
            }

            public function getDatabasePlatform(): AbstractPlatform
            {
                return $this->wrappedDriver->getDatabasePlatform();
            }

            public function getSchemaManager(Connection $conn, AbstractPlatform $platform): AbstractSchemaManager
            {
                /** @psalm-suppress DeprecatedMethod */
                return $this->wrappedDriver->getSchemaManager($conn, $platform);
            }

            public function getExceptionConverter(): ExceptionConverter
            {
                return $this->wrappedDriver->getExceptionConverter();
            }
        };
    }
}