Ez\DbLinker\Driver\Connection\RetryConnection

Complexity

Total Complexity

15

Size/Duplication

Total Lines	148
Duplicated Lines	0 %

Coupling/Cohesion

Components	2
Dependencies	5

Importance

Changes	7
Bugs	1	Features	2

15 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	4	1
A	transactionLevel()	0	4	1
A	callWrappedConnectionAndRetry()	0	6	1
A	wrap()	0	8	1

<?php

namespace Ez\DbLinker\Driver\Connection;

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

class RetryConnection implements Connection, ConnectionWrapper
{
    use ConnectionWrapperTrait;

    private $wrappedConnectionParams;
    private $retryStrategy;
    private $transactionLevel = 0;

    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
     */
    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()
    {
        $this->wrappedConnection = null;
    }

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

    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();
    }
}