Ez\DbLinker\Driver\Connection\RetryConnection

Complexity

Total Complexity

21

Size/Duplication

Total Lines	180
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Importance

Changes

0

17 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	prepare()	0	7	1
A	query()	0	3	1
A	quote()	0	3	1
A	exec()	0	3	1
A	lastInsertId()	0	3	1
A	beginTransaction()	0	4	1
A	commit()	0	4	1
A	rollBack()	0	4	1
A	errorCode()	0	3	1
A	errorInfo()	0	3	1
A	close()	0	8	3
A	transactionLevel()	0	4	1
A	callWrappedConnectionAndRetry()	0	6	1
A	wrap()	0	8	1
A	wrappedConnection()	0	7	2
A	wrappedDriver()	0	7	2

<?php

namespace Ez\DbLinker\Driver\Connection;

use Doctrine\DBAL\Driver\Connection;
use Doctrine\DBAL\Driver\PDOConnection;
use Doctrine\DBAL\DriverManager;
use Ez\DbLinker\RetryStrategy;

class RetryConnection implements Connection
{
    private $wrappedConnectionParams;
    private $retryStrategy;
    private $transactionLevel = 0;
    private $wrappedConnection;
    private $wrappedDriver;

    use CallAndRetry;

    public function __construct(Array $wrappedConnectionParams, RetryStrategy $retryStrategy)
    {
        $this->wrappedConnectionParams = $wrappedConnectionParams;
        $this->retryStrategy = $retryStrategy;
    }

    /**
     * Prepares a statement for execution and returns a Statement object.
     *
     * @param string $prepareString
     *
     * @return \Doctrine\DBAL\Driver\Statement
     */
    public function prepare($prepareString) {
        return new RetryStatement(
            $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args()),
            $this,
            $this->retryStrategy
        );
    }

    /**
     * Executes an SQL statement, returning a result set as a Statement object.
     *
     * @return \Doctrine\DBAL\Driver\Statement
     * @throws \Exception
     */
    public function query() {
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Quotes a string for use in a query.
     *
     * @param string  $input
     * @param integer $type
     *
     * @return string
     */
    public function quote($input, $type = \PDO::PARAM_STR) {
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Executes an SQL statement and return the number of affected rows.
     *
     * @param string $statement
     *
     * @return integer
     */
    public function exec($statement) {
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Returns the ID of the last inserted row or sequence value.
     *
     * @param string|null $name
     *
     * @return string
     */
    public function lastInsertId($name = null) {
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Initiates a transaction.
     *
     * @return boolean TRUE on success or FALSE on failure.
     */
    public function beginTransaction() {
        $this->transactionLevel++;
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Commits a transaction.
     *
     * @return boolean TRUE on success or FALSE on failure.
     */
    public function commit() {
        $this->transactionLevel--;
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Rolls back the current transaction, as initiated by beginTransaction().
     *
     * @return boolean TRUE on success or FALSE on failure.
     */
    public function rollBack() {
        $this->transactionLevel--;
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Returns the error code associated with the last operation on the database handle.
     *
     * @return string|null The error code, or null if no operation has been run on the database handle.
     */
    public function errorCode() {
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    /**
     * Returns extended error information associated with the last operation on the database handle.
     *
     * @return array
     */
    public function errorInfo() {
        return $this->callWrappedConnectionAndRetry(__FUNCTION__, func_get_args());
    }

    public function close()
    {
        if ($this->wrappedConnection instanceof PDOConnection) {
        } elseif (method_exists($this->wrappedConnection, "getWrappedResourceHandle")) {
            $this->wrappedConnection->getWrappedResourceHandle()->close();
        }
        $this->wrappedConnection = null;
    }

    public function transactionLevel()
    {
        return $this->transactionLevel;
    }

    /**
     * @param $method
     * @param array $arguments
     * @return mixed
     * @throws \Exception
     */
    private function callWrappedConnectionAndRetry($method, array $arguments)
    {
        return $this->callAndRetry(function () use ($method, $arguments) {
            return \call_user_func_array([$this->wrappedConnection(), $method], $arguments);
        }, $this->retryStrategy, $this);
    }

    protected function wrap()
    {
        $connection = $this->wrappedConnection = DriverManager::getConnection(
            $this->wrappedConnectionParams
        );
        $this->wrappedConnection = $connection->getWrappedConnection();
        $this->wrappedDriver = $connection->getDriver();
    }

    /**
     * @inherit
     * @return \Doctrine\DBAL\Driver\PDOConnection
     */
    public function wrappedConnection()
    {
        if ($this->wrappedConnection === null) {
            $this->wrap();
        }
        return $this->wrappedConnection;
    }

    public function wrappedDriver()
    {
        if ($this->wrappedDriver === null) {
            $this->wrap();
        }
        return $this->wrappedDriver;
    }

}