League\Tactician\Doctrine\DBAL\PingConnectionMiddleware

Complexity

Total Complexity

5

Size/Duplication

Total Lines	40
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	execute()	0	8	2
A	ping()	0	16	2

<?php

declare(strict_types=1);

namespace League\Tactician\Doctrine\DBAL;

use Doctrine\DBAL\Connection;
use ErrorException;
use League\Tactician\Middleware;
use Throwable;

use function restore_error_handler;
use function set_error_handler;

/**
 * Verifies if there is a connection established with the database. If not it will reconnect.
 */
final class PingConnectionMiddleware implements Middleware
{
    private Connection $connection;

    public function __construct(Connection $connection)
    {
        $this->connection = $connection;
    }

    /**
     * Reconnects to the database if the connection is expired.
     *
     * @return mixed
     */
    public function execute(object $command, callable $next)
    {
        if (! $this->ping($this->connection)) {
            $this->connection->close();
            $this->connection->connect();
        }

        return $next($command);
    }

    private function ping(Connection $connection): bool
    {
        set_error_handler(static function (int $severity, string $message, string $file, int $line): bool {
            throw new ErrorException($message, $severity, $severity, $file, $line);
        });

        try {
            $dummySelectSQL = $connection->getDatabasePlatform()->getDummySelectSQL();

            $connection->executeQuery($dummySelectSQL);

            return true;
        } catch (Throwable $exception) {
            return false;
        } finally {
            restore_error_handler();
        }
    }
}