Ray\AuraSqlModule\AuraSqlConnectionInterceptor

Complexity

Total Complexity

4

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Importance

Changes	3
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	getConnection()	0	8	2
A	invoke()	0	9	1
A	__construct()	0	5	1

<?php

declare(strict_types=1);

namespace Ray\AuraSqlModule;

use Aura\Sql\ConnectionLocatorInterface;
use Aura\Sql\ExtendedPdoInterface;
use Ray\Aop\MethodInterceptor;
use Ray\Aop\MethodInvocation;
use Ray\AuraSqlModule\Annotation\Read;
use ReflectionProperty;

use function in_array;

class AuraSqlConnectionInterceptor implements MethodInterceptor
{
    public const PROP = 'pdo';

    private ConnectionLocatorInterface $connectionLocator;

    /** @var string[] */
    private array $readsMethods = [];

    /**
     * @phpstan-param array<string> $readMethods
     *
     * @Read("readMethods")
     */
    #[Read('readMethods')]
    public function __construct(ConnectionLocatorInterface $connectionLocator, array $readMethods)
    {
        $this->connectionLocator = $connectionLocator;
        $this->readsMethods = $readMethods;
    }

    /**
     * {@inheritDoc}
     */
    public function invoke(MethodInvocation $invocation)
    {
        $connection = $this->getConnection($invocation);
        $object = $invocation->getThis();
        $ref = new ReflectionProperty($object, self::PROP);
        $ref->setAccessible(true);
        $ref->setValue($object, $connection);

        return $invocation->proceed();
    }

    /** @param MethodInvocation<object> $invocation */
    private function getConnection(MethodInvocation $invocation): ExtendedPdoInterface
    {
        $methodName = $invocation->getMethod()->name;
        if (in_array($methodName, $this->readsMethods, true)) {
            return $this->connectionLocator->getRead();
        }

        return $this->connectionLocator->getWrite();
    }
}